Nonreciprocal Brownian Optimizer
Implementation & benchmark of arXiv:2607.27740 — Non-reciprocity drives a Brownian dimer out of equilibrium
Source paper: Non-reciprocity drives a Brownian dimer out of equilibrium arXiv:2607.27740 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a single parameter iterate by two coupled replicas with unequal cross-couplings: replica 1 receives a force proportional to k_1(theta_1-theta_2), while replica 2 receives a force proportional to k_2(theta_2-theta_1), with k_1 not equal to k_2. The asymmetric coupling creates a controlled circulating component in the stochastic training dynamics, potentially helping escape flat saddles or correlated minibatch-noise traps without requiring an external periodic schedule. The coupling must be projected into the analytically stable region and annealed toward zero before final convergence.
Formulas
Mathematical statement
For one scalar coordinate of the dimer, let x=(x_1,x_2)^T be the two replica parameters, let gamma>0 be a relaxation scale, let k_0>0 be the restoring coefficient, and let k_1,k_2 be directed coupling strengths. Ignoring the loss gradient locally, the overdamped dynamics is dx=-A x dt+sqrt(2D)dW, where A=gamma^{-1}[[k_0+k_1,-k_1],[-k_2,k_0+k_2]], D=(T/gamma)I, T is the effective minibatch-noise temperature, and W is standard Brownian motion. The deterministic dynamics is asymptotically stable when trace(A)>0 and det(A)>0; for k_0>0 this reduces to k_0+k_1+k_2>0. The stationary covariance Sigma solves the continuous Lyapunov equation A Sigma+Sigma A^T=2D. The stationary probability current is J(x)=(D Sigma^{-1}-A)x p_ss(x), where p_ss is the Gaussian with covariance Sigma. Detailed balance requires J=0, which holds for reciprocal coupling k_1=k_2 under isotropic noise but is generically violated when k_1 is not equal to k_2. A measurable nonequilibrium strength is the entropy-production rate sigma_EP=integral J(x)^T D^{-1}J(x)/p_ss(x) dx, which is nonnegative, vanishes at reciprocity, and is generically quadratic in the asymmetry delta=k_1-k_2 near delta=0.
Implementation notes
1. Integration point: maintain two full parameter copies theta1 and theta2, compute independent minibatch gradients g1 and g2, and update them with directed cross-couplings. Start with scalar or layerwise shared coupling k1,k2 rather than a dense coupling matrix. Use k0 as a weak tether to the current averaged parameters theta_bar=(theta1+theta2)/2, or implement the displayed equations in a two-state auxiliary optimizer while the network prediction uses theta_bar. 2. Pseudocode: sample two minibatches; compute g1=grad(L(theta1,B1)) and g2=grad(L(theta2,B2)); draw independent standard Gaussian noises xi1 and xi2; set theta1 <- theta1-eta*(g1+k1*(theta1-theta2))+sqrt(2*eta*T)*xi1 and theta2 <- theta2-eta*(g2+k2*(theta2-theta1))+sqrt(2*eta*T)*xi2; evaluate the model using theta_bar. Apply k0 or weight decay and periodically reduce T,k1,k2. 3. Compute the predicted stability boundary from the local Jacobian: estimate the largest eigenvalue of the symmetric Hessian block or use a short power iteration, then require the discrete update spectral radius rho(I-eta A_local)<1; empirically log covariance Sigma, lagged velocity, and an antisymmetric-current estimator. The Lyapunov equation and current formula are imported from the paper; minibatch temperature T, local Hessians, and current histograms are estimated. 4. First cheap experiment: train a two-layer MLP on MNIST or CIFAR-10 with SGD, two-replica SGD, and the nonreciprocal version at matched compute and noise. Sweep delta=k1-k2 while fixing s=k1+k2 and test delta=0, 0.5s, s, and 2s. In a quadratic-loss toy problem, verify divergence begins when the measured spectral radius crosses one and remains within 20% of the predicted boundary. At fixed s, measure an entropy-production proxy from short parameter trajectories; it should be zero or statistically indistinguishable from zero at delta=0 and scale approximately as delta^2 for small delta. Test whether moderate asymmetry reduces escape time from an initialized saddle without increasing final loss.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Significantly worse ✗
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built the two-replica nonreciprocal optimizer math prototype and a small MLP experiment. The mechanism manifested: continuous stability transitioned at the predicted k0+k1+k2=0 boundary (observed between -1.0 and -0.9), the discrete Euler boundary was predicted at eta=1.0 and observed at 1.00002, and entropy production was zero at reciprocity while scaling as 0.35714·delta² with negligible fit-ratio variation. The MLP comparison did not show a practical win: baseline final loss was 0.67615 versus 0.68896 reciprocal and 0.68581 nonreciprocal.
- Agent confidence
- 9/10
- Baseline
- MLP final loss 0.676155; deterministic single-replica SGD
- Idea
- Nonreciprocal two-replica final loss 0.685811 (k1=0.95,k2=0.25); EP at delta=0 was 8.15e-32 and quadratic coefficient was 0.357143
Limitations: Only a tiny synthetic full-batch MLP was tested, not MNIST/CIFAR or minibatch-gradient temperature estimation. The optimizer used fixed noise/couplings and did not test annealing, saddle escape time, broad hyperparameter sweeps, matched FLOPs in a realistic training regime, or statistical repeats.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.27740)
Significantly WORSE than the tuned baseline (+42.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 4.8192
- Idea mean
- 6.8408
- Effect (Δ)
- 2.0217 (+42.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±2.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 6 configs
Only the prescribed tabular track and mlp_tiny model were tested; no vision, sequence, or dynamics track was run. The implementation used annealed deterministic coupling without explicit Brownian noise, and the mechanism signature used a simple epoch-level first-coordinate circulation proxy rather than a full stationary entropy-production estimate. The idea has approximately twice the parameter-update compute because it trains two replicas.
How to run: python3 bench_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
Implemented the nonreciprocal two-replica optimizer on the prescribed tabular optimizer track with the shared mlp_tiny architecture, paired seeds, baseline sweep, learning-rate parity, and 8-seed permutation testing. The trained replicas showed nonzero separation/circulation, confirming the mechanism signature, but the idea performed significantly worse: test MSE 6.8408 versus baseline 4.8192, paired delta +2.0217, permutation p=0.0081. Therefore it does not transfer as a practical win on this benchmark.
- Baseline
- mlp_tiny tabular, Adam, best lr=0.006 and weight_decay=0.0001; full 8-seed test MSE mean 4.8191644549, std 0.1716614496
- Idea
- two replicas with annealed k1=0.20, k2=0.05, best tested lr=0.003 and weight_decay=0.0001; full 8-seed test MSE mean 6.8408182263, std 0.1725772231
- Mechanism signature
- observed_epoch_area_seed0=-1.9684885011e-08, mean replica separation=0.0770557180, confirmed=true
- Paired delta
- idea - baseline = +2.0216537714; 0/8 idea wins; permutation p=0.0081
Benchmark result
Significantly WORSE than the tuned baseline (+42.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- mlp_tiny
- Paired seeds
- 8
- Baseline mean
- 4.8192
- Idea mean
- 6.8408
- Effect (Δ)
- 2.0217 (+42.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±2.8%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 6 configs
Only the prescribed tabular track and mlp_tiny model were tested; no vision, sequence, or dynamics track was run. The implementation used annealed deterministic coupling without explicit Brownian noise, and the mechanism signature used a simple epoch-level first-coordinate circulation proxy rather than a full stationary entropy-production estimate. The idea has approximately twice the parameter-update compute because it trains two replicas.
How to run: python3 bench_experiment.py
Artifacts
- bench_experiment.py 7.1 KB View Raw
- bench_report.json 4.2 KB View Raw
- experiment.py 5.5 KB View Raw
- report.md 1.3 KB View
- report_bench_2026-08-31T192247.md 4.6 KB View
- results.json 5.7 KB View Raw