Metric Learning: Harnessing the Power of Machine Learning in Nanophotonics

• Published in: ACS photonics 2023-04, Vol.10 (4), p.900-909
• Main Authors: Zandehshahvar, Mohammadreza, Kiarashi, Yashar, Zhu, Muliang, Bao, Daqian, H Javani, Mohammad, Pourabolghasem, Reza, Adibi, Ali
• Format: Article
• Language: English
• ISSN: 2330-4022; 2330-4022
• DOI: 10.1021/acsphotonics.2c01331

We present a novel metric-learning approach based on combined triplet loss and mean-squared error for providing more functionality (e.g., more effective similarity measures) to the machine-learning algorithms used for the knowledge discovery and inverse design of nanophotonic structures compared to commonly used mean-squared error and mean-absolute error. We demonstrate the main shortcoming of the existing metrics (or loss functions) in mapping the nanophotonic responses into lower-dimensional spaces in keeping similar responses close to each other. We show how a systematic metric-learning paradigm can resolve this issue and provide physically interpretable mappings of the nanophotonic responses while facilitating the visualization. The presented metric-learning paradigm can be combined with almost all existing machine-learning and deep-learning approaches for the investigation of nanophotonic structures. Thus, the results of this paper can have a transformative impact on using machine learning and deep learning for knowledge discovery and inverse design in nanophotonics.