---
title: Atomic Physics Neural Networks
url: https://www.emergentmind.com/topics/atomic-physics-neural-networks-apnn
type: topic
---

# Atomic Physics Neural Networks

Atomic Physics Neural Networks (APNN) denotes a family of neural-network methods in which atomic-scale physics is embedded directly into the representation, architecture, training objective, or workflow. In the literature, the term spans several distinct but related uses: covariant networks for atomic potential energy surfaces, atomic neural networks for molecules and materials, neural quantum states for electronic and nuclear many-body problems, neural classifiers embedded in atomic configuration-interaction pipelines, physics-informed neural networks for atomic and optical quantum control, and physics-aware networks for atomic-scale structure determination from microscopy. Across these settings, the defining motif is not a single architecture but the explicit incorporation of physical structure such as local environments, symmetry, conservation laws, projection models, or Hamiltonian dynamics [1803.01588], [1910.03376], [2408.00477], [2603.19942].

## 1. Scope and defining characteristics

Within atomistic modeling, APNN usually refers to models that operate on atoms as fundamental units and construct local, symmetry-respecting representations of each atom’s environment before aggregating atomic contributions into extensive observables such as total energy, forces, virial stress, or other physico-chemical quantities. A standard formulation is the atomic-energy ansatz
\[
E(\mathbf{R}) = \sum_i E_i\big(G_i(\mathbf{R})\big), \qquad \mathbf{F}_i=-\nabla_{\mathbf{R}_i}E(\mathbf{R}),
\]
with locality enforced by finite cutoffs and scalar invariance obtained from internal coordinates or symmetry-adapted features [1910.03376]. In a more representation-theoretic formulation, each subsystem carries a covariant state $\psi_j$ in a representation space of $SO(3)$, and parent states are built by aggregation rules of the form
\[
\psi_j=\Phi_j(\hat r_{ch_1},\ldots,\hat r_{ch_k},\psi_{ch_1},\ldots,\psi_{ch_k}),
\]
with irreducible $SO(3)$ structure preserved by Clebsch–Gordan reduction and same-$\ell$ blockwise mixing [1803.01588].

The label also extends beyond interatomic potentials. In atomic CI, APNN describes neural selectors that classify relativistic configurations by importance inside iterative pCI workflows, thereby reducing otherwise intractable configuration spaces [2408.00477], [2503.01379]. In atomic electron tomography, APNN describes physics-aware networks that encode projection geometry, HAADF contrast, local symmetry, and coordination-shell structure to recover atomic coordinates and species under low-dose conditions [2603.19942]. In quantum control, the term is used for PINN-based control of open quantum systems, where the GKSL or Schrödinger dynamics are embedded into the loss so that learned controls remain tied to atomic or optical Hamiltonians [2206.06287].

| Family | Core task | Representative papers |
|---|---|---|
| Covariant atomistic networks | Atomic potentials and many-body energy mapping | [1803.01588], [1903.04366] |
| Atomic neural-network libraries | PES, forces, stress, dipoles, charges | [1910.03376], [1911.01365] |
| Neural wave functions | Excited states, ionization, nuclear ground states | [2312.15216], [2108.06836] |
| CI-integrated APNN | Selection of important relativistic configurations | [2408.00477], [2503.01379] |
| Physics-informed control and imaging | Quantum control and atomic tomography | [2206.06287], [2603.19942] |
| Nuclear-observable regression | Charge radii and atomic masses | [1212.6319], [2501.01352] |

A common misconception is that APNN denotes a single standardized model class. The cited literature instead uses the term as an umbrella over several architectures whose shared feature is the direct encoding of atomic-scale physics into learning.

## 2. Symmetry-structured interatomic potentials and atomic energy mapping

A central APNN lineage concerns the learning of interatomic potentials and potential energy surfaces. N-body Networks formulate this problem as a compositional directed acyclic graph in which each node corresponds to a physical subsystem, carries a position $r_j\in\mathbb{R}^3$ and a covariant state $\psi_j$, and aggregates child subsystems through local tensorial rules. Rotational covariance is exact: activations transform under block-diagonal direct sums of Wigner $D$ matrices, nonlinearities are implemented by tensor products followed by Clebsch–Gordan decomposition, and learnable weights are permitted only within fixed-$\ell$ irreducible blocks. The resulting architecture is hierarchical, Fourier-space, and explicitly $N$-body in structure rather than descriptor-based in the usual sense [1803.01588].

PiNN and its PiNet architecture represent a more application-oriented APNN framework for molecules and materials. PiNet defines pairwise interaction features
\[
I_{ij}^t=\mathrm{PI}\!\left(\vec P_i^t,\vec P_j^t,r_{ij}\right),
\qquad
\vec P_i^{t+1}=\mathrm{IP}\!\left(\sum_j \vec I_{ij}^t\right),
\]
with radial basis expansions filtered by a smooth Behler cutoff,
\[
f_c(r)=
\begin{cases}
0.5[\cos(\pi r/R_c)+1], & r<R_c\\
0, & r\ge R_c ,
\end{cases}
\]
and zero biases in the II-layer to guarantee smooth PES behavior at the cutoff. PiNN implements both PiNet and Behler–Parrinello HDNNs, provides analytical virial pressure under periodic boundary conditions, and interfaces with ASE and a development version of AMS. Reported results include MAE \(=0.012\) eV on QM9, MAE \(=0.029\) eV/atom on Materials Project crystals, and MAE \(=0.042\) eV/atom on a perovskite benchmark [1910.03376].

The APNN literature also distinguishes between accurate total-energy fitting and correct atomic-energy inference. In the atomic-energy mapping analysis for Si, the DFT total energy is written as
\[
E_\text{tot}^\text{DFT}=\sum_i E_\text{at}(\mathcal{R}_i;R_c),
\]
and the neural approximation as
\[
E_\text{tot}^\text{DFT}\simeq \sum_i E_\text{at}^\text{NN}(\mathbf{G}_i;r_c).
\]
The work emphasizes invariant points in feature space, where all atoms share the same local descriptor and the atomic energy is fixed by \(E_\text{tot}^\text{DFT}/N\). It shows that networks can achieve energy RMSE \(0.9\) meV/atom and force RMSE \(0.10/0.11\) eV/Å on crystalline Si while still learning an incorrect EOS away from equilibrium, and that in a Si(100)-(2×2) slab at 100 K the bulk and surface mapping errors can be \(-108\) meV/atom and \(+76\) meV/atom despite a total-energy RMSE of \(0.3\) meV/atom [1903.04366]. This established a technical caution: APNN potentials may exhibit ad hoc mapping unless training sets span volumes, structures, and feature-space connectivity.

A deliberately simple dense-network variant was demonstrated for Al by fitting per-atom energies from EAM and MEAM using the spherical-coordinate input
\[
X=[R_1,\ldots,R_{54},\Theta_1,\ldots,\Theta_{54},\Phi_1,\ldots,\Phi_{54}],
\]
with a nine-layer fully connected network. More than \(77\%\) of EAM samples and more than \(81\%\) of MEAM samples had relative error below \(0.2\%\); over \(93\%\) of EAM samples and over \(95\%\) of MEAM samples had absolute error below \(10\) meV [1911.01365]. This suggests that even non-invariant representations can approximate classical reference potentials accurately on restricted tasks, although the broader APNN literature strongly favors explicit invariance and smoothness.

## 3. Neural wave functions and many-body structure in atomic and nuclear physics

A second APNN lineage uses neural networks as variational many-body ansätze. In the excited-state framework based on FermiNet with effective core potentials, the many-electron Hamiltonian is written as
\[
\hat H = -\tfrac{1}{2}\sum_i \nabla_i^2 + \sum_{i<j}\frac{1}{r_{ij}} + \sum_A \hat V_A^{\text{ECP}} + V_{nn},
\]
with semi-local ECP operators
\[
\hat V_A^{\text{ECP}}(\mathbf r)=V_{\text{loc},A}(r)+\sum_{\ell=0}^{\ell_{\max}}V_{\ell,A}(r)\hat P_\ell .
\]
The neural wave function has the FermiNet-like form
\[
\Psi_\theta(\mathbf R)=\sum_{k=1}^K
\det\big(\Phi^\uparrow_k(\mathbf R)\big)\det\big(\Phi^\downarrow_k(\mathbf R)\big)\exp\big(J_\theta(\mathbf R)\big),
\]
and excited states are targeted sequentially with an orthogonality-penalized loss
\[
\mathcal{L}_n(\theta)=E_n(\theta)+\alpha\sum_{i=1}^{n-1} S_{ni}(\theta).
\]
The reported vertical excitation energies for atoms from Li to Br and molecules including LiH, BeH, CO, H\(_2\)O, H\(_2\)S, H\(_2\)CSi, and benzene have mean absolute errors typically in the \(1\)–\(5\) mHa range, with almost all level errors within \(\sim 10\) mHa versus theoretical best estimates or experiment [2312.15216].

Neural quantum states also appear in nuclear APNN. For nuclei up to \(A=6\), the variational ansatz is
\[
\langle RS|\Psi_V^{ANN}\rangle=\exp[U(R,S)]\tanh[V(R,S)]\langle RS|\Phi\rangle,
\]
where \(\Phi\) is an antisymmetrized mean-field state with good \(J^\pi T\), while \(U\) and \(V\) are permutation-invariant neural correlators of Deep Sets form
\[
F(R,S)=\rho_F\!\left(\sum_{i\neq j}\phi_F(r_i,s_i,r_j,s_j)\right).
\]
Benchmarks against hyperspherical harmonics show close agreement for \(^2\)H, \(^3\)H, \(^3\)He, \(^4\)He, \(^6\)He, and \(^6\)Li. For example, for \(^6\)Li with the NN-only Hamiltonian the ANN energy is \(-42.04(1)\) MeV versus \(-42.51(5)\) MeV from HH, while with NN+3N it is \(-30.82(3)\) MeV versus \(-31.00(8)\) MeV [2108.06836]. The underbinding relative to HH was attributed to nodal limitations of the mean-field part.

At the level of learned nuclear observables, a small ANN trained on around \(900\) nuclei yielded the empirical charge-radius formula
\[
R_c(A)=1.234\,A^{0.28},
\]
with root-mean-square deviation \(\sigma \approx 1.4\times 10^{-2}\) fm from experiment [1212.6319]. A physics-informed fully connected neural network for atomic masses adopted a macroscopic–microscopic decomposition with outputs \((E_\text{exp},E_{LD},E_\text{exp}-E_{LD})\) and reached a test RMSD of \(0.122\) MeV on AME2020 and \(0.191\) MeV when extrapolating from AME2016 to newly added AME2020 nuclei [2501.01352]. Together, these works indicate that APNN in the many-body context ranges from direct wave-function representation to surrogate learning of derived observables.

## 4. APNN inside atomic configuration interaction and high-precision codes

A distinct atomic-physics usage of APNN is as a classifier embedded in the configuration-interaction workflow. In this setting the CI expansion
\[
|\Psi^k\rangle=\sum_\alpha c_\alpha^k|\Phi_\alpha\rangle
\]
is not replaced by a network; rather, a network learns which relativistic configurations should be retained before the CI Hamiltonian is built and diagonalized. Configuration importance is defined by
\[
w_\Gamma^k=\sum_{\alpha\in\Gamma}|c_\alpha^k|^2,
\qquad
w_\Gamma=\max_k w_\Gamma^k,
\]
and the network performs binary classification according to whether \(w_\Gamma\ge x\) for a threshold \(x\) that is lowered iteratively [2408.00477].

The APNN-pCI workflow of Bilous, Cheung, and Safronova uses binary encodings of relativistic subshell occupancies as features and a dense MLP with four ReLU hidden layers and a softmax output. Labels come from previous pCI diagonalizations, while a balancing set of randomly added configurations prevents the classifier from collapsing onto already discovered subspaces. For Ni\(^{12+}\), the full pool of \(862{,}788\) relativistic configurations was reduced to a final APNN CI set of \(145{,}490\) configurations, a \(\approx 5.9\times\) reduction; the total pCI wall time fell from \(283.5\) h to \(71.2\) h, with final deviations from large direct-CI references of \(3.8, 2.7, 3.6, 3.6,\) and \(3.8\) cm\(^{-1}\) across the five lowest levels [2408.00477]. For Fe\(^{16+}\), reported RMS deviations are \(\approx 1.36\) cm\(^{-1}\) for five even-parity levels and \(\approx 1.47\) cm\(^{-1}\) for seventeen odd-parity levels.

The later Python package formalizes this methodology as a “Neural manager” on top of pCI and other high-performance atomic codes. It treats the APNN component as code-agnostic and delegates format-specific operations to an adapter layer. In the Fe\(^{16+}\) demonstration, the full even-parity CI basis contained \(151{,}422\) relativistic configurations, the prior run used \(28{,}348\), the APNN loop started from \(34{,}501\), and five iterations produced \(37{,}284\), \(35{,}859\), \(49{,}617\), \(60{,}629\), and \(60{,}070\) configurations. Timings were \(137\) min for full CI, \(3\) min for the prior run, \(5\) min for the start, and \(12\), \(13\), \(24\), \(32\), and \(36\) min for the five APNN iterations, giving \(122\) min total for the APNN route in this medium-size example [2503.01379]. The authors note that the benefit grows when full CI is impossible rather than merely expensive.

This CI-centered usage of APNN differs conceptually from neural potentials or neural wave functions. The network does not approximate the Hamiltonian or the eigenstate directly; it approximates the combinatorial selection rule that would otherwise require prohibitive perturbative screening or brute-force diagonalization.

## 5. Physics-informed control and atomic-scale measurement

APNN also includes models that encode the governing dynamics or measurement operator directly into the loss. In open-system quantum control, Norambuena et al. formulate the dynamics through the GKSL master equation
\[
\frac{d\rho}{dt}=-i[H(t;\theta,u(t)),\rho]+\sum_k\gamma_k\left(L_k\rho L_k^\dagger-\tfrac12\{L_k^\dagger L_k,\rho\}\right),
\]
or through the Schrödinger equation for closed systems. Their PINN outputs state and control trajectories using a hard initial-condition constraint
\[
\mathbf{x}(t)=\mathbf{x}(0)+f(t)\mathbf{N}_x(t),\qquad
\mathbf{u}(t)=\mathbf{u}(0)+f(t)\mathbf{N}_u(t),\qquad f(t)=1-e^{-t},
\]
and minimizes a loss containing physics residuals, fidelity terms, control penalties, and constraint penalties [2206.06287]. In a two-level open system with damping, the PINN found a time-dependent Stark-like control \(\xi(t)\) achieving Uhlmann fidelity \(F\approx 0.99\) for \(t\gtrsim 20\), with each density-matrix component within \(\sim 1\%\) of the target steady state. In a three-level \(\Lambda\)-system, the baseline result was \(p_2=0.97\), \(t_f=2.0\), and \(\mathcal A=7.3\), compared with STIRAP at \(p_2=0.98\), \(\mathcal A\approx 128.6\), \(t_f\approx 35\) [2206.06287]. The work explicitly treats these PINNs as APNNs when applied to TLS, \(\Lambda\)-systems, and NV-center-like platforms.

On the measurement side, the physics-aware, two-stage neural networks for atomic electron tomography define a forward model
\[
y_i=P_i x+n_i,
\qquad
I_i(\theta)=\Pi_\theta O(u,v,w)=\int O(u,v,w)\,dw,
\]
with low-dose Poisson noise, additive Gaussian noise, correlated backgrounds, and missing-wedge distortions. Stage 1, GLARE, is a global–local 3D ResUNet with FiLM modulation and an \(L_1\) regression loss against physically constructed positive target volumes. Stage 2, DAST, computes \(225\) 3D Zernike coefficients per atom, concatenates them with normalized coordinates, builds a \(k\)-NN graph with \(k=12\), and applies neighbor-restricted graph attention for elemental classification [2603.19942].

The quantitative results are among the clearest demonstrations of APNN in atomic-scale imaging. On a benchmark of \(42{,}588\) reconstructed volumes, the typical-noise test gave median F1 \(0.9978\) and median RMSD \(0.1002\) Å for GLARE, compared with \(0.9666\) and \(0.2444\) Å for direct tracing. Overall under typical conditions, the full PANN pipeline reduced atomic coordinate error from \(\approx 0.24\) Å to \(\approx 0.10\) Å and raised the combined atomic recovery rate from \(\approx 93\%\) to \(\approx 99\%\). Full DAST achieved \(99.50\%\) mean classification accuracy, whereas coordinate-only DAST plateaued at \(\sim 55\%\) validation/test accuracy. On low-dose experimental data at \(1/6\) dose, two independent PANN reconstructions agreed at \(95\%\), versus \(75\%\) for conventional processing; for CsPbBr\(_3\) at \(5{,}000\) e·Å\(^{-2}\), fine-tuned GLARE-CPB achieved F1 \(=0.9715\) and RMSD \(=0.1269\) Å [2603.19942].

These examples show two complementary APNN modes: one constrains control synthesis by dynamical equations, and the other constrains reconstruction by the imaging operator, local symmetry descriptors, and coordination-aware graph structure.

## 6. Methodological principles, limitations, and terminological ambiguity

Across the literature, several design principles recur. One is explicit symmetry handling: translational and rotational invariance, permutation invariance, or full rotational covariance are built into descriptors, message passing, or irreducible representations rather than left to data augmentation alone [1803.01588], [1910.03376]. A second is physically structured decomposition: atomic energies are summed over local environments, nuclear masses are split into \(E_{LD}\) and residual terms, CI selection is performed at the level of relativistic configurations rather than individual determinants, and tomography separates global distortion correction from local symmetry-based species classification [2501.01352], [2503.01379], [2603.19942]. A third is structure-preserving regularization: smooth cutoff functions, hard initial-condition constraints, trace or positivity penalties, and local-neighborhood attention are used to keep learned objects physically admissible [1910.03376], [2206.06287].

The main limitations are equally consistent. In interatomic potentials, accurate total-energy fitting can conceal incorrect atomic-energy mapping, so invariant-point diagnostics and training-set connectivity matter [1903.04366]. In quantum control of open systems, exact GKSL satisfaction guarantees complete positivity and trace preservation only in the continuous limit; in practice, numerical residuals may drift unless Cholesky-like or exponential-map parameterizations are added [2206.06287]. In atomic electron tomography, the workflow assumes no atomic dynamics during tilt acquisition and remains modular rather than end-to-end from raw tilt series to atomic model [2603.19942]. In nuclear mass and radius prediction, extrapolation remains limited by sparse data near drip lines and by the lack of calibrated uncertainty quantification [1212.6319], [2501.01352]. In CI selection, performance depends on the quality of the prior CI seed and on the schedule of selection thresholds [2408.00477], [2503.01379].

A further source of confusion is nomenclature. The acronym APNN is not unique across arXiv usage. In radiative-transfer literature it also denotes “Asymptotic-Preserving Neural Networks,” where the emphasis is multiscale kinetic-to-diffusion limits rather than atomic-scale structure or atomic physics [2501.08166]. In the atomic-physics sense, however, the cited works converge on a coherent idea: neural networks become APNNs when they do not merely fit atomic data, but instead encode the governing symmetries, operators, or hierarchical structures of atomic-scale physics itself.

A plausible implication is that APNN should be understood less as a narrow subfield than as a design doctrine. Whether the task is a potential-energy surface, an excited-state wave function, a CI basis, a control field, or a tomography reconstruction, the defining move is the same: replace generic function approximation by a neural architecture whose hypothesis space has already been shaped by the physics of atoms.

Source: https://www.emergentmind.com/topics/atomic-physics-neural-networks-apnn