Bounded Telegraph Exploration for Optimizers
Implementation & benchmark of arXiv:2608.29226 — Ornstein-Uhlenbeck Process Driven by Multiple Dichotomous Noises
Source paper: Ornstein-Uhlenbeck Process Driven by Multiple Dichotomous Noises arXiv:2608.29226 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Add a bounded colored exploration force to an optimizer by filtering a sum of independent two-state telegraph signals through a stable linear relaxation equation. Unlike Gaussian momentum noise, the perturbation has a strict amplitude bound and a tunable finite correlation time, reducing rare destructive parameter excursions while retaining structured exploration.
Formulas
Mathematical statement
For each parameter coordinate or parameter block, define u(t) by du/dt = -gamma u(t) + sum_k v_k s_k(t), where gamma > 0 is the relaxation rate, v_k > 0 are amplitudes, and s_k(t) in {-1,+1} are independent symmetric telegraph processes switching at rate lambda_k. The stationary process obeys |u(t)| <= V/gamma, where V = sum_k v_k. For symmetric switching, E[u] = 0, Var(u) = sum_k v_k^2/[gamma(gamma+2 lambda_k)], and the autocovariance is C_u(tau) = sum_k v_k^2[exp(-2 lambda_k |tau|) - exp(-gamma |tau|)]/[gamma^2 - 4 lambda_k^2] when gamma is not 2 lambda_k. These relations give exact amplitude, variance, and correlation-time targets for optimizer noise.
Implementation notes
Integrate the mechanism into a momentum-like optimizer state rather than directly perturbing weights. For parameter vector theta, maintain telegraph force u and update theta <- theta - eta [g(theta) + alpha u], where g is the minibatch gradient, eta is the learning rate, and alpha scales exploration. Use one scalar telegraph process per parameter block or layer to avoid storing a sign for every parameter. With step size dt, flip each sign s_k independently with probability 1 - exp(-lambda_k dt), then apply the exact linear relaxation update u <- exp(-gamma dt) u + [1 - exp(-gamma dt)] sum_k v_k s_k/gamma. Pseudocode: initialize u = 0 and random signs; for each minibatch compute g; flip signs; update u; optionally clamp only roundoff excursions to [-u_max,u_max]; apply theta <- theta - eta(g + alpha u). The paper supplies the bounded-support and stationary-moment mechanism; estimate empirical variance and autocorrelation from the run. First test a small MLP and ResNet-18 on CIFAR-10 against SGD and SGD with Gaussian OU noise at matched variance. Sweep gamma, K, and lambda_k. The quantitative prediction is |u| <= u_max at every step, with violations below numerical precision. Measured stationary variance should match the closed form within 10%, and the autocorrelation should fit the predicted sum of exponentials. As K increases with fixed total variance, standardized excess kurtosis should decrease toward zero.
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 tested bounded telegraph exploration with exact discrete relaxation. The strict bound held in every run (maximum observed |u|/u_max=0.999999), and the variance law matched reasonably well overall. However, the supplied covariance formula is invalid because it predicts C(0)=0; the corrected covariance matched simulation much better, while kurtosis scaling was noisy and the quadratic optimizer showed no meaningful win over matched-variance OU noise.
- Agent confidence
- 9/10
- Baseline
- Matched-variance Gaussian OU: final quadratic loss 0.02503; average last-500 loss 0.03497.
- Idea
- Telegraph: final quadratic loss 0.02832; average last-500 loss 0.03463; maximum observed bound ratio 0.999999; covariance RMSE 0.0355 for the corrected formula versus 0.2371 for the supplied formula.
Limitations: Only a toy quadratic optimizer was tested, not CIFAR-10, an MLP, or ResNet. No wall-clock/FLOP or generalization measurement was performed, and the finite Monte Carlo horizon made the kurtosis sweep noisy.
How to run: python3 telegraph_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- report.md 1.3 KB View
- results.json 3.4 KB View Raw
- telegraph_experiment.py 4.7 KB View Raw