✓✓ Beats tuned baseline 2026

One-Bang Gradient-Noise Preparation

Implementation & benchmark of arXiv:2607.12955 — Optimal preparation and reachable-state constraints in the Mpemba effect

Usefulness7/10
Difficulty5/10
Novelty7/10

Source paper: Optimal preparation and reachable-state constraints in the Mpemba effect arXiv:2607.12955 ⓘ · analyzed Aug 30, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Use a bounded stochasticity control during an initial preparation window to shape the gradient or parameter-update distribution before ordinary training. The control is restricted to its minimum or maximum value, with at most one switch, because the reduced moment dynamics are affine in the control; this gives a falsifiable alternative to smooth noise or learning-rate annealing.

Formulas

$$T=\frac{1}{d}\mathbb{E}[\|g\|^{2}],\qquad a_{2}=\frac{d}{d+2}\frac{\mathbb{E}[\|g\|^{4}]}{\mathbb{E}[\|g\|^{2}]^{2}}-1.$$
$$\dot T=F(T,a_{2})+B_T(T,a_{2})u,\qquad \dot a_{2}=G(T,a_{2})+B_a(T,a_{2})u,\qquad u\in[u_{\min},u_{\max}].$$
$$\mathcal{H}=p_T[F+B_Tu]+p_a[G+B_au],\qquad S(t)=\frac{\partial\mathcal{H}}{\partial u}=p_TB_T+p_aB_a.$$
$$u^{\star}(t)=\begin{cases}u_{\min},&S(t)>0,\\u_{\max},&S(t)<0,\end{cases}\qquad \text{with at most one switch predicted during preparation.}$$

Mathematical statement

For an update vector g in R^d, define its effective temperature T=E[||g||^2]/d and excess kurtosis a_2=(d/(d+2))E[||g||^4]/E[||g||^2]^2-1, paralleling the paper's kinetic state variables. Let u(t) in [u_min,u_max] be the controlled gradient-noise variance or learning-rate multiplier, and approximate the preparation dynamics by dT/dt=F(T,a_2)+B_T(T,a_2)u and da_2/dt=G(T,a_2)+B_a(T,a_2)u. For terminal objective J=phi(T(t_p),a_2(t_p)), Pontryagin's Hamiltonian is H=p_T[F+B_Tu]+p_a[G+B_au], where p_T and p_a are costates. Its control coefficient, or switching function, is S(t)=p_TB_T+p_aB_a. Minimizing J gives u=u_min when S>0 and u=u_max when S<0, with signs reversed for maximization. Thus an affine reduced model predicts a bang-bang schedule. Reachable states are obtained by integrating the two extreme controls; target kurtosis values outside the resulting interval are unattainable under the chosen optimizer and batch-noise amplitude.

Implementation notes

Integrate this at the optimizer or scheduler level for the first t_p updates of training. Choose a bounded control u to represent injected Gaussian gradient noise, dropout probability, or a multiplicative learning-rate/noise scale; the cleanest MVP is g_t=\widehat g_t+\sqrt{u_t}\epsilon_t, with epsilon_t sampled from a standard normal distribution and u_t in [u_min,u_max]. Maintain exponential moving averages of m_2=||g_t||^2 and m_4=||g_t||^4, then estimate T=m_2/d and a_2=(d/(d+2))m_4/m_2^2-1. First fit the local reduced dynamics by finite differences: run short constant-control probes at u_min and u_max, regress Delta(T,a_2) on (T,a_2,u), and estimate F,G,B_T,B_a. Specify a terminal target, such as minimizing a_2 at fixed T or maximizing the subsequent loss-decay rate over the next K updates. Numerically integrate the fitted two-dimensional model under both extreme controls, use the better endpoint, and implement a single-switch schedule: u_max for k<k_s and u_min afterward, or the reverse. The paper contributes the affine-control and Pontryagin mechanism; the moment coefficients and reachable interval must be estimated empirically for each network and optimizer. The first cheap experiment should use a two-layer MLP on MNIST or CIFAR-10, comparing Adam or SGD with constant noise, cosine noise annealing, random schedules, and the one-switch controller. Measure gradient kurtosis, loss after the preparation window, and later loss-relaxation rate r_K=[L(t)-L(t+K)]/K. The quantitative prediction is that the best preparation is boundary-valued with one switch, while interior smooth schedules should not improve the terminal objective beyond estimation error; the achieved a_2(t_p) should lie inside the envelope generated by constant u_min and u_max trajectories. Vary the control range and test whether the attainable a_2 interval expands monotonically and whether the best relaxation-rate asymmetry saturates at the predicted reachable boundary.

Verification

Beats tuned baseline

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Confirmed ✓

Stage 2 · Practical benchmark result: Significant win ✓

Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works

Stage 1 — Mechanism check agent confidence 7/10

Built a reproducible affine reduced-dynamics verifier and a small CUDA/CPU MLP preparation experiment. Endpoint displacement scaled exactly linearly with control range, the terminal objective selected a boundary control, and a one-switch schedule beat 3,000 random binary schedules (0.0241 vs 0.0267 objective). In the neural test, one-switch noise gave slightly lower final validation loss than constant, cosine, and random preparation, but the gain was small and accuracy was unchanged.

Agent confidence
7/10
Baseline
Constant-mid noise: validation loss 0.128286, accuracy 0.96875, post-preparation relaxation rate 0.0006741. Cosine: 0.128350, 0.96875, 0.0006720. Random: 0.128749, 0.96875, 0.0006791.
Idea
One-switch noise: validation loss 0.127486, accuracy 0.96875, post-preparation relaxation rate 0.0006561. Toy range ratios observed [0.25, 0.50, 1.00, 1.50] versus predicted [0.25, 0.50, 1.00, 1.50]; boundary optimum u=1.0; one-switch objective 0.02408 versus random-binary best 0.02666 and median 0.03919.

Limitations: The reduced dynamics are a specified affine toy model rather than coefficients fitted from measured neural-network moment trajectories. The neural experiment used one seed, one synthetic dataset, one optimizer and noise range, and did not test MNIST/CIFAR-10, multiple seeds, FLOPs, or broad robustness.

How to run: python3 one_bang_experiment.py

Stage 2 — Benchmark vs. tuned baseline (arXiv:2607.12955)

Beats the tuned baseline by 38.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
4.2488
Idea mean
2.6338
Effect (Δ)
-1.6149 (−38.0%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±15.0%
Mechanism
Confirmed ✓
Practical effect
Helps
Baseline tuning
swept over 6 configs
Limitations:

Only the registered tabular/Friedman#1 track with mlp_tiny was tested. Vision, sequence, and dynamics tracks were not run; reduced coefficients F,G,B_T,B_a were not fitted from trajectories; and wall-clock/FLOP efficiency was not measured.

How to run: python3 bench_one_bang.py

Verdict computed by deterministic test code from paired-seed statistics — not by the language model.

Stage 2 — Benchmark latest Worked ✓

Agent confidence: 8/10

The registered tabular Friedman#1 benchmark shows a significant improvement from one-bang gradient-noise preparation: mean test MSE decreased from 4.248755 to 2.633840 across all 8 paired seeds. The paired delta is -1.614916 with permutation p=0.0081, so the bench verdict is idea better (significant).

Baseline
Adam baseline, best configuration lr=0.006 and weight_decay=0.0001; 8-seed mean MSE 4.248755484819412.
Delta mean
-1.6149159371852875
Idea
One-bang Adam, same best configuration, U_HIGH=0.0004 for the first 80 of 160 preparation updates followed by u=0; 8-seed mean MSE 2.6338395476341248.
Idea wins
8
P value
0.0081

Benchmark result

Beats the tuned baseline by 38.0% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
4.2488
Idea mean
2.6338
Effect (Δ)
-1.6149 (−38.0%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±15.0%
Mechanism
Confirmed ✓
Practical effect
Helps
Baseline tuning
swept over 6 configs
Limitations:

Only the registered tabular/Friedman#1 track with mlp_tiny was tested. Vision, sequence, and dynamics tracks were not run; reduced coefficients F,G,B_T,B_a were not fitted from trajectories; and wall-clock/FLOP efficiency was not measured.

How to run: python3 bench_one_bang.py

Artifacts

Implementation overview ⬇ Download all as ZIP 6 files · code, reports and structured results