---
title: CSP-MACE-Å for Faster Crystal Structure Prediction
url: https://www.emergentmind.com/topics/csp-mace-a
type: topic
---

# CSP-MACE-Å for Faster Crystal Structure Prediction

Searching arXiv for the cited paper and closely related models to ground the article in current literature.
CSP-MACE-Å is a machine learning interatomic potential for crystal structure prediction (CSP) that is intended to replace density functional theory (DFT) in the evaluation and reranking of molecular crystal candidates. Its defining feature is an additive decomposition of the total lattice energy into intramolecular and intermolecular components, with the intermolecular term itself partitioned into a MACE-POLAR contribution, an analytical XDM-style dispersion correction, and a learned residual $\Delta$ model trained to reproduce B86bPBE-XDM intermolecular energies. In the reported evaluation, the method achieves performance comparable to PBE DFT with the Neumann-Perrin dispersion correction on an AstraZeneca set and performance close to B86bPBE-XDM DFT on a blind-test set, while running multiple orders of magnitude faster than DFT [2605.28905].

## 1. Definition and scope

CSP-MACE-Å is presented as a machine learning interatomic potential designed for crystal structure prediction workflows in which large numbers of candidate molecular crystal structures must be relaxed, ranked, and, in some cases, reranked by approximate free energies rather than lattice energies alone [2605.28905]. The central objective is to provide DFT-level ranking fidelity at substantially lower computational cost.

The model is formulated around the decomposition
$$
E_{\rm total} = E_{\rm intra} + E_{\rm inter}\,.
$$
Here, $E_{\rm intra}$ is the intramolecular, or “monomer,” contribution, defined as the sum of gas-phase energies of each molecule in the cell, while $E_{\rm inter}$ is the intermolecular, or “binding,” contribution [2605.28905]. This decomposition is not merely notational; it structures both model construction and training. The intramolecular part is handled by a MACE-POLAR model trained on single-molecule data, whereas the intermolecular part is assembled from physically and statistically distinct terms.

This suggests that CSP-MACE-Å is tailored to the particular error structure of molecular-crystal energetics: intramolecular conformational energetics, short-to-mid-range intermolecular interactions, long-range dispersion, and residual DFT-level corrections are treated as separable modeling targets rather than collapsed into a single universal lattice-energy predictor.

## 2. Energy decomposition and model architecture

The intramolecular component $E_{\rm intra}$ is predicted by the MACE-POLAR architecture in its “medium” size configuration [2605.28905]. MACE-POLAR is described as consisting of a stack of higher-order equivariant message-passing layers, a long-range electrostatics module using Ewald or multipole summation of learned partial charges, and a final feed-forward network mapping per-atom embeddings to atomic energy contributions $e_i$, such that
$$
E_{\rm intra}=\sum_{i\in\,\text{all atoms in isolated molecules}} e_i(\mathbf{h}_i)\,.
$$
In this definition, the molecular energies are evaluated in the gas phase and summed across the molecules in the crystal cell [2605.28905].

The intermolecular contribution is split into three parts:
$$
E_{\rm inter}
=
E_{\rm inter}^{\rm MACE\text{-}POLAR}
+
E^{\rm XDM}_{\rm disp}
+
\Delta E_{\rm inter}\,.
$$
The first term, $E_{\rm inter}^{\rm MACE\text{-}POLAR}$, is defined as the total MACE-POLAR energy on the crystal minus the gas-phase $E_{\rm intra}$ contribution [2605.28905]. The second term is a fixed, pairwise-additive XDM-style dispersion correction with the functional form
$$
E^{\rm XDM}_{\rm disp}
=
-\tfrac12
\sum_{n\in\{6,8,10\}}
\sum_{i<j}
\frac{C_{n,ij}}{R_{ij}^n + R_{\rm vdW,ij}^n}\,,
$$
where $R_{ij}$ is the interatomic distance, $C_{6,ij}$, $C_{8,ij}$, and $C_{10,ij}$ are fixed averaged XDM coefficients, and $R_{\rm vdW,ij}$ is the sum of van der Waals radii for atoms $i$ and $j$ [2605.28905].

The third term is a learned residual:
$$
\Delta E_{\rm inter}
=
E_{\rm inter}^{\rm DFT}
-
\bigl(
E_{\rm inter}^{\rm MACE\text{-}POLAR}
+
E^{\rm XDM}_{\rm disp}
\bigr)\,.
$$
Residual targets of the same form are also used for forces and virials [2605.28905]. In effect, the model places physically motivated structure into the intermolecular energy and reserves machine learning capacity for the remaining discrepancy with B86bPBE-XDM.

A plausible implication is that the architecture is designed to reduce the burden on the residual learner by isolating the most systematic long-range component analytically and the monomer physics through a separate high-capacity intramolecular model.

## 3. Training data and optimization procedure

The intramolecular model is trained on the OMol25 dataset, which contains 100 million single-molecule DFT calculations at the $\omega$B97M-V/def2-TZVPD level of theory [2605.28905]. The chemistry is described as having broad coverage of small and medium-sized organic molecules, including heterocycles and aliphatics, and the targets are total energies, atomic forces, and optionally dipoles [2605.28905]. These characteristics establish the monomer term as a gas-phase, electronically detailed component trained on very large-scale DFT data.

The residual intermolecular model is trained on 50 000 periodic B86bPBE-XDM crystal DFT calculations with energies, forces, and stresses [2605.28905]. The residuals are computed as
$$
\Delta E=E^{\rm DFT}-E^{\rm MACE\text{-}POLAR}-E^{\rm XDM}_{\rm disp}\,.
$$
The $\Delta$ model itself is a MACE-POLAR “medium” architecture trained on these residuals with mean-squared-error losses on energy, forces, and stress [2605.28905]. The reported optimization details are Adam, a typical learning rate of $10^{-3}$, early stopping after convergence on a held-out 5% set, and a batch size of approximately one crystal per batch [2605.28905].

The separation between OMol25-based monomer training and crystal-based residual training is a central procedural feature. It assigns broad intramolecular chemical coverage to the monomer model and crystal-specific DFT correction to the intermolecular residual model. This suggests a hybrid transfer strategy in which single-molecule electronic structure data and periodic crystal data are combined without forcing a single model to learn both regimes from scratch.

## 4. Evaluation protocol and quantitative performance

The reported evaluation uses two test suites [2605.28905]. The first, the “AZ Set,” comprises 19 AstraZeneca compounds, described as 17 small molecules plus 1 salt, each with approximately 1 000 crystal candidates from a force-field prescreen. The second, the “Blind-Test Set,” comprises 28 compounds from the first through seventh CSP blind tests and includes neutral organics, salts, cocrystals, flexible molecules, and polymorphs [2605.28905]. The key metrics are the rank of the experimental structure, where 1 is best, and $\Delta E$, the energy gap in kJ/mol to the lowest-energy candidate [2605.28905].

The average results reported for the two sets are as follows.

| Evaluation set | Model | $\langle{\rm Rank}\rangle$, $\langle \Delta E\rangle$ |
|---|---|---|
| AZ Set (19 compounds) | PBE + Neumann–Perrin (E) | 3.68, 0.68 kJ/mol |
| AZ Set (19 compounds) | CSP-MACE-Å (E) | 3.58, 0.66 kJ/mol |
| AZ Set (19 compounds) | CSP-MACE-Å (A) | 2.11, 0.36 kJ/mol |
| AZ Set (19 compounds) | UMA-OMC (E) | 25.26, 1.75 kJ/mol |
| AZ Set (19 compounds) | UMA-OMC (A) | 23.26, 1.59 kJ/mol |
| AZ Set (19 compounds) | MACE-POLAR-1 (E) | 5.58, 1.32 kJ/mol |
| AZ Set (19 compounds) | MACE-POLAR-1 (A) | 3.00, 0.58 kJ/mol |
| Blind-Test Set (28 compounds) | B86bPBE-XDM (E) | 3.25, 0.91 kJ/mol |
| Blind-Test Set (28 compounds) | CSP-MACE-Å (E) | 3.86, 0.94 kJ/mol |
| Blind-Test Set (28 compounds) | CSP-MACE-Å (A) | 2.96, 0.80 kJ/mol |
| Blind-Test Set (28 compounds) | UMA-OMC (E) | 3.89, 1.11 kJ/mol |
| Blind-Test Set (28 compounds) | UMA-OMC (A) | 3.50, 0.89 kJ/mol |
| Blind-Test Set (28 compounds) | MACE-POLAR-1 (E) | 7.46, 1.85 kJ/mol |
| Blind-Test Set (28 compounds) | MACE-POLAR-1 (A) | 5.43, 1.18 kJ/mol |

On the AstraZeneca set, CSP-MACE-Å with lattice energies alone is reported to be comparable to PBE DFT with the Neumann-Perrin dispersion correction, with average rank 3.58 versus 3.68 and average $\Delta E$ 0.66 versus 0.68 kJ/mol [2605.28905]. On the blind-test set, CSP-MACE-Å with lattice energies is reported to be close to B86bPBE-XDM DFT, with average rank 3.86 versus 3.25 and average $\Delta E$ 0.94 versus 0.91 kJ/mol [2605.28905].

Across the full evaluation suite, CSP-MACE-Å is reported to outperform both MACE-POLAR-1 and UMA-OMC [2605.28905]. Because the comparison spans neutral organics, salts, cocrystals, flexible molecules, and polymorph systems, the performance claim is not restricted to a narrow crystal subclass.

## 5. Harmonic free-energy reranking and temperature dependence

A major aspect of the reported workflow is reranking by the Helmholtz free energy under the harmonic approximation rather than by static lattice energy alone [2605.28905]. The free energy is evaluated as
$$
A_{\rm harm}(T)
=
E_{\rm lattice}
+
k_B T \sum_i
\ln\!\bigl(\tfrac{\hbar\omega_i}{k_B T}\bigr)\,,
$$
where $\{\omega_i\}$ are phonon frequencies obtained by finite-difference supercell diagonalization [2605.28905].

In both evaluation sets, reranking with harmonic free energies improves performance relative to ranking by energy alone [2605.28905]. Across both sets, reranking by $A_{\rm harm}$ reduces the average rank by approximately 30% and lowers $\Delta E$ by approximately 40% [2605.28905]. Numerically, on the AstraZeneca set CSP-MACE-Å improves from average rank 3.58 and $\Delta E$ 0.66 kJ/mol in energy ranking to 2.11 and 0.36 kJ/mol in harmonic free-energy ranking; on the blind-test set it improves from 3.86 and 0.94 kJ/mol to 2.96 and 0.80 kJ/mol [2605.28905].

The method was also applied to five polymorphic systems, including sulfathiazole, mexiletine·HCl, and AZD1305, over the temperature range 0–600 K under the harmonic model [2605.28905]. CSP-MACE-Å is reported to reproduce the qualitative ordering and phase-transition trends in all five cases, with errors in exact transition temperatures of approximately 20–50 K [2605.28905].

These results place harmonic phonon contributions at the center of the model’s practical utility rather than as a peripheral post-processing step. A plausible implication is that the main gain from CSP-MACE-Å is not only cheaper lattice-energy evaluation but also the expansion of free-energy-based reranking to a materially larger subset of candidate structures.

## 6. Comparative position among related machine-learning potentials

The comparison against MACE-POLAR-1 and UMA-OMC clarifies the design choices of CSP-MACE-Å [2605.28905]. MACE-POLAR-1 is described as lacking an explicit long-range dispersion term and as being trained only on gas-phase OMol25; it underbinds crystals, with $\Delta E\sim1.3$ kJ/mol error and rank $\sim6$ in the AZ set [2605.28905]. UMA-OMC is described as being trained on $\sim25$ million PBE-D3 crystal configurations and as capturing dispersion implicitly, but showing larger errors, with $\Delta E\sim1.7$ kJ/mol and rank $\sim25$ on the AZ set [2605.28905].

Within this comparative framing, CSP-MACE-Å is characterized by a hybrid design: high-quality intramolecular modeling, explicit XDM dispersion, and a crystal-trained residual correction [2605.28905]. The reported results indicate that this combination yields near-DFT crystal-ranking accuracy while avoiding two distinct failure modes suggested by the baselines: insufficient intermolecular attraction in the absence of explicit long-range dispersion, and lower ranking fidelity when dispersion is learned only implicitly from crystal data.

A common misconception would be to interpret CSP-MACE-Å as a pure end-to-end lattice-energy network. The reported construction does not support that reading. Instead, it is an additive composite potential whose components reflect distinct physical and data-driven roles: monomer energetics, analytical dispersion, and residual correction to B86bPBE-XDM-level intermolecular energetics [2605.28905].

## 7. Computational implications for crystal structure prediction workflows

The reported acceleration relative to DFT is substantial [2605.28905]. Single-point evaluation plus geometry optimization under CSP-MACE-Å is stated to be approximately $10^3$–$10^4\times$ faster than DFT, while free-energy evaluation including phonons is approximately $10^2$–$10^3\times$ faster [2605.28905]. This speedup enables relaxation and energy ranking of $\mathcal{O}(10^3)$ candidates, in contrast to $\mathcal{O}(10^2)$ for DFT, and renders free-energy reranking of the top 50–100 structures feasible [2605.28905].

In the context of CSP, where force-field prescreens often produce large candidate pools and the experimentally realized polymorph may not rank first under approximate lattice energies, this computational regime changes the practical allocation of accuracy. Rather than restricting high-level calculations to a very small set of structures, CSP-MACE-Å permits a larger portion of the search space to be evaluated at DFT-comparable accuracy and a nontrivial subset to be reranked by harmonic free energies.

The practical significance described for this workflow is “greater confidence when derisking solid forms” [2605.28905]. This suggests a role not only in academic benchmark studies but also in industrial solid-form assessment, where salts, cocrystals, and polymorph stability trends are operationally important. The reported results indicate that CSP-MACE-Å is positioned as a surrogate for DFT within CSP pipelines rather than merely as a prescreening heuristic, with its principal contribution lying in the combination of DFT-level ranking behavior, harmonic free-energy capability, and multiple-orders-of-magnitude speedup [2605.28905].

Source: https://www.emergentmind.com/topics/csp-mace-a