Order-Adaptive Integral Optimizer
Source paper: Order-Adaptive Distributed Integral Control arXiv:2609.00688 ⓘ · analyzed Sep 2, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace a fixed optimizer memory order with a nested family of gradient-integral controllers. Training begins with a first-order update and activates additional accumulated-gradient states only after an exponentially smoothed residual fails to decrease for several decision intervals; newly activated gains are ramped from zero, so the parameter update remains continuous and previously learned states are preserved. The optimizer should use little memory on easy problems and acquire longer memory on ill-conditioned or delayed-gradient problems.
Formulas
Mathematical statement
For parameters \(\theta\in\mathbb{R}^d\), gradient \(g(\theta)=\nabla L(\theta)\), and order \(p\), maintain integral-chain states \(z_1,\ldots,z_p\in\mathbb{R}^d\) with \(\dot z_1=g\) and \(\dot z_{r+1}=z_r\). The continuous optimizer is \(\dot\theta=-a_0g-\sum_{r=1}^{p}a_rz_r\). Around a quadratic mode \(L(\theta)=\frac12q\theta^2\), where \(q>0\), each eigenmode has characteristic polynomial \(P_{p,q}(\lambda)=\lambda^{p+1}+a_0q\lambda^p+a_1q\lambda^{p-1}+\cdots+a_pq\). Choose positive coefficients satisfying the paper's adjacent log-concavity condition \(a_l^2>4a_{l-1}a_{l+1}\); this is a conservative design rule for separated, nonoscillatory roots. Candidate orders are nested because order \(p+1\) retains \(z_1,\ldots,z_p\) and adds only \(z_{p+1}\). Let \(r_k\) be an exponential moving average of gradient norm or prediction residual. If \(r_k\) fails to contract below a threshold for \(K\) decision intervals, increase \(p\). A smooth activation \(\gamma_p(t)\in[0,1]\) multiplies the newly added gain and prevents discontinuous updates.
Implementation notes
1. Integration point: implement this as a PyTorch optimizer around any MLP, CNN, Transformer, or language-model training loop. Start with order \(p=0\), meaning ordinary gradient descent or AdamW without changing the network architecture. For each parameter tensor, maintain optional states \(z_1,z_2,\ldots,z_{p_{\max}}\); use a global order decision initially, then test layerwise order as a follow-up. 2. Pseudocode: compute gradients \(g\); update \(r\leftarrow\beta r+(1-\beta)\|g\|\); if the contraction test fails for \(K\) checks, increment \(p\); initialize the new state \(z_{p+1}=0\), set its activation \(\gamma=0\), and ramp \(\gamma\leftarrow\min(1,\gamma+1/H)\) over \(H\) optimizer steps; update \(z_1\leftarrow z_1+\eta g\), \(z_{r+1}\leftarrow z_{r+1}+\eta z_r\), and \(\theta\leftarrow\theta-\eta[a_0g+\sum_r\gamma_ra_rz_r]\). Apply norm clipping to every state to prevent integral windup. 3. Computed from the paper: nested order expansion, residual-triggered finite decision intervals, continuous transitions, and polynomial gain constraints. Estimated empirically: gradient contraction \(r_k/r_{k-1}\), Hessian curvature range, and the discrete-time stability boundary. For a local linear model, estimate dominant Jacobian eigenvalues by Lanczos and compare the measured Euler stability limit \(\eta_c\) with \(\eta_c=\min_j[-2\operatorname{Re}(\lambda_j)/|\lambda_j|^2]\). 4. First cheap experiment: train a 2-layer MLP on MNIST and a small Transformer on Penn Treebank, comparing SGD, fixed momentum, fixed order-1 integral memory, and adaptive orders \(0\rightarrow1\rightarrow2\). Use \(\beta=0.95\), \(\rho=0.98\), \(K=20\), \(H=50\), and coefficients satisfying the displayed inequality. The primary falsifiable signature is a plateau in \(r_k\) followed by an order transition, after which residual decay accelerates without a discontinuous loss spike. On quadratic regression, the measured divergence boundary should agree with the Jacobian prediction within 20%; on easy well-conditioned data, the adaptive method should remain at \(p=0\) for most training, while on ill-conditioned data it should activate higher order and produce faster late-time residual decay.
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.