---
title: Neuroevolution Potential (NEP/UNEP-v1)
url: https://www.emergentmind.com/topics/neuroevolution-potential-nep-unep-v1
type: topic
---

# Neuroevolution Potential (NEP/UNEP-v1)

The Neuroevolution Potential (NEP/UNEP-v1) is a machine-learned interatomic potential framework that couples systematically crafted local descriptors with compact neural networks, trained via black-box evolutionary optimization. Designed for deployment in large-scale atomistic simulations, especially within the GPU-accelerated GPUMD environment, NEP/UNEP-v1 delivers near-DFT accuracy while maintaining empirical-potential-level computational efficiency. This unified model is applicable across diverse chemistries—including pure elements, alloys, oxides, semiconductors, and porous frameworks—with high fidelity to both structural and thermal properties, even for million-atom and nanosecond timescale molecular dynamics [2409.04430, 2311.04732, 2501.11191].

## 1. Mathematical Formulation and Descriptor Structure

NEP/UNEP-v1 represents the total potential energy $E$ of an $N$-atom system as a sum of atomic energies $E_i$:
\[
E = \sum_{i=1}^N E_i
\]
Each site energy $E_i$ is modeled via a feed-forward neural network of fixed width, whose input is a descriptor vector $\mathbf{q}_i$ encoding the local environment of atom $i$:
\[
E_i = \mathbf{w}^e \cdot \tanh( W^u \mathbf{q}_i - \mathbf{b}^u ) - b^e
\]
where $W^u, \mathbf{b}^u, \mathbf{w}^e, b^e$ are species-dependent trainable weights and biases; $\tanh$ is used as the activation function [2604.13848, 2311.04732]. For multi-element systems, the total model comprises separate neural submodels (optionally bagged in an ensemble) for each atomic species.

The descriptor vector $\mathbf{q}_i$ concatenates radial ($q^{\text{rad}}_n$) and angular ($q^{\text{ang}}_{nl}$, with higher body-orders possible) symmetry functions. The canonical forms employed are:
\[
q^{\text{rad}}_{i, n} = \sum_{j \neq i} f_c(r_{ij}) R_n(r_{ij})
\]
\[
q^{\text{ang}}_{i, n l} = \sum_{j \neq i} \sum_{k \neq i, j} f_c(r_{ij}) f_c(r_{ik}) R_n(r_{ij}) R_n(r_{ik}) P_l(\cos\theta_{jik})
\]
where $R_n(r)$ are Chebyshev or polynomial radial functions, $P_l$ are Legendre polynomials, and $f_c(r)$ is a smooth cosine cutoff. For alloy and multicomponent problems, Chebyshev coefficients $c^{IJ}_{nk}$ are trained per element-pair; higher descriptor orders (e.g., up to five-body) and separate cutoffs for radial and angular channels are used in NEP4/UNEP-v1 [2311.04732, 2604.01642].

All descriptors are invariant to global rotations, translations, and permutation of like atoms, and are normalized to unit variance over training data.

## 2. Training Methodology: Separable Natural Evolution Strategy (SNES)

NEP/UNEP-v1 employs a derivative-free optimization via a Separable Natural Evolution Strategy (SNES), which evolves a population of model parameter vectors. At each generation, candidate parameter sets are sampled from a multivariate Gaussian distribution, scored on a loss function, ranked, and used to update the Gaussian's mean and variance along the natural gradient:
- **Loss Function** (for parameter vector $\theta$): 
  \[
  L(\theta) = w_E \Vert E_{\mathrm{pred}} - E_{\mathrm{ref}} \Vert^2 + w_F \Vert F_{\mathrm{pred}} - F_{\mathrm{ref}} \Vert^2 + w_V \Vert V_{\mathrm{pred}} - V_{\mathrm{ref}} \Vert^2
  \]
  with typical weightings $w_E : w_F : w_V = 1:100:1$ for balancing global energy, atomic forces, and virial tensor errors [2409.04430, 2311.04732].

- **SNES Procedure**: For each generation, sample a population of candidate parameters, evaluate the loss over a (possibly batched) training subset, rank and assign fitness utilities, and update the mean and variance via natural gradient steps. Typically, 30–100 offspring per generation and $10^3$–$10^6$ generations are employed for converged models, with mini-batch loss to expedite evaluation [2604.13848, 2505.13179].

For some recent applications, analytic gradients and first-order optimizers (Adam) have been used for enhanced convergence (GNEP) [2507.00528].

## 3. Model Versions, Descriptor Advances, and Data Generation

The original NEP (e.g., NEP1) targeted single- and binary-atomic systems with type-averaged descriptors, while NEP2 introduced per-basis-function element-pair scaling to significantly enhance accuracy in multi-component alloys [2109.10643]. NEP4/UNEP-v1 incorporates expanded descriptor sets (e.g., up to five-body terms, more Chebyshev channels), per-element network weights, and bagging/ensembling for uncertainty quantification [2311.04732, 2604.01642].

Model training sets are built from extensive high-fidelity DFT databases. The unified 16-element metal potential (UNEP-v1) is trained entirely on unary and binary alloy data, with active learning and farthest-point sampling in descriptor space ensuring maximal chemical coverage. This technique enables generalization to arbitrary $n$-component alloys and solid solutions without explicit $n>2$ training data [2311.04732, 2505.13179, 2604.01642]. For nonmetal systems (MOFs, oxides, etc.), similar descriptor choices and active-learning sampling strategies are employed [2409.04430, 2505.18993, 2412.02191].

Tables of typical hyperparameters:

| Parameter          | Metals (UNEP-v1) | MOFs | Oxides |
|--------------------|------------------|------|--------|
| Radial cutoff $R_c$    | 6.0 Å           | 6.0 Å| 6.0 Å  |
| Angular cutoff    | 5.0 Å           | 4.5 Å| 5.0 Å  |
| Radial channels   | 5 (×Chebyshev 9) | 8–10 | 8–10   |
| Angular channels  | $\ell_\text{max}=4$ | 8   | 2–4    |
| Hidden layer size | 80               | 40–64| 40–100 |
| Training length   | $10^6$ gens      | $10^6$–$10^7$ | $2\times10^3$ (Adam) |
| Batch size        | $10^4$           | 500–1000 | 64   |

## 4. GPUMD Implementation and Computational Performance

NEP/UNEP-v1 is natively integrated in GPUMD, with all descriptor construction, neural network forward/backward passes, and force/virial/heat-current evaluation performed in CUDA kernels. Separate pre-allocated buffers store bead replicas, partial forces, and virials for PIMD workflows. GPUMD takes standard NEP model files, with element-specific settings included for all supported species [2409.04430, 2410.13535].

Performance is highly optimized for linear scaling and single- or multi-GPU deployment. Measured throughput routinely exceeds $10^7$–$10^8$ atom-step/s on modern A100, H100, or RTX 4090 hardware, enabling million-atom and multi-nanosecond molecular dynamics (MD) trajectories [2410.13535, 2505.13179, 2604.01642]. The total model size per species is of order kilobytes; neighbor list and coordinate data dominate memory consumption [2604.01642].

Relative to other state-of-the-art MLIPs (e.g., GAP, ACE, DP, GRACE), NEP/UNEP-v1 is typically one to two orders of magnitude faster at inference, with only a modest penalty over classical EAM-type potentials, but delivers DFT-quality accuracy for energies, forces, and virials across a wide variety of systems [2311.04732, 2410.13535, 2604.01642].

## 5. Benchmarks, Accuracy, and Applications

### Accuracy
- **Metals (UNEP-v1, 16 elements):** Training RMSEs of 2–5 meV/atom (energy), 60–100 meV/Å (force), 0.2–0.4 GPa (virial); test RMSEs remain <25 meV/atom and 120 meV/Å for alloys and complex solid solutions [2311.04732, 2505.13179, 2604.01642].
- **MOFs/Water/Al:** RMSEs of $<$1.2 meV/atom (energy), 30–53 meV/Å (force), and 0.8–14 meV/atom (virial) have been achieved [2409.04430].  
- **GaN, other semiconductors, oxides:** Model and hyperparameter optimization gives force RMSEs $<$50–100 meV/Å and energy RMSEs $<$1 meV/atom [2502.05580, 2601.10174, 2412.02191].

### Large-Scale Applications
NEP/UNEP-v1 supports a range of applications:
- **Path-integral MD with nuclear quantum effects** (LiH, MOFs, water, Al), capturing isotope effects and quantum corrections [2409.04430].
- **Heat transport simulations** for pure and polycrystalline graphene, metals, and MOFs with system sizes up to 1.4 million atoms; accurate thermal conductivity predictions and transferability benchmarks (including D3 dispersion for van der Waals systems) [2410.13535, 2310.05279, 2505.13179].
- **Hybrid MC/MD for chemical order and radiation damage** in concentrated alloys and complex materials [2404.13694, 2311.04732].
- **Structural and mechanical response** for oxides (Al$_2$O$_3$, Ga$_2$O$_3$), cement gels, tobermorite, and Ca–Si–H systems, robust to polymorphism and amorphization [2505.18993, 2412.02191, 2601.10174].
- **Autonomous active learning** for dataset expansion in large or extrapolative cells, with D-optimality-driven selection and fragment embedding for scalable training data acquisition [2604.13848].

### Transferability and Limitations
NEP/UNEP-v1 is demonstrated to transfer accurately from unary/binary training sets to arbitrary $n$-component alloys and mixed-chemistry phases [2311.04732, 2501.11191]. However, explicit long-range electrostatics, charge transfer, and out-of-domain chemistry require augmenting the framework with additional descriptors or coupling to charge-equilibration models.

## 6. Active Learning, Uncertainty Quantification, and Model Updates

Active learning in NEP/UNEP-v1 is based on D-optimality and ensemble uncertainty. During MD, atomic environments with extrapolation grade above threshold, detected via the D-optimality MaxVol criterion, are extracted, embedded in locally periodic fragments, and labeled with DFT for incremental retraining [2604.13848]. For uncertainty quantification, bagging ensembles of NEP submodels are used; the variance across the ensemble informs model error and reliability [2604.01642].

Automated retraining, fragment labeling, and model hot-swapping are directly supported in GPUMD, enabling seamless on-the-fly updates and adaptation to new environments with minimal computational overhead.

## 7. Impact and Outlook

The NEP/UNEP-v1 framework enables routine, high-fidelity atomistic simulations at scales and speeds previously inaccessible to DFT-based approaches and outperforms standard empirical models in both accuracy and transferability for alloys, oxides, and complex frameworks. Its design principles—local systematic descriptors, compact ANNs, and global evolutionary optimization—allow for rapid extension to new chemistries, automated active-learning deployment, and direct incorporation in multi-resolution methods.

Planned developments include further extension toward full periodic-table “foundation” NEPs, hybridization with charge and long-range descriptors, deeper architectures for strongly correlated materials, and workflow automation for self-guided data generation and model management [2311.04732, 2603.01234, 2603.17367]. This suggests NEP/UNEP-v1 remains a competitive standard in the current landscape of machine-learned potentials for large-scale molecular simulation.

Source: https://www.emergentmind.com/topics/neuroevolution-potential-nep-unep-v1