XG-Attention-WGAN PIC: Utilizing XGboost-Attention-WGAN for Photonics Integrated Circuit Design
Mahmood Hasani, Atena Shircharandabi, Masiha Rabiei, Mobin Motaharifar, Mehrdad Ghasemi, Yaser Mike Banad, Sarah Sharif
Published: 2025/10/4
Abstract
Photonic Integrated Circuits (PICs) are fundamental for optical computing, communication, quantum information processing, and precision sensing. However, traditional numerical simulations for designing PIC components are computationally intensive and struggle with high-dimensional parameter spaces. This paper introduces XG-Attention-WGAN PIC, a novel framework that synergistically combines Wasserstein Generative Adversarial Networks (WGANs) with eXtreme Gradient Boosting (XGBoost) to enhance parameter prediction and inverse design in PICs. We utilize Finite-Difference Time-Domain simulations to generate high-fidelity training data, which is augmented by WGAN-generated synthetic data, yielding a root mean squared error (RMSE) of 0.26089. When integrated with XGBoost, this error is reduced to 0.008. The integration of a 64-head self-attention mechanism within the WGAN generator significantly improves data quality and model efficiency over 1000 training epochs. Demonstrated on microring resonators, our approach not only achieves superior prediction accuracy and design optimization but also autonomously discovers a novel, experimentally realizable geometry with enhanced Q-factor performance. The proposed framework provides a scalable, data-driven strategy for developing high-performance PIC components, with promising implications for quantum computing and advanced optical systems.