Towards universal property prediction in Cartesian space: TACE is all you need
2509.14961v1
stat.ML, cond-mat.mtrl-sci, cs.LG, physics.chem-ph
2025-09-20
Авторы:
Zemin Xu, Wenbo Xie, Daiqian Xie, P. Hu
Резюме на русском
#### Контекст
Машинное обучение (ML) на протяжении многих лет основывалось на сферических гармонических представлениях для моделирования атомных симуляций и исследований материалов. Однако, существуют значительные огранчивания в том, что большинство моделей не включают в себя геометрию молекул и свойств кристаллических структур, что может привести к неточностям или нестабильности в прогнозировании. Необходимость развития моделей, которые могут полностью использовать варианты взаимодействия и материальных свойств в тензорном виде, была очевидна для развития физично-фундаментальных и прикладных исследований.
#### Метод
"Tensor Atomic Cluster Expansion and Tensor Moment Potential" (TACE) представляет собой новую подходную модель, которая впервые предлагает представление тензорных свойств в целостном виде для получения материальных свойств. Она создана в виде гибкой архитектуры, позволяющей учитывать взаимодействия между атомами в пространстве Cartesian, что обеспечивает естественное взаимодействие геометрии и тензорных свойств. Для этого TACE использует декомпозицию атомных окружений в ирредуцируемые тензоры Cartesian, чтобы гарантировать симметрические представления, которые соответствуют квантовым и теоретическим ограничениям. Более того, TACE включает универсальные подключаемые модули для работы с различными атрибутами, такими как запасные значения, заряды, магнитные моменты и внешние поля. Это позволяет модели стабильно реагировать на внешние факторы, которые могут влиять на структуру и свойства материалов.
#### Результаты
TACE протестирована на широком спектре задач, включая моделирование физических и химических свойств, а также выполнение моделирования вне-области (out-of-domain benchmarks). Модель демонстрирует высокую точность и стабильность по сравнению с другими ведущими моделями в области материальных моделей. Например, в задачах геометрического моделирования, таких как анализ характеристик и отзывов, TACE показала значительное превосходство в высокоточном прогнозировании. Также, модель обеспечивает эффективный и точный прогноз в различных условиях, включая магнитные и заряженные системы.
#### Значимость
Новая модель TACE позволяет расширить границы ML в области атомных симуляций и исследований материалов. Она может использоваться в различных прикладных дисциплинах, включая материаловедение, химию и физику, чтобы получить более точные и стабильные прогнозы. Это открывает новые возможности для разработки новых материалов, проблем контроля материальных свойств, а также для улучшения существующих моделей.
#### Выводы
TACE полностью переносит сферич
Abstract
Machine learning has revolutionized atomistic simulations and materials
science, yet current approaches often depend on spherical-harmonic
representations. Here we introduce the Tensor Atomic Cluster Expansion and
Tensor Moment Potential, the first unified framework formulated entirely in
Cartesian space for the systematic prediction of arbitrary structure-determined
tensorial properties. TACE achieves this by decomposing atomic environments
into a complete hierarchy of (irreducible) Cartesian tensors, ensuring
symmetry-consistent representations that naturally encode invariance and
equivariance constraints. Beyond geometry, TACE incorporates universal
embeddings that flexibly integrate diverse attributes including basis sets,
charges, magnetic moments and field perturbations. This allows explicit control
over external invariants and equivariants in the prediction process. Long-range
interactions are also accurately described through the Latent Ewald Summation
module within the short-range approximation, providing a rigorous yet
computationally efficient treatment of electrostatic interactions. We
demonstrate that TACE attains accuracy, stability, and efficiency on par with
or surpassing leading equivariant frameworks across finite molecules and
extended materials, including in-domain and out-of-domain benchmarks, spectra,
hessians, external-field response, charged systems, magnetic systems,
multi-fidelity training, and heterogeneous catalytic systems. Crucially, TACE
bridges scalar and tensorial modeling and establishes a Cartesian-space
paradigm that unifies and extends beyond the design space of
spherical-harmonic-based methods. This work lays the foundation for a new
generation of universal atomistic machine learning models capable of
systematically capturing the rich interplay of geometry, fields and material
properties within a single coherent framework.