Rapid Inverse Design of High Degree of Freedom Meta-Atoms Based on the Image-Parameter Diffusion Model

• Published in: Journal of lightwave technology Vol. 42; no. 15; pp. 5269 - 5278
• Main Authors: Zhu, Lu, Hua, Wei, Lv, Cong, Liu, Yuanyuan
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.08.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Algorithms; Atomic structure; Degrees of freedom; Design methodology; Design optimization; Design parameters; Diffusion model; Diffusion processes; Diffusion rate; Free form; Gaussian process; high DOFS; Image quality; Inverse design; Iterative methods; meta-atom; Metamaterials; Metasurfaces; Neural networks; Noise generation; Normal distribution; one-to-many; Optimization; Predictive models; Process parameters; Spectral sensitivity
• ISSN: 0733-8724; 1558-2213; 1558-2213
• DOI: 10.1109/JLT.2024.3391924

The inverse design of metamaterials is the mapping of spectral response (low dimensions) to the structure (high dimensions). In the process of inverse design, there will be a "one-to-many" problem (similar spectral responses are generated by multiple meta-atoms), especially for designs with high degrees of freedom (DOFs). Although traditional generative models (VAE, GAN) combined with optimization algorithms can convert inverse design into forward prediction to solve the "one-to-many" problem, this method requires multiple iterative calculations and has low design efficiency, making it difficult to achieve fast and accurate inverse design. This paper proposes a novel inverse design method based on the image-parameter diffusion model. Through a step-by-step process of adding noise that transforms the structural parameters and images into Gaussian distributions, our method can gradually remove the noise starting from the distribution and generate high DOFs meta-atoms that satisfy the transmission coefficient of additional inputs. Moreover, this method avoids multiple iterative optimizations and ensures more efficient and high-quality generation results. The experimental results show that our method solves the "one-to-many" problem of high-DOFs meta-atoms and can generate free-form structures with only one design iteration, which is superior to conventional design methods in terms of model generation speed, generation accuracy, and quality.