---
title: CHGNet Foundation Potential
url: https://www.emergentmind.com/topics/chgnet-foundation-potential
type: topic
---

# CHGNet Foundation Potential

The CHGNet Foundation Potential is a universal, charge-aware graph neural network (GNN) machine learning interatomic potential designed to encode both atomic and electronic degrees of freedom within a single framework. Its core architecture, data-driven pretraining, and fine-tuning strategies position it as a foundation model for modeling chemically and structurally diverse materials systems with near-density functional theory (DFT) accuracy and vastly reduced computational cost. CHGNet’s integration of magnetic moments as proxies for local orbital occupancies allows it to capture charge effects and phase behavior inaccessible to conventional MLIPs.

## 1. Architecture and Mathematical Formulation

CHGNet represents each atom as a node in a graph, with edges encoding pairwise distances and angular information. The initial node embeddings are derived from atomic numbers, and edges use radial Bessel expansions of interatomic distances. Multiple message-passing layers update atom and edge features through nonlinear functions and linear transformations. The architecture also processes three-body angular terms using a bond (angle) graph, with angular features expanded in a Fourier basis.

Mathematically, after $T$ message-passing iterations, each atom $i$ has a latent descriptor $h_i = v_i^{(T)} \in \mathbb{R}^{64}$, processed through a feed-forward “energy head” to yield atomic energy contributions $\epsilon_i$. The total energy is $E_{\text{CHGNet}}(R) = \sum_{i=1}^N \epsilon_i$. Forces and stress tensors are obtained by autodifferentiation with respect to atomic positions and strain, respectively.

A unique feature is the explicit prediction of site magnetic moments $m_i = L_m(v_i^{(3)})$, which regularizes the node embeddings to encode orbital occupancy and charge effects. These magnetic moments are not static inputs but are part of the output heads during multitask training, encouraging correlation between predicted atom environments and their electronic configuration [2302.14231, 2502.21179].

## 2. Foundation Model Pretraining

CHGNet is pretrained on the Materials Project Trajectory (MPtrj) dataset, which contains $\sim$1.5 million inorganic structure snapshots encompassing DFT-computed energies, forces, stresses, and magnetic moments for $\sim$94 chemical elements. The pretraining loss is a weighted mean squared error (or Huber loss) over these four targets:
\[
\mathcal{L} = \text{MSE}(E^\text{pred}, E^\text{DFT}) + w_F\,\text{MSE}(\mathbf{F}^\text{pred}, \mathbf{F}^\text{DFT}) + w_{\sigma}\,\text{MSE}(\sigma^\text{pred}, \sigma^\text{DFT}) + w_m\,\text{MSE}(m^\text{pred}, m^\text{DFT})
\]
with typical weights $w_F = 1$, $w_\sigma = 0.1$, and $w_m = 0.1$. Optimization is performed using Adam or RAdam with initial learning rates near $10^{-3}$ [2302.14231, 2507.02334].

On the full MPtrj test set, pretrained CHGNet attains mean absolute errors of 30 meV/atom (energy), 77 meV/Å (forces), 0.348 GPa (stress), and 0.032 $\mu_B$ (magmom) [2302.14231].

## 3. Fine-Tuning and Domain Adaptation

The foundation CHGNet can be efficiently fine-tuned for specific materials classes or underrepresented chemical environments. When applied to high-throughput discovery of Li-ion conductors, CHGNet is further refined on $\sim$2,800 DFT-computed transition-state geometries (generated by running CHGNet-NEB on Li-quaternary systems, filtered for relevant chemistries and symmetries) [2507.02334].

Fine-tuning retains the GNN architecture and loss structure, using the same MSE-based targets with normalized weights, a RAdam optimizer (learning rate $1 \times 10^{-3}$), and training for 500 epochs. Final held-out test performance is sharply improved relative to the parent model, achieving MAE$_E = 2$ meV/atom, MAE$_F = 13$ meV/Å, and MAE$_{\sigma} = 13$ mGPa. Critically, the mean absolute error for predicted migration barriers on the NEB-derived transition state dataset drops from 0.24 to 0.07 eV (train) and 0.23 to 0.09 eV (test), with $R^2$ increasing from 0.94 to 0.98 [2507.02334].

Systematic softening (i.e., force underestimation) sometimes observed in pretrained uMLIPs is mitigated by fine-tuning on targeted DFT data. For e.g., in layered WS$_2$ and MoS$_2$, fine-tuning on $\sim$100 DFT structures brings force errors down to 100 meV/Å and energy errors to $\sim$1 meV/atom, verifying performance against both DFT and experimental EXAFS spectra [2509.08498].

## 4. Integration with Atomistic Simulation Workflows

CHGNet can be deployed as a plug-in force provider for atomistic simulation packages (e.g., ASE) in both NEB and MD workflows.

**NEB workflow** (automated):

1. Structure generation: CIF→POSCAR conversion, $\sim$10 Å supercell construction.
2. Enumeration: All $N \times N$ hops among $N$ inequivalent Li Wyckoff sites, both forward and reverse.
3. Path interpolation: Image-dependent pair potential (IDPP).
4. Force evaluation: 7-image CI-NEB with CHGNet calculator, using spring forces and convergence for $|F| \lesssim 0.1$ eV/Å.
5. Post-processing: Extraction of migration barriers, statistical analysis over all distinct hops.

**MD workflow**:

- Ensemble: NVT with Nosé–Hoover thermostat, 1 fs (CHGNet) or 2 fs (AIMD) timestep.
- Conductivity: 3$\times$200 ps trajectories at each $T$, splitting trajectory for mean square displacement (MSD) analysis, extracting diffusion coefficient $D(T)$ and Arrhenius activation energy. Room-temperature ionic conductivity is estimated via the Nernst–Einstein relation $\sigma(T) = n e^2 z^2 D(T)/(k_B T)$ [2507.02334].

CHGNet’s deployment yields $\sim$100$\times$ speedup for NEB (10$^4$ s/path DFT+CI-NEB vs 10$^2$ s CHGNet-NEB) and $\sim$200$\times$ greater sampling in MD (1.7 h/1,000 AIMD steps vs 2$\times$10$^5$ CHGNet steps in 3.2 h) [2507.02334].

## 5. Benchmarking, Validation, and Performance Comparison

**Energy and force metrics**: Fine-tuned CHGNet reduces energy MAE from 9 → 2 meV/atom and force MAE from 24 → 13 meV/Å (transition-state set) [2507.02334].

**Migration barrier prediction**: For NEB-calculated Li$^+$ migration in Li$_2$MnO$_3$, fine-tuning lowers MAE from 0.21 (pretrained) to 0.056 eV; for LiTi$_2$(PO$_4$)$_3$, fine-tuned CHGNet achieves 0.40 eV vs DFT 0.38 eV [2507.02334].

**Diffusivity and conductivity**: In NASICON Li$_{1.5}$Al$_{0.5}$Ti$_{1.5}$(PO$_4$)$_3$, CHGNet-MD gives $E_a=0.21$ eV ($E_a^\text{AIMD}=0.22$ eV) and direct MD $\sigma$(300 K)$=2.8$ mS/cm (vs 5.1 mS/cm from AIMD), whereas the pretrained model overestimates mobility ($E_a\approx 0.15$ eV, $\sigma\approx 57$ mS/cm) [2507.02334].

**Experimental validation**: In layered chalcogenides (WS$_2$, MoS$_2$), fine-tuning on 50–100 DFT frames delivers EXAFS agreement at the level of experimental uncertainty for Debye–Waller factors and spectral amplitudes, requiring only minor computational resources for retraining [2509.08498].

**Phonon properties**: CHGNet achieves force RMSE (131.76 meV/Å) comparable to state-of-the-art uMLPs but exhibits higher errors for second/third-order IFCs (7.79 eV/Å$^2$ and 15.98 eV/Å$^3$) and large errors predicting lattice thermal conductivity (MAE$\sim$1 dex versus DFT/experiment). This suggests that, in its current form, CHGNet underperforms as a universal force provider for phonon-mediated thermal transport predictions relative to models such as EquiformerV2 [2509.03401].

## 6. Discovery and Screening of New Functional Materials

High-throughput CHGNet-accelerated NEB/MD calculations enable enumeration and rapid screening of candidate fast-Li$^+$ ion conductors. In a materials discovery campaign, 66 compounds with at least one Li$^+$ hop barrier <0.5 eV were identified. Notably, orthorhombic Pnma polymorphs (LiMgPO$_4$ with 0.27 eV and LiTiPO$_5$ with 0.28 eV barriers), as well as their aliovalent-doped derivatives (Li$_{0.5}$Mg$_{0.5}$Al$_{0.5}$PO$_4$, Li$_{0.5}$TiPO$_{4.5}$F$_{0.5}$), display room-temperature ionic conductivities (0.19 mS/cm, 0.024 mS/cm) and formation energies close to the hull ($<$30–50 meV/atom) [2507.02334].

This workflow, combining high-throughput NEB, efficient fine-tuning, and accurate barrier estimation, allows systematic mapping of migration pathways and conductivity metrics, establishing a paradigm for ML-accelerated exploration of novel battery electrolytes.

## 7. Capabilities, Limitations, and Perspectives

**Strengths**:
- Broad chemical and structural transferability due to extensive pretraining on $\sim$1.6 million DFT configurations.
- Charge-awareness via magnetic-moment proxy, enabling electronic structure effects and phase complexity to be encoded in the model.
- Rapid, automated integration into NEB/MD pipelines, with speedups enabling previously intractable disorder and kinetic pathway sampling.
- Efficient fine-tuning (<1k DFT structures) corrects softening and restores DFT accuracy for both equilibrium and transition-state configurations [2507.02334, 2509.08498].

**Limitations**:
- Underestimation of high-energy images (10–50 meV/atom) persists, motivating growth of transition-state training data.
- Pretrained and fine-tuned models are specialized to Li-containing materials; extension to, e.g., Na or Mg systems, requires new labeled data [2507.02334].
- Global charge neutrality and explicit long-range electrostatics are not enforced; charge is inferred via magnetic moments, which are not applicable to nonmagnetic species [2302.14231]. For explicit charge transfer, potentials such as QET, which directly solve for atomic charges, may provide superior accuracy for systems dominated by long-range Coulomb interactions [2511.07249].
- For phonon-dominated properties—second/third-order IFCs, lattice thermal conductivity—CHGNet currently exhibits systematically larger errors compared to leading uMLPs [2509.03401].

A practical perspective is that, for equilibrium and activated-process modeling in charge- and spin-active solids, CHGNet now functions as a robust, general-purpose foundation potential. Where domain-specialized accuracy is required in poorly represented chemistries or electronic regimes, fine-tuning or $\Delta$-learning using CHGNet’s internal atomic embeddings via Gaussian process regression remains computationally inexpensive and effective [2502.21179].

---

**Key References**:
- “High-Throughput NEB for Li-Ion Conductor Discovery via Fine-Tuned CHGNet Potential” [2507.02334]
- “CHGNet: Pretrained universal neural network potential for charge-informed atomistic modeling” [2302.14231]
- “Benchmarking CHGNet Universal Machine Learning Interatomic Potential Against DFT and EXAFS: Case of Layered WS2 and MoS2” [2509.08498]
- “A Comprehensive Assessment and Benchmark Study of Large Atomistic Foundation Models for Phonons” [2509.03401]
- “$Δ$-model correction of Foundation Model based on the models own understanding” [2502.21179]
- “A Fast, Accurate, and Reactive Equivariant Foundation Potential” (for comparison of charge-awareness paradigms) [2511.07249]

Source: https://www.emergentmind.com/topics/chgnet-foundation-potential