用对称性约束的神经网络精准预测石墨烯能带结构
Symmetry-Constrained Multi-Scale Physics-Informed Neural Networks for Graphene Electronic Band Structure Prediction
- 通过多头架构强制晶体对称性,分路径学习狄拉克物理与鞍点特征
- 训练损失下降99.99%,狄拉克点间隙误差仅30.3 μeV,平均误差低于54 meV
- 适合材料科学加速发现,尤其关注二维材料电子结构的研究者
精确预测二维材料的电子能带结构仍是基础挑战,现有方法难以兼顾计算效率与物理准确性。我们提出对称性约束的多尺度物理信息神经网络SCMS-PINN v35,直接学习石墨烯能带结构,并通过多头架构严格强制晶格对称性。模型引入三个专用ResNet-6路径——K-head用于狄拉克物理,M-head用于鞍点,General head用于平滑插值——在31个从波矢提取的物理信息特征上运行。采用渐进式狄拉克约束调度,权重参数从5.0逐步增至25.0,实现从全局拓扑到局部关键物理的分层学习。在10,000个波矢上训练300轮,训练损失降低99.99%(从34.597降至0.003),验证损失为0.0085。模型预测狄拉克点间隙误差仅30.3 μeV,价带和导带平均误差分别为53.9 meV和40.5 meV。所有十二个C$_{6v}$对称操作通过系统平均强制执行,确保对称性完全保留。该框架为拓展物理信息学习至更广泛的二维材料提供了基础,助力加速材料发现。
原文摘要 · Abstract (English)
Accurate prediction of electronic band structures in two-dimensional materials remains a fundamental challenge, with existing methods struggling to balance computational efficiency and physical accuracy. We present the Symmetry-Constrained Multi-Scale Physics-Informed Neural Network (SCMS-PINN) v35, which directly learns graphene band structures while rigorously enforcing crystallographic symmetries through a multi-head architecture. Our approach introduces three specialized ResNet-6 pathways -- K-head for Dirac physics, M-head for saddle points, and General head for smooth interpolation -- operating on 31 physics-informed features extracted from k-points. Progressive Dirac constraint scheduling systematically increases the weight parameter from 5.0 to 25.0, enabling hierarchical learning from global topology to local critical physics. Training on 10,000 k-points over 300 epochs achieves 99.99\% reduction in training loss (34.597 to 0.003) with validation loss of 0.0085. The model predicts Dirac point gaps within 30.3 $μ$eV of theoretical zero and achieves average errors of 53.9 meV (valence) and 40.5 meV (conduction) across the Brillouin zone. All twelve C$_{6v}$ operations are enforced through systematic averaging, guaranteeing exact symmetry preservation. This framework establishes a foundation for extending physics-informed learning to broader two-dimensional materials for accelerated discovery.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。