Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Spectral/Spatial Tensor Atomic Cluster Expansion with Universal Embeddings in Cartesian Space

About

Equivariant atomistic machine learning models have largely been built on spherical-tensor representations, where explicit angular-momentum coupling introduces substantial complexity and systematic extensions beyond energies and forces remain challenging, often requires problem-specific architectural choices. Here we introduce the Tensor Atomic Cluster Expansion (TACE), which unifies scalar and tensorial modeling in Cartesian and space by decomposing local environments into irreducible Cartesian tensors (ICT) constructing a controlled many-body hierarchy with atomic cluster expansion (ACE). In addition to performing ACE in the frequency domain, we propose an efficient Clebsch-Gordan-free alternative in the spatial domain. TACE provides universal invariant (e.g., fidelity tags and charges) and equivariant (e.g., external electric fields and non-collinear magnetic moments) embeddings and predicted tensorial observables are handled on equal footing and enabling explicit control at inference. We demonstrate the accuracy, stability, and efficiency across finite molecules and extended materials, including in-domain and out-of-domain benchmarks, spectra, Hessian, external-field responses, charged systems, and multi-fidelity/head training. We further show its robustness on nonequilibrium/reactive datasets and controlled scaling when extending to large foundation model datasets.

Zemin Xu, Wenbo Xie, P. Hu• 2025

Related benchmarks

TaskDatasetResultRank
Energy barrier predictionAcetylene hydrogenation on Pd(111) and PdAg surfaces 1.0 (test)
Energy Barrier (eV)0.85
32
Energy and force prediction3BPA (Dihedral slices)
Energy RMSE7.6
16
Interatomic Potential Prediction3BPA 300 K
Energy RMSE (meV)2.69
16
Interatomic Potential Prediction3BPA 600 K
Energy RMSE (meV)8.1
16
Interatomic Potential Prediction3BPA 1200 K
Energy RMSE (meV)23.42
16
Interatomic Potential PredictionLiquid water 32-water system revPBE0-D3 (test)
Energy RMSE (meV/atom)0.58
11
Atomic distance predictionAg(111) surface adsorption
Main Chain Carbon Distance (Å)2.97
10
Interatomic potential modelingGAP 17 (test)
RMSE (Energy)8.55
7
Energy barrier predictionPdAg surfaces
MAE0.07
7
Dipole moment predictionMonomer water system (val)
RRMSE0.6
6
Showing 10 of 23 rows

Other info

Follow for update