Impedance-Calibrated Learning-Rate Control
Implementation & benchmark of arXiv:2609.02458 — Impedance in Periodically Driven Stochastic Systems
Source paper: Impedance in Periodically Driven Stochastic Systems arXiv:2609.02458 ⓘ · analyzed Sep 3, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Treat local neural-network training as a driven linear system and periodically modulate the learning rate by a small sinusoid. Estimate the transfer function from this modulation to loss or gradient observables, fit its relaxation poles, and set the learning rate below the measured instability boundary.
Formulas
Mathematical statement
Linearize training dynamics as \(\dot{x}=Ax+Bu\), where \(x\) is parameter displacement, \(A\) is the local training Jacobian, \(u\) is a scalar learning-rate modulation, and \(B\) is its input direction. For observable \(y=Cx\), with observation row vector \(C\), the frequency response is \(H(i\omega)=C(i\omega I-A)^{-1}B\). For real stable modes, \(H(i\omega)=\sum_k a_k/(1+i\omega\tau_k)\), where \(\tau_k=-1/\lambda_k>0\) are relaxation times, \(\lambda_k\) are stable eigenvalues, and \(a_k\) are residues. For a quadratic local loss, discrete gradient descent has mode update \(z_{k,t+1}=(1-\eta\lambda_k)z_{k,t}\), stable when \(0<\eta<2/\lambda_k\).
Implementation notes
Integrate this as a wrapper around SGD, initially avoiding Adam so the identified dynamics remain interpretable. Every K optimizer steps, apply a small learning-rate probe \(u_t=\epsilon\sin(\omega t)\) for several frequencies spanning approximately one decade below and above the estimated inverse relaxation time. Record minibatch loss, gradient norm, and optionally a random projection of parameter displacement. For each frequency, estimate the complex response with lock-in correlations, \(\widehat H(\omega)=2\langle y_t e^{-i\omega t}\rangle/(\epsilon T)\), over an integer number of probe periods. Fit one to four positive time constants and residues by nonlinear least squares. The paper-derived quantities are the response poles and RC-style decomposition; the Jacobian, residues, and effective curvatures are estimated empirically. Infer \(\widehat{\lambda}_{\max}=2/\widehat{\eta}_{\mathrm{crit}}\) from the largest learning rate whose oscillations still decay, then operate at \(\gamma\widehat{\eta}_{\mathrm{crit}}\), with \(\gamma=0.7\). First experiment: train a three-layer MLP on MNIST and compare fixed SGD, cosine decay, and impedance control at equal update count. Sweep fixed learning rates around the predicted boundary. The falsifiable signature is a sharp transition from decaying to growing or alternating loss oscillations near \(\eta_c\); the impedance estimate should predict \(\eta_c\) within 20 percent, while controlled training should remain below the boundary and reduce settling time.
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 8/10
Built a reproducible lock-in transfer-function sanity check, quadratic stability test, and small CUDA digits-MLP comparison in experiment.py. The core math was confirmed: corrected lock-in recovery had 0.82% relative error, and the quadratic transition occurred at the predicted eta=2/lambda_max=0.4. The training controller showed a promising result versus the selected fixed and cosine baselines, with test loss 0.0755 versus 0.1098 and 0.1616, but this is not a demonstrated impedance-control win because the implementation used Hessian power estimates rather than fitted sinusoidal poles/residues, and fixed-rate eta=0.6 performed better.
- Agent confidence
- 8/10
- Baseline
- Fixed SGD eta=0.08: test loss 0.1098, accuracy 97.78%; cosine SGD: test loss 0.1616, accuracy 96.67%; fixed-rate eta=0.6 reached sampled final loss 0.0265.
- Idea
- Curvature-controlled SGD: test loss 0.0755, accuracy 98.00%, final learning rate 0.35; lock-in transfer relative error 0.0082; predicted critical eta 0.400, grid estimate 0.395; quadratic eta=0.39 decayed while eta=0.41 and 0.50 grew.
Limitations: Only one random seed and one small sklearn digits dataset were tested. The real-training prototype does not implement the proposed multi-frequency probing, pole/residue nonlinear fit, or loss/gradient lock-in estimator; it estimates curvature directly with a short Hessian-vector power iteration. No FLOP-normalized or repeated-seed comparison, larger model, MNIST, or robust instability-boundary sweep was performed.
How to run: python3 experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- experiment.py 6.3 KB View Raw
- report.md 1.8 KB View
- results.json 10.3 KB View Raw