Conservative Density-Functional Network
Implementation & benchmark of arXiv:2608.08137 — Cubic-Equivariant Neural Density Functional Theory for Three-Dimensional Lattice Fluids
Source paper: Cubic-Equivariant Neural Density Functional Theory for Three-Dimensional Lattice Fluids arXiv:2608.08137 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Predict a scalar excess free-energy functional of a complete density field and obtain the direct-correlation output by automatic differentiation, instead of independently predicting each output-site value. This enforces the integrability and reciprocity constraints of a thermodynamic force field and gives a Lyapunov-like scalar that can control iterative density inference.
Formulas
Mathematical statement
Let rho=(rho_r) be a lattice density field, beta=(k_B T)^(-1), and F_exc[rho] the excess Helmholtz free-energy functional. The direct correlation is the functional derivative c^(1)[rho](r)=-delta(beta F_exc[rho])/delta rho_r. Parameterize a neural scalar functional Phi_hat_theta[rho] and define c_hat_theta^(1)[rho](r)=-partial Phi_hat_theta/partial rho_r. Its Hessian is symmetric, partial c_hat_r/partial rho_s=partial c_hat_s/partial rho_r, because it is the negative Hessian of one scalar. For iterative equilibrium inference with ideal contribution log rho_r, external potential V_r, and chemical potential mu, use rho_r^+=(1-alpha)rho_r+alpha sigma(mu-V_r+c_hat_theta^(1)[rho](r)), where sigma is a bounded occupancy map and 0<alpha<=1 is a damping parameter. A local stability diagnostic is alpha times lambda_max(J)<1, where J is the Jacobian of the undamped map at the fixed point.
Implementation notes
Use a fully convolutional scalar-output network Phi_theta that maps a complete periodic or padded density profile rho to one scalar. To preserve extensivity, sum local contributions: Phi_theta[rho]=sum_r q_theta(rho)_r, with q produced by a cubic-equivariant CNN; then call reverse-mode automatic differentiation to obtain c_hat=-grad(Phi,rho). Train with Monte Carlo targets c_target using an L2 loss, optionally add a free-energy anchor loss on known homogeneous states and a Hessian reciprocity penalty only if numerical differentiation reveals implementation errors. Pseudocode is: rho.requires_grad_(True); Phi=network(rho).sum(); c_hat=-autograd.grad(Phi,rho,create_graph=True); loss=mean((c_hat-c_target)^2); update theta. At inference, solve the Euler-Lagrange fixed point with damped updates above and estimate the largest Jacobian eigenvalue by 10 to 20 power iterations using Jacobian-vector products; set alpha=min(alpha_max,0.8/lambda_max). The paper supplies the thermodynamic relation between c^(1) and the free-energy functional; network values, target noise, and lambda_max are estimated. First experiment: train on 3D lattice hard-cube Monte Carlo profiles in randomized external potentials, comparing direct c prediction with the conservative model at equal parameter count. Test homogeneous equations of state, planar-wall profiles, and random initial densities. The prediction is that the reciprocity residual ||J-J^T||_F/||J||_F is below 1e-4 for the conservative model but remains measurable for direct prediction, and fixed-point divergence begins near alpha_c approximately 1/lambda_max(J), with the damped model remaining stable for alpha below 0.8/lambda_max. Measure whether free-energy descent and wall-profile error improve out of distribution.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built and ran a scalar convolutional free-energy network whose differentiated output has exactly zero reciprocity residual, while an equal-parameter direct predictor had residual 0.0314. The measured Jacobian multiplier matched 1-alpha+alpha*lambda to about 1e-8, confirming the linearization mechanism, but the proposed alpha*lambda<1 stability rule was false for lambda>1: damping increased the positive-mode multiplier above one. Thus integrability manifested, but the full claimed mechanism did not pass the required predictions.
- Agent confidence
- 9/10
- Baseline
- Direct predictor: test MSE 0.00719, reciprocity residual 3.14e-2, 241 parameters
- Idea
- Conservative functional: test MSE 0.00504, reciprocity residual 0.0, path-integral error 4.79e-7, 241 parameters; stability sweep confirmed multiplier formula but contradicted alpha*lambda<1 as a sufficient condition
Limitations: Only a 1D synthetic periodic convolution problem was tested, not 3D hard-cube Monte Carlo, wall profiles, free-energy descent, or realistic out-of-distribution inference. CUDA was attempted but failed in the local PyTorch convolution backend, so the successful run used CPU.
How to run: python3 experiment.py --cpu
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- experiment.py 6.6 KB View Raw
- report.md 1.4 KB View
- results.json 7.9 KB View Raw