一个模型搞定所有晶体对称性,自动适配230种空间群。
A Single Architecture for Representing Invariance Under Any Space Group
- 用傅里叶基约束设计统一架构,自动适应任意空间群
- 在材料性质预测上表现媲美专用模型,支持零样本迁移
- 适合材料科学、凝聚态物理中的对称性建模任务
将已知对称性融入机器学习模型可显著提升预测精度、鲁棒性和泛化能力。然而,实现特定对称性的精确不变性通常需为每个对称群定制专用架构,限制了可扩展性并阻碍了相关对称性间的知识迁移。在晶体学中,这种挑战尤为突出——三维空间共有230种空间群。本文提出一种新方法,构建单一机器学习架构,能自动调整权重以实现任意输入空间群的不变性。该方法基于显式刻画群操作对傅里叶系数的约束,构造对称适应的傅里叶基。将这些约束编码至神经网络层,实现不同空间群间的权重重用,使模型能够利用群间的结构相似性,在特定群数据稀缺时仍具表现力。我们在材料性质预测任务中验证了该方法的有效性,实现了与专用模型相当的性能,并成功完成零样本学习,推广至未见空间群。
原文摘要 · Abstract (English)
Incorporating known symmetries in data into machine learning models has consistently improved predictive accuracy, robustness, and generalization. However, achieving exact invariance to specific symmetries typically requires designing bespoke architectures for each group, limiting scalability and preventing knowledge transfer across related symmetries. In the case of the space groups, symmetries critical to modeling crystalline solids in materials science and condensed matter physics, this challenge is particularly salient as there are 230 such groups in three dimensions. In this work we present a new approach to such crystallographic symmetries by developing a single machine learning architecture that is capable of adapting its weights automatically to enforce invariance to any input space group. Our approach is based on constructing symmetry-adapted Fourier bases through an explicit characterization of constraints that group operations impose on Fourier coefficients. Encoding these constraints into a neural network layer enables weight sharing across different space groups, allowing the model to leverage structural similarities between groups and overcome data sparsity when limited measurements are available for specific groups. We demonstrate the effectiveness of this approach in achieving competitive performance on material property prediction tasks and performing zero-shot learning to generalize to unseen groups.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。