Nonreciprocal Two-Replica Optimizer
Implementation & benchmark of arXiv:2607.20420 — Optimal Finite-Time Control of Nonreciprocal Brownian Dimers: Thermodynamic Anomaly and Multiple Transitions
Source paper: Optimal Finite-Time Control of Nonreciprocal Brownian Dimers: Thermodynamic Anomaly and Multiple Transitions arXiv:2607.20420 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Represent a trainable parameter block by a center state \(c\) and an auxiliary separation state \(r\), and couple them asymmetrically so that the auxiliary state can transiently push the parameter center in useful directions. Bound the auxiliary control using either hard clipping or smooth saturation. This tests whether the paper's distinct transition mechanisms can regulate exploratory optimizer motion without destabilizing training.
Formulas
Mathematical statement
For the dimer, define \(c=(x_1+x_2)/2\), \(r=x_2-x_1-q_0\), \(Y=(\lambda_1+\lambda_2)/2\), \(R=\lambda_2-\lambda_1-q_0\), \(\alpha=(a-b)/2\), and \(\rho=1+a+b\). Averaging and subtracting the particle equations gives a center mode driven by the separation mode, \(\dot c=-(c-Y)+\alpha r+\xi_c(t)\), and a relaxing separation mode, \(\dot r=-\rho r+R+\xi_r(t)\), with noise terms obtained by combining the two independent particle noises. The deterministic fixed-trap system is stable when \(\rho>0\); use the paper's extension-restoring regime \(\rho\geq1\). Transfer this structure to optimization by replacing the restoring center force with the minibatch gradient, retaining auxiliary relaxation \(\rho\), and using \(\alpha r\) as a controlled exploratory correction. The auxiliary control \(R\) must be bounded to prevent the unregularized finite-time anomaly.
Implementation notes
1. Integration point: replace the SGD update for a selected parameter block \(\theta\) by center state \(c\) and auxiliary state \(r\) with the same shape. Use \(c\) in the forward pass, initialize \(c=\theta\) and \(r=0\), and retain \(r\) in optimizer state. Compute the minibatch gradient \(g_k=\nabla_c L(c_k)\). 2. Pseudocode: set an unconstrained auxiliary command \(\widehat R_k=-\beta g_k-\gamma r_k\); apply either \(R_k=\operatorname{clip}(\widehat R_k,-R_{\max},R_{\max})\) or \(R_k=R_{\max}\tanh(\widehat R_k/R_{\max})\); update \(r_k\leftarrow(1-\eta\rho)r_k+\eta R_k\); update \(c_k\leftarrow c_k-\eta g_k+\eta\alpha r_k\); copy \(c_k\) into the network parameters. Use \(\rho\geq1\), and optionally add minibatch Gaussian noise to \(R_k\) to mimic thermal forcing. 3. Quantities computed directly are gradients, \(r_k\), \(R_k\), and the correction norm \(\|\alpha r_k\|\). The paper supplies the modal decomposition and relaxation condition; \(\alpha,\beta,\gamma\), and \(R_{\max}\) are hyperparameters estimated by a small sweep. For explicit Euler, the isolated auxiliary mode is linearly stable only when \(0<\eta\rho<2\). 4. First experiment: train a two-layer MLP on MNIST using SGD, the same optimizer with \(\alpha=0\), and the nonreciprocal optimizer over \(\alpha\in\{0,0.25,0.5,1\}\), \(\rho\in\{1,2,4\}\), and hard versus smooth saturation. Record loss, gradient norm, auxiliary norm, correction norm, and the power spectrum of parameter motion. The quantitative prediction is a divergence boundary near \(\eta_c=2/\rho\), with auxiliary perturbations growing geometrically by factor \(|1-\eta\rho|\) beyond the boundary. As \(R_{\max}\) increases, hard clipping should show an abrupt jump in mean control magnitude, while smooth tanh saturation should show continuous growth; test whether the measured stability boundary agrees with \(2/\rho\) within 20%.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.