Efficient Equivariant High-Order Crystal Tensor Prediction via Cartesian Local-Environment Many-Body Coupling
Dian Jin, Yancheng Yuan, Xiaoming Tao
摘要
End-to-end prediction of high-order crystal tensor properties from atomic structures remains challenging: while spherical-harmonic equivariant models are expressive, their Clebsch-Gordan tensor products incur substantial compute and memory costs for higher-order targets. We propose the Cartesian Environment Interaction Tensor Network (CEITNet), an approach that constructs a multi-channel Cartesian local environment tensor for each atom and performs flexible many-body mixing via a learnable channel-space interaction. By performing learning in channel space and using Cartesian tensor bases to assemble equivariant outputs, CEITNet enables efficient construction of high-order tensor. Across benchmark datasets for order-2 dielectric, order-3 piezoelectric, and order-4 elastic tensor prediction, CEITNet surpasses prior high-order prediction methods on key accuracy criteria while offering high computational efficiency.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper12
- E(n) Equivariant Graph Neural NetworksVictor Garcia Satorras, Emiel Hoogeboom, Max WellingICML 2021 · 被引用 1,432 次
- Equivariant message passing for the prediction of tensorial properties and molecular spectraKristof Schütt, Oliver T. Unke, Michael GasteggerICML 2021 · 被引用 736 次
- Scalars are universal: Equivariant machine learning, structured like classical physicsSoledad Villar, David W. Hogg, Kate Storey-Fisher, Weichi Yao 等NeurIPS 2021 · 被引用 185 次
- Periodic Graph Transformers for Crystal Material Property PredictionKeqiang Yan, Yi Liu, Yuchao Lin, Shuiwang JiNeurIPS 2022 · 被引用 167 次
- Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNsSaro Passaro, C. Lawrence ZitnickICML 2023 · 被引用 157 次
相关 Paper
- A Space Group Symmetry Informed Network for O(3) Equivariant Crystal Tensor PredictionKeqiang Yan, Alexandra Saxton, Xiaofeng Qian, Xiaoning Qian 等ICML 2024 · 被引用 13 次
- TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular PotentialsGuillem Simeon, Gianni De FabritiisNeurIPS 2023 · 被引用 100 次
- Higher-Rank Irreducible Cartesian Tensors for Equivariant Message PassingViktor Zaverkin, Francesco Alesiani, Takashi Maruyama, Federico Errica 等NeurIPS 2024 · 被引用 19 次
- Revisiting the Canonicalization for Fast and Accurate Crystal Tensor Property PredictionHaowei Hua, Jingwen Yang, Wanyu Lin, Pan ZhouAAAI 2026 · 被引用 1 次
- GotenNet: Rethinking Efficient 3D Equivariant Graph Neural NetworksSarp Aykent, Tian XiaICLR 2025
