Global Field Heterogeneous Graph Neural Networks for Accelerating Quantum Transport Calculation

Xiaoxin Xie,Zhijiang Wang,Yuchen Wang,Fei Liu
DOI: https://doi.org/10.1109/sispad62626.2024.10733057
2024-01-01
Abstract:In this work, we propose an attention-based global field heterogeneous graph neural network (GFGNN) to characterize global field and dynamic features of the open system, aiming to accelerate or even bypass the computationally demanding self-consistent iterations of NEGF and substantially improve the efficiency of quantum transport calculations. Representing the device with a heterogeneous graph largely preserves its intrinsic physical characteristics, while the global graph attention network effectively captures the propagation of nonlocal physical information and mitigates prediction accuracy issues due to device scaling. GFGNN has been verified to have strong predictive power on 2D Mos2 DG-MOSFETs, with MAE (mean absolute error) as low as 2. $1\sim 3.3\text{meV}$ for potential profile prediction. By incorporating GFGNN into the NEGF computing framework, an acceleration of 182.22 $\sim6$ 35.71 % can be achieved while maintaining the accuracy of transport property calculations.
What problem does this paper attempt to address?