Lee-Yang Phase-Diagram Loss for Neural Potentials
Implementation & benchmark of arXiv:2608.08546 — Lee-Yang Theory Guided Force Field Refinement Based on Phase Diagrams
Source paper: Lee-Yang Theory Guided Force Field Refinement Based on Phase Diagrams arXiv:2608.08546 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Train a neural energy model using a loss that matches the modulus of its partition function in a small complex neighborhood of target phase-transition points. Instead of fitting only local energies or a selected order parameter, the model is forced to place its finite-size Lee-Yang zero minima at the correct temperature, pressure, or chemical-potential coordinates, providing a global thermodynamic constraint.
Formulas
Mathematical statement
For a system with neural energy U_theta(x), inverse temperature beta = 1/(k_B T), extensive observable M(x), and complex field h = h_r + i h_i, define Z_theta(beta,h) = integral dx exp[-beta(U_theta(x) - h M(x))]. Relative to samples from a real reference state h_r, the normalized complex partition function is phi_theta(h_i;h_r) = average[exp(i beta h_i M)] over the neural model at h_r. Lee-Yang zeros satisfy Z = 0, equivalently phi = 0 after normalization; in finite systems the minimum h_i*(h_r,N) = argmin over h_i > 0 of |phi_theta(h_i;h_r)| estimates the closest zero. The training loss compares log moduli across state points and imaginary-field values, with epsilon preventing numerical singularities. Near a first-order transition, the closest zero is expected to approach the real axis approximately as N^(-1), while it remains separated from the real axis away from coexistence.
Implementation notes
Integrate this as an auxiliary loss for a neural interatomic potential, neural Hamiltonian, or differentiable energy-based model. For each training state q_j = (T_j,p_j), run short equilibrium simulations using the current U_theta, collect configurations x_n and the extensive observable M_n associated with the field being complexified. For a liquid-gas transition, use particle number or volume; for a magnetic transition, use magnetization; for a structural transition, use an extensive bias variable that distinguishes the phases. Pseudocode: (1) sample x_n from the current model at q_j; (2) for each imaginary-field value h_i,k compute phi_hat_jk = (1/N_s) sum_n exp(i beta_j h_i,k M_n); (3) compute the log-modulus loss against target curves or target minimum locations; (4) backpropagate through differentiable energy evaluations or use a reweighting estimator with the simulation samples treated as fixed; (5) update theta and periodically resample because the distribution changes. The paper supplies the complex partition-function objective and the interpretation of zeros; sampling variance, target curves, and finite-size corrections must be estimated empirically. The first cheap experiment is a two-dimensional Lennard-Jones fluid with a small neural correction to the analytic potential, system sizes N = 64, 128, and 256, coexistence temperatures from reference simulation as targets, and ordinary energy/force fitting as baseline. The quantitative prediction is that the learned model's minimum h_i*(N) decreases approximately as N^(-1) at the target transition but remains bounded away from zero at off-transition temperatures. Accept the mechanism only if the inferred transition agrees with the reference within 20 percent and the fitted finite-size exponent is reasonably close to 1 for the first-order case. Held-out enthalpy and heat capacity should also be measured as secondary validation.
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 numerically verified a differentiable Lee–Yang partition-function loss MVP plus an exactly solvable two-phase sweep. The normalized complex-partition identity was correct to 1.1e-16, coexistence gave h1 proportional to N^-1 with observed exponent -1.0000, and the finite-sample modulus floor followed the predicted Ns^-1/2 scaling with exponent -0.540. However, away from coexistence the minimum location still scaled as N^-1 rather than remaining bounded away from zero, and the neural modulus fit did not outperform the moment baseline and showed an identifiability limitation.
- Agent confidence
- 9/10
- Baseline
- Moment fit: parameter error 6.60e-06 in the initial zero-field comparison; 0.1227 in the nonzero-field optimization comparison.
- Idea
- Lee–Yang curve fit: exact analytic curve recovery in the sweep; nonzero-field neural fit parameter error 0.1235, final loss 1.74e-09; coexistence exponent -1.0000 versus predicted -1; sampling-noise exponent -0.540 versus predicted -0.5; off-coexistence minimum exponent -1.000 instead of the claimed bounded-away-from-zero behavior.
Limitations: This was an exactly solvable two-state toy system, not a Lennard-Jones fluid or neural interatomic potential. No molecular-dynamics sampling, pressure/temperature phase diagram, force fitting, held-out thermodynamic observables, resampling drift, or GPU training was tested.
How to run: python3 lee_yang_toy.py && /home/maxwelhelp/main/bin/python3 ly_mvp.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- lee_yang_toy.py 4.9 KB View Raw
- ly_mvp.py 3.1 KB View Raw
- mvp_results.json 0.4 KB View Raw
- report.md 1.7 KB View
- results.json 2.0 KB View Raw