PRIME: Physics-Related Intelligent Mixture of Experts for Transistor Characteristics Prediction
Zhenxing Dou, Yijiao Wang, Tao Zou, Zhiwei Chen, Fei Liu, Peng Wang, Weisheng Zhao
TL;DR
This work tackles the difficulty of accurately predicting transistor I-V characteristics across multiple operating regions by introducing PRIME, a physics-related intelligent mixture of experts. PRIME replaces a static, single-model approach with a three-expert mixture whose gating network dynamically assigns region-specific importance, yielding substantial improvements (60%–84%) over state-of-the-art models on gate-all-around transistors. By incorporating high-order derivative regularization and leveraging TCAD-derived data, PRIME demonstrates robust generalization across shapes and holds promise for more accurate and efficient circuit simulation via SPICE-like analysis. The approach bridges physics knowledge and data-driven learning to deliver precise, region-aware transistor modeling suitable for design and fabrication workflows.
Abstract
In recent years, machine learning has been extensively applied to data prediction during process ramp-up, with a particular focus on transistor characteristics for circuit design and manufacture. However, capturing the nonlinear current response across multiple operating regions remains a challenge for neural networks. To address such challenge, a novel machine learning framework, PRIME (Physics-Related Intelligent Mixture of Experts), is proposed to capture and integrate complex regional characteristics. In essence, our framework incorporates physics-based knowledge with data-driven intelligence. By leveraging a dynamic weighting mechanism in its gating network, PRIME adaptively activates the suitable expert model based on distinct input data features. Extensive evaluations are conducted on various gate-all-around (GAA) structures to examine the effectiveness of PRIME and considerable improvements (60\%-84\%) in prediction accuracy are shown over state-of-the-art models.
