✓✓ Beats tuned baseline 2026

Slow-Mode-Canceling Optimizer Packet

Implementation & benchmark of arXiv:2608.25141 — Thermalization packets and optimal ice cubes

Usefulness7/10
Difficulty6/10
Novelty8/10

Source paper: Thermalization packets and optimal ice cubes arXiv:2608.25141 ⓘ · analyzed Sep 1, 2026

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

Idea description

Train two parameter replicas with symmetric coupling, treating one replica as a prepared thermalization packet for the other. Estimate the slow local Hessian direction and initialize or periodically reset the packet so that the coupled state has zero projection onto that mode; the target should then relax according to the next-slowest mode rather than the original bottleneck.

Formulas

$$\dot{x}=-Hx+\kappa(y-x),\qquad \dot{y}=-Hy+\kappa(x-y),\qquad \kappa>0.$$
$$Hv_i=\lambda_i v_i,\quad c_i=\frac{v_i^\top x+v_i^\top y}{2},\quad d_i=\frac{v_i^\top x-v_i^\top y}{2},\qquad \dot c_i=-\lambda_i c_i,\quad \dot d_i=-(\lambda_i+2\kappa)d_i.$$
$$v_1^\top y(0)=-v_1^\top x(0)\ \Rightarrow\ c_1(0)=0,\qquad r_{\mathrm{after}}\approx\min\{\lambda_2,\lambda_1+2\kappa\}.$$
$$\theta_{t+1}=\theta_t-\eta[g_\theta+\kappa(\theta_t-\phi_t)],\qquad \phi_{t+1}=\phi_t-\eta[g_\phi+\kappa(\phi_t-\theta_t)].$$

Mathematical statement

Near a stationary point, let the target and packet parameter errors be x=theta-theta^star and y=phi-theta^star, and approximate the loss gradient by grad L(theta^star+x) approximately Hx, where H is the local Hessian. Use coupled gradient-flow dynamics dot x=-Hx+kappa(y-x) and dot y=-Hy+kappa(x-y), with coupling strength kappa>0. For a Hessian eigenvector v_i satisfying Hv_i=lambda_i v_i, define scalar coordinates a_i=v_i^T x and b_i=v_i^T y. Their common and difference coordinates are c_i=(a_i+b_i)/2 and d_i=(a_i-b_i)/2, obeying dot c_i=-lambda_i c_i and dot d_i=-(lambda_i+2kappa)d_i. If lambda_1<lambda_2 is the slowest nonzero Hessian eigenvalue, ordinary initialization has asymptotic error proportional to c_1(0) exp(-lambda_1 t), while packet preparation b_1(0)=-a_1(0) gives c_1(0)=0 and removes this mode exactly. The resulting asymptotic rate is at least min(lambda_2,lambda_1+2kappa), subject to nonlinear and stochastic errors.

Implementation notes

1. Integration point: add a second trainable parameter vector phi with the same architecture as the ordinary model parameter vector theta. At every minibatch, compute both losses and update both replicas with gradients plus a parameter-coupling term. Use the target prediction from theta; use phi only as the packet and optionally average predictions late in training. The discrete Euler updates are theta <- theta-eta[g_theta+kappa(theta-phi)] and phi <- phi-eta[g_phi+kappa(phi-theta)]. 2. Pseudocode: initialize theta and phi, estimate a low-curvature direction v1 using 5--20 Hessian-vector products and Lanczos or inverse iteration, then estimate the displacement x relative to a nearby reference theta_star or an initial checkpoint. Set phi so that v1^T(phi-theta_star)=-v1^T(theta-theta_star), while leaving orthogonal components unchanged. Run the coupled updates and re-estimate v1 every 100--1000 steps. 3. Computed versus estimated: the modal equations and cancellation condition are analytic; v1, lambda_1, and lambda_2 are estimated from Hessian-vector products or by fitting exponential decay rates of projected errors. Minibatch noise will continually repopulate the cancelled mode, so measure its replenishment rate. 4. First cheap experiment: use a two-layer MLP on MNIST and compare SGD, an unprepared two-replica optimizer, and the prepared optimizer at matched total parameter-update FLOPs. Measure loss, validation accuracy, and error projected onto v1. The falsifiable prediction is that the late-time projected error changes from A exp(-lambda_1 t) to approximately B exp(-min(lambda_2,lambda_1+2kappa)t). Increasing kappa should improve the decay rate only until lambda_1+2kappa approximately equals lambda_2. A cancellation ratio |c_1(0)|/|a_1(0)| below 0.1 should yield at least a tenfold reduction in slow-mode amplitude. For discrete updates, instability should appear when eta(lambda_max+2kappa) reaches approximately 2.

Verification

Beats tuned baseline

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Not reproduced ✗

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

How to read this: The implementation beat the baseline, but the claimed mechanism did not reproduce at benchmark scale. The gain may come from another effect; treat the cause as unresolved.

Stage 1 — Mechanism check agent confidence 9/10

Built and numerically verified the coupled two-replica optimizer on a diagonal quadratic Hessian. Exact slow-mode cancellation was observed, the measured rate matched the discrete prediction across the kappa sweep to numerical precision, saturation occurred at kappa=0.3 as predicted, and the Euler stability boundary was predicted at kappa=19.0 with divergence observed at kappa=19.05. The prepared packet reached 1.05% of the unprepared final loss in this toy case.

Agent confidence
9/10
Baseline
Unprepared coupled replicas: final loss 0.0370561 at eta=0.05, kappa=0.35, 100 steps; slow-mode amplitude after 30 steps was 0.860384 of initial target amplitude.
Idea
Prepared packet: final loss 0.000388824 under identical settings, or 0.01049x baseline; exact initial c1 cancellation and c1=0 after 30 steps. Rate sweep predicted/observed continuous-equivalent rates matched: 0.100251/0.100251, 0.201007/0.201007, 0.404054/0.404054, 0.712544/0.712544, and 0.712544/0.712544 for kappa 0, 0.05, 0.15, 0.3, and 0.6. Stability critical kappa was predicted as 19.0; kappa=19.05 and 19.5 diverged while kappa<=19.0 did not.

Limitations: Only a deterministic diagonal quadratic/local-Hessian model was tested; no MLP, MNIST, stochastic minibatch replenishment, Hessian-vector estimation, nonlinear effects, wall-clock or FLOP comparison, or practical packet-reset schedule was evaluated.

How to run: python3 slow_mode_experiment.py

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

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

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
10.0418
Idea mean
7.6948
Effect (Δ)
-2.3471 (−23.4%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±5.3%
Mechanism
Not confirmed ✗
Practical effect
Helps
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular track was tested. The experiment used a tiny MLP and 400-sample Friedman#1 data; no vision, sequence, dynamics, larger architectures, wall-clock/FLOP accounting, Hessian-vector/Lanczos estimation, or packet re-estimation schedule was tested.

How to run: python3 bench_experiment.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

Implemented the prepared two-replica coupled optimizer on the structurally matched Friedman#1 tabular benchmark with the shared mlp_tiny architecture. Across 8 paired seeds it significantly improved test MSE versus the tuned baseline. Initial cancellation occurred, but the projected common mode was repopulated during training, so the persistent mechanism prediction was not confirmed.

Baseline
Tuned baseline lr=0.006, weight_decay=0.0: mean test MSE 10.0418401957 over 8 seeds.
Idea
Prepared packet, lr=0.006, kappa=0.03: mean test MSE 7.6947646737 over 8 seeds; delta=-2.3470755219, 8/8 wins, permutation p=0.0081.
Mechanism
Initial cancellation ratio 3.7e-10; observed ratio 0.2168 after epoch 1 and 1.0423 at the end; persistent cancellation not confirmed.

Benchmark result

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

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
10.0418
Idea mean
7.6948
Effect (Δ)
-2.3471 (−23.4%; negative = idea better)
Wins
8 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±5.3%
Mechanism
Not confirmed ✗
Practical effect
Helps
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular track was tested. The experiment used a tiny MLP and 400-sample Friedman#1 data; no vision, sequence, dynamics, larger architectures, wall-clock/FLOP accounting, Hessian-vector/Lanczos estimation, or packet re-estimation schedule was tested.

How to run: python3 bench_experiment.py

Artifacts

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