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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Energy barrier prediction | Acetylene hydrogenation on Pd(111) and PdAg surfaces 1.0 (test) | Energy Barrier (eV)0.85 | 32 | |
| Energy and force prediction | 3BPA (Dihedral slices) | Energy RMSE7.6 | 16 | |
| Interatomic Potential Prediction | 3BPA 300 K | Energy RMSE (meV)2.69 | 16 | |
| Interatomic Potential Prediction | 3BPA 600 K | Energy RMSE (meV)8.1 | 16 | |
| Interatomic Potential Prediction | 3BPA 1200 K | Energy RMSE (meV)23.42 | 16 | |
| Interatomic Potential Prediction | Liquid water 32-water system revPBE0-D3 (test) | Energy RMSE (meV/atom)0.58 | 11 | |
| Atomic distance prediction | Ag(111) surface adsorption | Main Chain Carbon Distance (Å)2.97 | 10 | |
| Interatomic potential modeling | GAP 17 (test) | RMSE (Energy)8.55 | 7 | |
| Energy barrier prediction | PdAg surfaces | MAE0.07 | 7 | |
| Dipole moment prediction | Monomer water system (val) | RRMSE0.6 | 6 |