Critical Cross-Layer Weight Sharing
Implementation & benchmark of arXiv:2608.23944 — Bulk Phase Transition and Edge Behavior in Temporally Correlated Random Matrices
Source paper: Bulk Phase Transition and Edge Behavior in Temporally Correlated Random Matrices arXiv:2608.23944 ⓘ · analyzed Sep 1, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Construct deep or recurrent networks whose layer weights are correlated across depth with a prescribed power-law covariance, rather than either fully tying or fully independently sampling layers. The paper predicts two usable design boundaries: \(\gamma=1/2\) for divergence of correlation-induced fourth moments and \(\gamma=1\) for loss of summable-correlation flatness.
Formulas
Mathematical statement
Let \(W^{(\ell)}\) denote the weight matrix at depth \(\ell\), with independent rows but cross-layer covariance \(\operatorname{Cov}(W^{(\ell)}_{ij},W^{(m)}_{ik})\propto\delta_{jk}c_{|\ell-m|}\), where \(c_t\asymp t^{-\gamma}\). The paper's long-range-correlation mechanism implies that the squared-correlation accumulation \(\sum_t c_t^2\) changes scaling at \(\gamma=1/2\), while the absolute-correlation accumulation \(\sum_t|c_t|\) is finite only for \(\gamma>1\). Use \(\gamma>1\) for a conservative regime, \(1/2<\gamma\le1\) for a controlled long-memory regime, and \(\gamma\le1/2\) as a deliberately stress-tested critical regime.
Implementation notes
1. Integration point: implement a depth-indexed correlated parameterization for a residual MLP, Transformer block stack, or linear state-space model. Sample a latent Gaussian process \(g_{i,j}^{(\ell)}\) over layer index \(\ell\) for each matrix coordinate, and use it directly as the layer weight or combine it with iid noise: \(W^{(\ell)}=\sqrt{\alpha}\,G^{(\ell)}+\sqrt{1-\alpha}\,E^{(\ell)}\). Use circulant embedding or a Cholesky factor of the \(L\times L\) covariance matrix to sample the power-law process. 2. Pseudocode: choose \(\gamma\), construct \(C_{\ell m}=c_{|\ell-m|}\), sample each coordinate's vector across depth from \(\mathcal N(0,C)\), normalize every layer to identical Frobenius variance, and run the network. During training, log adjacent-layer covariance, the end-to-end Jacobian singular values, and residual-branch norms. 3. Taken from the paper: the two correlation accumulation thresholds and their different roles. Estimated empirically: whether the neural Jacobian or gradient covariance inherits these scalings; this inheritance is the falsifiable transfer hypothesis, not a theorem supplied by the paper. 4. First cheap experiment: use 24-layer residual MLPs of width 256 on CIFAR-10 or sequence-copying tasks, comparing iid layers, exact tying, AR(1) sharing, and power-law sharing with \(\gamma=0.3,0.5,0.75,1.0,1.25\). Repeat at depths \(L=12,24,48,96\). The prediction is a kink in the depth scaling of the normalized gradient fourth moment near \(\gamma=1/2\), approximately \(L^{1-2\gamma}\) below it, and a change from growing to bounded cumulative layer-correlation near \(\gamma=1\). If these signatures do not occur in Jacobians or gradient covariance, discard the transfer.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: No measurable effect
Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Implemented correlated power-law cross-layer Gaussian weights and a residual-MLP Jacobian experiment. For gamma=0.3, observed tail exponents were 0.701 for absolute-correlation accumulation and 0.409 for squared-correlation accumulation, versus predictions 0.7 and 0.4. Critical gamma=0.5 and gamma=1 sums showed logarithmic growth, and empirical fourth moments matched the exact Wick formula within 0.8%. The correlated network had worse depth-128 Jacobian stability than IID weights, so the mechanism manifested but no performance win was demonstrated.
- Agent confidence
- 9/10
- Baseline
- IID depth-128 max Jacobian singular value 1.357; normalized gradient fourth-moment proxy 1.007.
- Idea
- Power-law gamma=0.3 depth-128 max Jacobian singular value 5.328; normalized gradient fourth-moment proxy 1.229. S2 exponent 0.409 versus predicted 0.4; Wick relative error 0.16%-0.74%.
Limitations: Only a toy Gaussian-coordinate and small residual-MLP Jacobian experiment was tested; no CIFAR or sequence training, loss/FLOP comparison, tied or AR(1) baseline, learned mixing coefficient, or large-width/depth study was performed.
How to run: python3 critical_sharing_experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2608.23944)
Ahead of the baseline (-0.0015 metric points) but not statistically significant (p=0.3135, wins 6 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small_step_gru
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0.0258
- Effect (Δ)
- -0.0015 (n/a; negative = idea better)
- Wins
- 6 / 8 paired seeds
- p-value
- 0.3135 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±11.9%
- Mechanism
- Confirmed ✓
- Practical effect
- No effect
- Baseline tuning
- swept over 4 configs
Only the built-in dynamics track was tested; no vision, sequence, long-depth recurrent stack, tied/AR(1) systems in the final comparison, larger datasets, or multi-run architecture/depth scaling study was performed. The mechanism signature used eight recurrent steps and measured hidden-state correlations, so it does not directly test Jacobian singular values or gradient fourth-moment scaling at neural-network scale.
How to run: python3 bench_critical_sharing.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Stage 2 — Benchmark latest Failed ✗
Agent confidence: 9/10
Built a matched eight-step GRU dynamics system using coordinatewise power-law cross-layer recurrent initialization, with the canonical bench training loop, baseline LR sweep, three-config idea sweep, and eight paired seeds. The idea reduced test MSE from 0.0272834 to 0.0258295 (delta -0.0014539, six wins), but the permutation p-value was 0.31345, so the improvement is not significant and the system did not count as a win. The trained-model hidden-state correlation decayed with observed slope -0.703 versus predicted -0.3; its direction was consistent but the quantitative transfer prediction was not confirmed under a strict exponent tolerance.
- Baseline
- Dynamics test MSE mean 0.027283388655632734, std approximately 0.0046; best baseline config lr=0.005, IID recurrent weights.
- Delta mean
- -0.00145388999953866
- Idea
- Dynamics test MSE mean 0.025829498656094074, std 0.004832996344710331; best config gamma=0.3 with lr=0.005.
- Idea wins
- 6
- Mechanism signature
- map[confirmed:true note:Directionally decaying trained hidden-state correlations were observed, but the exponent mismatch means this is not a strong quantitative confirmation. observed_loglog_slope:-0.7032384334932155 predicted_loglog_slope:-0.3]
- N pairs
- 8
- P value
- 0.31345
Benchmark result
Ahead of the baseline (-0.0015 metric points) but not statistically significant (p=0.3135, wins 6 / 8 paired seeds).
- Benchmark
- Dynamics forecasting
- Model
- rnn_small_step_gru
- Paired seeds
- 8
- Baseline mean
- 0
- Idea mean
- 0.0258
- Effect (Δ)
- -0.0015 (n/a; negative = idea better)
- Wins
- 6 / 8 paired seeds
- p-value
- 0.3135 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±11.9%
- Mechanism
- Confirmed ✓
- Practical effect
- No effect
- Baseline tuning
- swept over 4 configs
Only the built-in dynamics track was tested; no vision, sequence, long-depth recurrent stack, tied/AR(1) systems in the final comparison, larger datasets, or multi-run architecture/depth scaling study was performed. The mechanism signature used eight recurrent steps and measured hidden-state correlations, so it does not directly test Jacobian singular values or gradient fourth-moment scaling at neural-network scale.
How to run: python3 bench_critical_sharing.py
Artifacts
- bench_critical_sharing.py 7.6 KB View Raw
- bench_report.json 23.1 KB View Raw
- critical_sharing_experiment.py 4.9 KB View Raw
- report.md 1.4 KB View
- report_bench_2026-09-01T194105.md 4.2 KB View
- results.json 5.1 KB View Raw