The paper introduces SOG-TNN, a variational ansatz that integrates tensor-product neural bases with sum-of-Gaussians kernel approximations to efficiently solve many-electron Schrödinger equations.
Methodologically, it employs one-dimensional feedforward subnetworks and model reduction techniques like weighted balanced truncation and SVD to reduce high-dimensional integrals.
Empirical results demonstrate spectral-like convergence and significant computational savings compared to sparse-grid and spherical-harmonic methods for systems such as H, He, Li, and Be.
Searching arXiv for the specified SOG-TNN papers and closely related antecedents.
The Sum-of-Gaussians Tensor Neural Network (SOG-TNN) is a variational neural-network ansatz for Schrödinger equations that combines a low-rank tensor neural network representation of the wave function with a sum-of-Gaussians approximation of Coulomb-type interaction kernels. In the form presented for the many-electron problem, SOG-TNN couples learned one-dimensional feedforward subnetworks, tensorized basis construction, kernel separabilization, and exact fermionic antisymmetry via a Slater determinant ansatz (Wu et al., 25 Mar 2026). A broader formulation for the high-dimensional Schrödinger equation uses the same central idea—tensor-product neural bases together with separable Gaussian decompositions of Coulomb interactions—to reduce high-dimensional integration to combinations of low-dimensional integrals, supplemented by range splitting and offline model reduction (Zhou et al., 14 Aug 2025).
1. Definition and problem setting
SOG-TNN is introduced for the many-electron Schrödinger equation in settings where the main computational difficulties are high dimensionality, strong correlation, and the nonseparable character of Coulomb interactions. In the one-dimensional soft-Coulomb formulation, the target equation is
The defining feature of SOG-TNN is the joint use of tensor-product neural bases and Gaussian kernel decompositions. This allows the wave function to be represented in low rank while transforming Coulomb terms into separable expressions. In the one-dimensional many-electron setting, the method is explicitly described as combining a tensor neural network representation, a sum-of-Gaussians (SOG) approximation of the Coulomb kernel, and a Slater determinant ansatz to enforce antisymmetry exactly (Wu et al., 25 Mar 2026).
2. Tensor neural representation of the wave function
In the many-electron soft-Coulomb construction, the wave function is approximated as
Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),
where P is the basis size, αp are linear coefficients, H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),0 are trainable parameters, H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),1 is the H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),2-th tensorized basis, and H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),3 is the antisymmetrization operator implemented by a Slater determinant (Wu et al., 25 Mar 2026).
Each one-electron orbital H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),4 is generated by a one-dimensional feedforward neural network,
Accordingly, the basis functions are not fixed orbitals or polynomials; they are learned nonlinear one-dimensional neural-network features combined tensorially (Wu et al., 25 Mar 2026).
The broader high-dimensional SOG-TNN formulation adopts an analogous low-rank tensor-product expansion. For H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),7,
so high-dimensional integrals are replaced by products of one-dimensional integrals (Zhou et al., 14 Aug 2025).
The high-dimensional paper also states a theoretical approximation estimate:
v(u)=1+u21.1
with a dimension-dependent constant v(u)=1+u21.2 that decays like v(u)=1+u21.3. This is presented as a formal justification that low-rank tensor structure can mitigate the curse of dimensionality (Zhou et al., 14 Aug 2025).
3. Sum-of-Gaussians decomposition and model reduction
The kernel treatment is the central computational mechanism in SOG-TNN. In the one-dimensional soft-Coulomb setting, the interaction kernel is approximated by a bilateral Gaussian series,
v(u)=1+u21.4
with
v(u)=1+u21.5
for tunable parameters v(u)=1+u21.6 and v(u)=1+u21.7. The approximation error is stated to behave like
A first reduction stage is weighted balanced truncation (WBT), which compresses the large Gaussian sum:
v(u)=1+u21.9
with VA0. A second reduction stage applies a separable Chebyshev expansion to each Gaussian kernel factor,
VA1
followed by an SVD compression of the coefficient matrix VA2, so that the retained rank satisfies VA3 (Wu et al., 25 Mar 2026).
For VA4, VA5, and VA6, the reported reduction is from about 220 Gaussians in the raw bilinear-series approximation to only 26–28 Gaussians after WBT, and SVD further reduces the Chebyshev rank by roughly two-thirds. The specific values reported are as follows (Wu et al., 25 Mar 2026).
VA7
VA8
VA9
13
26
229
15
27
270
20
28
361
The high-dimensional SOG-TNN paper uses the classical Gaussian integral identity
TijΨ=−Ψ,1≤i=j≤N0
to derive a bilateral sum-of-Gaussians approximation of the Coulomb kernel,
That paper supplements the decomposition by a range-splitting scheme dividing Gaussian indices into short-, mid-, and long-range sets. Short-range terms use an asymptotic expansion, long-range terms use low-rank Chebyshev expansion, and mid-range terms use SVD-based model reduction. This is described as the principal mechanism for reducing the number of expensive two-dimensional integrals in electron-electron interactions (Zhou et al., 14 Aug 2025).
4. Antisymmetry and the Slater determinant ansatz
For fermionic systems, SOG-TNN incorporates antisymmetry structurally rather than through penalties in the one-dimensional soft-Coulomb formulation. The antisymmetrization operator for spin-TijΨ=−Ψ,1≤i=j≤N4 electrons is
TijΨ=−Ψ,1≤i=j≤N5
This is explicitly the determinant structure of a Slater determinant (Wu et al., 25 Mar 2026).
Because exchanging any two same-spin electrons changes the determinant sign, the wave function automatically satisfies
TijΨ=−Ψ,1≤i=j≤N6
The spin sectors are handled separately according to
TijΨ=−Ψ,1≤i=j≤N7
and the full wave function is formed from the product of spin-resolved antisymmetrized factors (Wu et al., 25 Mar 2026).
This strict enforcement of antisymmetry is a defining distinction of the many-electron soft-Coulomb SOG-TNN formulation. By contrast, the earlier high-dimensional SOG-TNN paper states that antisymmetry constraints are enforced for same-spin electrons via exchange operators TijΨ=−Ψ,1≤i=j≤N8, and its algorithmic outline includes a loss with Pauli penalties (Zhou et al., 14 Aug 2025). This suggests that the later formulation with a Slater determinant ansatz shifts antisymmetry from a training constraint to an exact architectural property.
5. Variational workflow, complexity, and deterministic energy evaluation
In the one-dimensional soft-Coulomb application, the workflow is stated as follows: truncate the spatial domain to TijΨ=−Ψ,1≤i=j≤N9, represent the wave function by SOG-TNN with antisymmetric Slater determinants, use SOG + WBT + SVD to convert electron-electron interactions into efficiently computable one-dimensional integrals, evaluate the energy functional deterministically,
and optimize with a hybrid strategy consisting of RAdam with cosine annealing and warm restarts, then another cosine annealing phase, then L-BFGS fine-tuning (Wu et al., 25 Mar 2026).
After factorization, the Hamiltonian assembly cost is given as
where H^=−21i=1∑NΔi+1≤i<j≤N∑∣ri−rj∣1−i=1∑Nk=1∑M∣ri−Rk∣Qk+1≤k<l≤M∑∣Rk−Rl∣QkQl,2 is the cost of the one-dimensional numerical quadrature. The substantive point is that the original high-dimensional electron-electron interaction is converted into a manageable set of one-dimensional integrals (Wu et al., 25 Mar 2026).
The high-dimensional SOG-TNN paper provides a more detailed algorithmic decomposition. Its per-step costs include
mass matrix: H^=−21i=1∑NΔi+1≤i<j≤N∑∣ri−rj∣1−i=1∑Nk=1∑M∣ri−Rk∣Qk+1≤k<l≤M∑∣Rk−Rl∣QkQl,3,
These expressions formalize the claim that the method is low-memory relative to full high-dimensional discretizations or dense multidimensional interaction tables (Zhou et al., 14 Aug 2025).
6. Convergence behavior and numerical performance
The later many-electron paper reports spectral-like convergence of the relative ground-state energy error
E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩0
with respect to basis size E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩1, using the mixed algebraic-exponential model
E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩2
For atoms H through O, the reported fits satisfy E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩3, with values including E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩4 for H, E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩5 for He, E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩6 for Li, E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩7 for Be, E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩8 for B, E0=Ψ∈VPminE[Ψ],E[Ψ]=⟨Ψ∣Ψ⟩⟨Ψ∣H^∣Ψ⟩9 for C, Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),0 for N, and Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),1 for O (Wu et al., 25 Mar 2026).
The same paper states that chemical accuracy is reached for all tested atoms with only Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),2, and that with Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),3 the errors are below Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),4 for H, He, Li, Be, B and below Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),5 for C, N, O. Reported high-accuracy energies were obtained with compact basis sizes up to oxygen, including H with Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),6, He with Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),7, Li with Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),8, Be with Ψ(r;Θ)=p=1∑PαpA(Φp↑(r↑;Θ↑))⋅A(Φp↓(r↓;Θ↓)),9, B with P0, C with P1, N with P2, and O with P3 (Wu et al., 25 Mar 2026).
The comparison with sparse-grid CI (SG-CI) is reported in similarly concrete terms. For Li, SOG-TNN reaches P4 accuracy with P5, while SG-CI needs about P6 bases. For C, SOG-TNN gets P7 accuracy with P8, whereas SG-CI remains around P9 even with tens of thousands of bases (Wu et al., 25 Mar 2026).
The earlier high-dimensional SOG-TNN paper reports numerical experiments on H, He, Li, and Be. It states that the method uses two hidden layers in each 1D subnet and sine activation, with Adam and a piecewise learning-rate decay introduced when needed to avoid training rebounds. For helium, it reports computed ground-state energy
αp0
against the reference
αp1
with relative accuracy about αp2. For lithium, it reports
αp3
against the reference
αp4
again at about αp5 accuracy. For beryllium, it reports
That paper also compares SOG-TNN against spherical-harmonic-expansion TNN (SHE-TNN). Representative values are reported as follows (Zhou et al., 14 Aug 2025).
The same comparison reports memory and time-per-step reductions: for He, H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),04 MiB versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),05 MiB and H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),06 ms versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),07 ms; for Li, H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),08 MiB versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),09 MiB and H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),10 ms versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),11 ms; for Be, H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),12 MiB versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),13 MiB and H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),14 ms versus H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),15 ms (Zhou et al., 14 Aug 2025).
7. Related formulations and interpretive context
SOG-TNN belongs to a broader family of tensorized neural constructions in which sums over local components are organized by low-rank tensor structure. A relevant antecedent is Tensorial Mixture Models (TMMs), which are not named SOG-TNN but are described as generative models with mixtures of local components and tensor-factorized global mixing weights, implemented by a Convolutional Arithmetic Circuit (ConvAC) (Sharir et al., 2016).
In TMMs, the input is decomposed into local parts H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),16, each governed by a hidden assignment H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),17, and the joint density is
The prior H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),19 is interpreted as an order-H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),20 tensor H:=i=1∑N(−21∇ri2−J=1∑MzJv(∣ri−RJ∣)+j=i+1∑Nv(rij))+I<J∑zIzJv(RIJ),21, and the paper connects its parameterization to classical tensor decompositions: CP decomposition corresponds to shallow ConvAC / shallow TMM, while Hierarchical Tucker corresponds to deep ConvAC / deep TMM (Sharir et al., 2016).
The local component family in TMMs is frequently Gaussian, and in sparse special cases the model reduces to a standard diagonal Gaussian mixture model,
This supports the characterization of TMMs as a strong antecedent for later architectures combining Gaussian components, tensor factorization, and neural implementation (Sharir et al., 2016).
The relationship is not identity. In TMMs, the Gaussian components are local factors in a generative density model, whereas in SOG-TNN the Gaussian structure is used primarily as a separable approximation of interaction kernels within a variational solver for Schrödinger equations. A plausible implication is that both lines of work share a tensorized “sum-over-components” logic, but they deploy it toward different computational ends: tractable marginalization and generative inference in TMMs, versus deterministic energy evaluation and low-rank many-body approximation in SOG-TNN (Sharir et al., 2016).
Taken together, the SOG-TNN literature presents a progression from a general high-dimensional tensor-neural solver with SOG kernel decomposition and range splitting (Zhou et al., 14 Aug 2025) to an improved many-electron architecture with model reduction techniques, exact Slater-determinant antisymmetry, and observed mixed algebraic-exponential spectral convergence for one-dimensional soft-Coulomb systems (Wu et al., 25 Mar 2026).