Bitter-lesson-pilled sampling-free propagator yielding 1.5e-5 raw MSE

Sharing my Phase 1 write-up for submission #327838.

My angle was deliberately the unfashionable one. Reading the front page literally — “given the weights of a neural network, can you predict its expected per-neuron activations more accurately than running it many times?” — QMC is running it many times, so I spent the campaign on the other thing: a fully sampling-free propagator, deterministic in the weights, with no random draws anywhere in the forward pass.

The bet was bitter-lesson-shaped. Rather than hand-design a closure, bake high-quality data and let a scalable architecture learn the optimal propagator from it. So the first real work was the datasets, which are public:

  • keenanpepper/whestbench-relu-mlp-means-500k — 500k train + 2k low-noise validation MLPs, per-layer post-ReLU mean targets, no weights stored (every MLP regenerates exactly from its seed via a spec-portable recipe). Target noise ~1.2e-8 train, ~7.5e-10 validation, the latter matching the official eval targets.
  • keenanpepper/whestbench-relu-mlp-moments-10k — weights plus full pairwise moments to fourth order at N=1e8. This is what made the oracle/teacher-forcing measurements possible, and it’s the one I’d point at if you want to do this kind of work.

The estimator carries a mean/second-moment state, an O(N²) third-cumulant plane state standing in for the O(N³) tensor (whose exact transport bills 1.4–3.4× the entire budget), and a recurrent per-neuron latent, all trained end-to-end by backprop through the 32-layer recursion. It scores 1.50e-5 raw / 1.95e-6 adjusted, which is 4.8× under its own analytic backbone. The approach works.

The part I think is worth more than the artifact is what happened when I ablated my own submission. The learned attention closure the whole architecture was designed around is numerically inert — deleting it changes the prediction in the fifth significant figure while it consumes 61% of billed cost. A rebuild with that subtree removed is bit-identical and lands under the 0.1 multiplier floor unconditionally, i.e. I left a free 1.50e-6 adjusted on the table by shipping compute the network had learned to ignore. Roughly ninety numbered experiments went into a mechanism that trained itself to zero, and I only found out by ablating the shipped weights for the write-up.

The write-up also prices the whole family with oracle states. Inject true means and true covariance at every layer and the recursion bottoms out at 8.1e-7 raw — an oracle state genuinely does beat sampling at equal arithmetic, but because both methods sit near the multiplier floor that 18× raw advantage is worth only ~2× on the scored metric. So the honest answer to the challenge question at width 256 / depth 32 is no, not yet: sampling still wins, the barrier is state fidelity rather than the concept, and the entire remaining research program is worth about a factor of two. Wu et al. beat sampling at sufficient width to depth 12; I lose at depth 32. Whether that crossover is a width effect, a depth effect, or a depth/width effect is the measurement I’d most like to see someone do.

Also included: five measured walls, the negative results that shaped the design, and a methodology section on the practices that paid for themselves (pre-registered kill rules with numbers; through-train before believing any negative — frozen substitution misled me six times out of six).

AI disclosure: I directed the research and made the calls; Claude Opus 5 and Fable 5 did a large share of the coding, analysis, and drafting throughout, and the manuscript text plus the ablation/attribution/lean-rebuild measurements in it were produced by Opus 5 from the shipped artifact. Per the rules I’ve credited it as a co-author.

Feedback and criticism very welcome, especially from anyone who has pushed a learned propagator further than this.

main.pdf (681.5 KB)

5 Likes

I don’t have alot of feedback yet but I would like to say thanks for putting the huggingface datasets out publicly that was a genuinely useful, and also just a very generous thing to do.

1 Like

Would like to second this. The moments dataset was very useful to make ‘oracle’ tests!

I’ve spent a bit of time on something very similar this week and I don’t think it’ll take much for me to modify that setup to do this measurement (for the analytic-baseline vs plain-MC version of the question). I’ll update here when that’s done.

Update, as promised — the measurement is done, for the analytic-baseline vs plain-MC version of the question.

Setup. Uncorrected k=2 Hermite/Mehler moment chain (zero fitted parameters — the analytic floor) vs plain MC, both estimating final-layer post-ReLU per-neuron means on fresh He-Gaussian challenge-spec nets. Truth: MC at N = 2×10⁶. FLOP convention up front: chain ≈ 4·d·w³, one sample ≈ 2·d·w² → FLOP-matched MC gets n = 2·w samples.

One methodological note that cost me a rewrite: per-net MSEs are heavy-tailed for both methods (MC max/median grows 10→22× from d=24 to d=48), so mean-based cell ratios at small net counts are unstable. The table below is median-based, deep rows at 48 nets/cell, per-net records in the repo.

Ratio = analytic MSE / matched-MC MSE (< 1 ⇒ analytic wins):

w=64 128 256 512 1024
d=4 0.11 0.05 0.02 0.01 0.01
d=8 0.49 0.21 0.11 0.05 0.03
d=12 0.61 0.60 0.24 0.14 0.07
d=16 0.64 0.83 0.37 0.21 0.12
d=24 2.70 1.20 0.74 0.44 0.25
d=32 3.24 2.68 1.04 0.64 0.34
d=48 2.42 1.51 1.07 1.33 0.78

Answer: both effects are real; width dominates, monotonically. d*(w) ≈ <24 / 24 / 32 / 32–48 / ≥48 across w = 64…1024. The Phase 1 operating point (256, 32) sits at median parity — 1.04, bootstrap 95% CI [0.75, 1.38]. Amusingly, the Phase-1 operating point lands right at parity under this convention

Consistency checks against the question’s anchors: at d=12 the analytic floor already wins at every width tested (0.61 → 0.07) —matching Wu et al. beating sampling there; and fitted corrections shift every ratio downward (typically 5–10×), so the corrected-analytic crossover — the setting your oracle comparison lives in — sits systematically deeper than this floor contour.

Two structural observations:

  1. Analytic error is depth-flat after state saturation (w=256: 7.3/7.4/7.4 ×10⁻⁵ at d=24/32/48) while sampling’s per-net tails grow with depth — the property that survives depths where MC stops being feasible.
  2. The multiplier floor can’t flip the scored game either way — a sampler rides any floor down by using fewer samples, so scored winners are set by per-FLOP parity alone (small analysis in scored_setting.py). The quantity that would flip it is covariance/moment targets: n ≫ w samples for a w×w covariance vs one 4dw³ chain pass. (Independently re-derives @trim_qewas’s flat-budget theorem from 18182 §3.5)

Code + per-cell data (self-contained): arc-whitebox-replication/analytic_vs_sampling at master · jamesrahenry/arc-whitebox-replication · GitHub