从电路复杂度视角揭示晶体等变GNN的表达能力极限
Fundamental Limits of Crystalline Equivariant Graph Neural Networks: A Circuit Complexity Perspective
- 用电路复杂度理论分析晶体等变GNN的计算能力边界
- 证明在合理资源下模型可被常数深度TC⁰电路模拟
- 为突破限制提供架构改进方向,适合对称学习研究者
图神经网络(GNN)已成为关系数据学习的核心范式。在材料科学中,等变图神经网络(EGNN)因其能保持欧几里得对称性和周期性边界条件,成为晶体结构预测的有力工具。尽管表现优异,其在周期性、对称性约束场景下的表达能力仍不明确。本文从电路复杂度视角刻画了EGNN在晶体结构预测中的内在计算与表达极限。分析了节点特征、原子坐标和晶格矩阵上各层的运算,证明在多项式精度、嵌入宽度d=O(n)(n个节点)、常数层深和常数深度、O(n)宽度的MLP实例化消息/更新/读出映射下,此类模型可被一个多项式大小的统一TC⁰阈值电路族模拟(具有显式的常数深度界)。将EGNN置于TC⁰框架中,为在现实资源约束下可解决的决策与预测问题提供了具体上限,并明确了需通过增加深度、丰富几何基元或拓宽层宽等修改才能突破此范式。该分析补充了无法直接应用于周期晶体的Weisfeiler-Lehman型结果,为晶体系统的对称感知图学习提供了复杂性理论基础。
原文摘要 · Abstract (English)
Graph neural networks (GNNs) have become a core paradigm for learning on relational data. In materials science, equivariant GNNs (EGNNs) have emerged as a compelling backbone for crystalline-structure prediction, owing to their ability to respect Euclidean symmetries and periodic boundary conditions. Despite strong empirical performance, their expressive power in periodic, symmetry-constrained settings remains poorly understood. This work characterizes the intrinsic computational and expressive limits of EGNNs for crystalline-structure prediction through a circuit-complexity lens. We analyze the computations carried out by EGNN layers acting on node features, atomic coordinates, and lattice matrices, and prove that, under polynomial precision, embedding width $d=O(n)$ for $n$ nodes, $O(1)$ layers, and $O(1)$-depth, $O(n)$-width MLP instantiations of the message/update/readout maps, these models admit a simulation by a uniform $\mathsf{TC}^0$ threshold-circuit family of polynomial size (with an explicit constant-depth bound). Situating EGNNs within $\mathsf{TC}^0$ provides a concrete ceiling on the decision and prediction problems solvable by such architectures under realistic resource constraints and clarifies which architectural modifications (e.g., increased depth, richer geometric primitives, or wider layers) are required to transcend this regime. The analysis complements Weisfeiler-Lehman style results that do not directly transfer to periodic crystals, and offers a complexity-theoretic foundation for symmetry-aware graph learning on crystalline systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。