Cyclic Lie-Bracket Residual Block
Implementation & benchmark of arXiv:2608.22514 — Diffusion limits of cyclic finite-velocity random motions along vector fields
Source paper: Diffusion limits of cyclic finite-velocity random motions along vector fields arXiv:2608.22514 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace one deterministic residual update with a short cyclic composition of learned vector fields evaluated for randomized, short run times. Because finite compositions of noncommuting flows generate directional-derivative and Lie-bracket terms, changing the cycle order gives the network an explicit, low-cost way to learn drift directions that are unavailable from the individual vector fields alone.
Formulas
Mathematical statement
The paper's finite-velocity construction is the telegraph process $X_t=x+v\xi\int_0^t(-1)^{N_s}\,\mathrm{d}s$, where $v$ is speed, $\xi\in\{-1,1\}$ is the initial direction, and $N_t$ is a Poisson switching process. Its multidimensional extension alternates vector fields $V_1,\ldots,V_p$ over short random run times under Kac-type scaling: run times shrink while speeds increase so that cumulative fluctuations remain finite. For two smooth vector fields, composing a short flow of $V_i$ followed by one of $V_j$ gives the second-order directional-derivative term $DV_j[V_i]$. Reversing the order changes this term by the Lie bracket $[V_i,V_j]=DV_j[V_i]-DV_i[V_j]$. If the first-order vector-field contributions are centered, the first-order drift cancels, while the accumulated second-order terms and run-time fluctuations remain observable in the diffusion-scale limit. The neural block directly implements the sequential composition; its order-dependent difference is an empirical approximation to the bracket-generated drift.
Implementation notes
1. Integration point: replace a standard residual MLP update $x\leftarrow x+F(x)$ in a small Transformer, ViT, or continuous-depth SSM with a cyclic block containing $p=2$ or $p=4$ vector-field modules $V_r:\mathbb{R}^d\to\mathbb{R}^d$. Each $V_r$ can be a bottleneck MLP with LayerNorm, linear down-projection, GELU, and linear up-projection. Enforce first-order centering by defining $V_p(x)=-\sum_{r<p}V_r(x)$, or by subtracting the average field from every field. 2. Pseudocode: sample independent $z_r\sim\mathcal N(0,1)$, set $a_r=z_r$, choose a permutation $\pi$ of $\{1,\ldots,p\}$, initialize y=x, and execute for r in pi: y = y + sqrt(epsilon) * a[r] * V[r](y). Return x + alpha*(y-x), where $\alpha$ is a learned residual gate initialized to $0.1$. To test order effects, use both $1\to2$ and $2\to1$ permutations with shared parameters. 3. The paper-derived computation is the sequential composition and its cross term $DV_j[V_i]$; Jacobian-vector products from autodiff can verify $[V_i,V_j]$ without forming full Jacobians. Estimate the empirical diffusion covariance with $\widehat C=T^{-1}\sum_t(\Delta x_t-\overline{\Delta x})(\Delta x_t-\overline{\Delta x})^\top/\varepsilon$. 4. First experiment: train a parameter- and FLOP-matched small ViT on CIFAR-10, comparing a standard residual MLP, the cyclic block with fixed order, randomized order, and commuting/linear vector fields. Use identical optimizer and schedule. Record validation accuracy, corruption robustness, gradient-norm variance, training loss at equal FLOPs, and $\|f_{1\to2}(x)-f_{2\to1}(x)\|$. The hypothesis is that a nonzero bracket signal improves accuracy or noise robustness and that random run times stabilize gradients; forcing the fields to commute should eliminate the order-dependent benefit.
Verification
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 a cyclic two-field residual composition. The mechanism manifested clearly: order-gap scaling with epsilon was 0.9995 versus predicted 1.0, joint field scaling was 1.9997 versus predicted 2.0, normalized gap matched the bracket norm (0.61108 vs 0.61098), and the commuting control was essentially zero (5.0e-17). In the secondary toy regression, cyclic training did not beat the baseline: test MSE was 0.00315 versus 0.000224 and it used 2274 versus 1170 parameters and about twice the runtime, so no practical win was established.
- Agent confidence
- 9/10
- Baseline
- Toy regression: train MSE 0.0002188, test MSE 0.0002243, loss at step 50 0.07765, loss at step 350 0.0002203, 1170 parameters, 1.70 s.
- Idea
- Toy regression: train MSE 0.002853, test MSE 0.003149, loss at step 50 0.09836, loss at step 350 0.002886, 2274 parameters, 3.44 s, learned order signal 4.74e-05. Math: epsilon exponent 0.9995, field-scale exponent 1.9997, normalized gap 0.61108 versus bracket prediction 0.61098, commuting gap 5.02e-17.
Limitations: The learning comparison is a tiny synthetic regression, not CIFAR-10/ViT or a parameter/FLOP-matched benchmark. The cyclic model uses stochastic scalar run times and was not tuned across schedules; robustness, gradient variance, larger architectures, and diffusion-limit behavior were not tested. The secondary model has more parameters than the baseline, so its poorer result is not a controlled efficiency comparison.
How to run: python3 run_experiment.py && /home/maxwelhelp/main/bin/python3 train_toy.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.22514)
Beats the tuned baseline by 52.8% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0004
- Idea mean
- 0
- Effect (Δ)
- -0.0002 (−52.8%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±23.8%
- Mechanism
- Not confirmed ✗
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the built-in dynamics track was tested, with 1000 training samples, 300 test samples, 12 epochs, and a small rnn_small model. No vision, sequence, corruption robustness, FLOP timing, gradient-variance analysis, commuting-field control, or larger-scale dynamics evaluation was run. The measured mechanism exponent did not match the predicted quadratic scaling.
How to run: python3 bench_cyclic.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Worked ✓
Agent confidence: 7/10
Implemented the cyclic Lie-bracket residual block on the structurally matched dynamics track using the same GRU encoder and paired 8-seed protocol. The tuned cyclic model achieved mean test MSE 0.0001866451 versus 0.0003950244 for the tuned baseline, winning on all 8 seeds; paired delta was -0.0002083793 with permutation p=0.0081. The trained-model order gap was nonzero, but its observed small-step exponent was 2.9885 rather than the predicted 2.0, so the mechanism prediction was not quantitatively confirmed.
- Baseline
- dynamics/rnn_small, 12 epochs, lr=0.006: mean test MSE 0.0003950244, std 0.0001292329; 8 per-seed results [0.00040816, 0.00056587, 0.00031052, 0.00062471, 0.00027074, 0.00039440, 0.00022947, 0.00035633]. Baseline lr sweep [0.001, 0.003, 0.006] means [0.00136616, 0.00073085, 0.00039502].
- Idea
- cyclic centered two-field GRU residual, 12 epochs, lr=0.006, step=0.1: mean test MSE 0.0001866451, std 0.0000630393; all 8 paired seeds improved. Idea step sweep [0.1, 0.15, 0.22] means [0.00018665, 0.00019070, 0.00020175]. Paired delta=-0.0002083793, permutation p=0.0081, 8/8 wins.
- Mechanism signature
- Trained-model order gap at step 0.1 was 0.0122054. Gaps at steps [0.01, 0.02, 0.04] were [1.3433e-05, 1.0711e-04, 8.4608e-04], giving observed exponent 2.9885 versus predicted 2.0; confirmed=false.
Benchmark result
Beats the tuned baseline by 52.8% (p=0.0081, wins 8 / 8 paired seeds; metric: lower is better).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small
- Paired seeds
- 8
- Baseline mean
- 0.0004
- Idea mean
- 0
- Effect (Δ)
- -0.0002 (−52.8%; negative = idea better)
- Wins
- 8 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±23.8%
- Mechanism
- Not confirmed ✗
- Practical effect
- Helps
- Baseline tuning
- swept over 3 configs
Only the built-in dynamics track was tested, with 1000 training samples, 300 test samples, 12 epochs, and a small rnn_small model. No vision, sequence, corruption robustness, FLOP timing, gradient-variance analysis, commuting-field control, or larger-scale dynamics evaluation was run. The measured mechanism exponent did not match the predicted quadratic scaling.
How to run: python3 bench_cyclic.py
Artifacts
- bench_cyclic.py 5.5 KB View Raw
- bench_report.json 2.8 KB View Raw
- report.md 1.8 KB View
- report_bench_2026-08-31T222908.md 5.2 KB View
- results.json 1.1 KB View Raw
- run_experiment.py 4.8 KB View Raw
- toy_results.json 0.5 KB View Raw
- train_toy.py 3.9 KB View Raw