Diverse ranking metamaterial inverse design based on contrastive and transfer learning

Abstract

Metamaterials, thoughtfully designed, have demonstrated remarkable success in the manipulation of electromagnetic waves. More recently, deep learning can advance the performance in the field of metamaterial inverse design. However, existing inverse design methods based on deep learning often overlook potential trade-offs of optimal design and outcome diversity. To address this issue, in this work we introduce contrastive learning to implement a simple but effective global ranking inverse design framework. Viewing inverse design as spectrum-guided ranking of the candidate structures, our method creates a resemblance relationship of the optical response and metamaterials, enabling the prediction of diverse structures of metamaterials based on the global ranking. Furthermore, we have combined transfer learning to enrich our framework, not limited in prediction of single metamaterial representation. Our work can offer inverse design evaluation and diverse outcomes. The proposed method may shrink the gap between flexibility and accuracy of on-demand design.