Star-Delta Hub Elimination
Implementation & benchmark of arXiv:2606.31596 — The Invariant Measure of Multiscale Markov Chains via Fast Arborescence Factorization
Source paper: The Invariant Measure of Multiscale Markov Chains via Fast Arborescence Factorization arXiv:2606.31596 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Remove a latent relay or hub token from an attention or graph layer and replace its two-hop influence by direct effective edges between retained tokens. The correction is a normalized rank-one update, so it can preserve hub-mediated communication while reducing the number of stored and processed states.
Formulas
Mathematical statement
For a star graph with hub s and surviving vertices \(\widehat V=V\setminus\{s\}\), the generalized star-delta reduction replaces the hub by effective weights \(\widehat r(x,x')=r(x,x')+r^*(x,x')\). Here \(r(x,s)\) is the incoming weight from surviving vertex x to hub s, \(r(s,x')\) is the outgoing hub weight, and \(r^{\mathrm{out}}(s)=\sum_{z\in\widehat V}r(s,z)\) is the total outgoing hub weight. The indirect contribution \(r^*(x,x')=r(x,s)r(s,x')/r^{\mathrm{out}}(s)\) is the normalized product of entering and leaving weights. In matrix form this is a rank-one outer product. For neural attention, A is a nonnegative directed attention or routing matrix, s is a relay token, and the retained-token matrix is \(A'_{xx'}=A_{xx'}+A_{xs}A_{sx'}/\sum_{z\neq s}A_{sz}\), followed by row normalization.
Implementation notes
Apply the reduction to a Transformer or graph neural network containing one or more designated relay tokens. At the attention stage, compute nonnegative weights A from the existing attention probabilities or from softplus routing logits. Choose a hub index s, retain all other tokens, and compute d_s=sum_{z != s} A[s,z]. Form the rank-one correction B=outer(A[keep,s], A[s,keep])/(d_s+epsilon), set A_prime=A[keep,keep]+B, and renormalize every surviving row. Then compute attention logits as Q_keep K_keep^T/sqrt(d) plus log(A_prime+epsilon), and multiply the resulting normalized attention by V_keep. For a strict Markov interpretation, build A from row-stochastic transition weights and apply row normalization after elimination; for numerical stability clamp d_s and epsilon to 1e-8. The mathematical quantities supplied by the paper are the normalized hub correction and its rank-one factorization; no spectral or probabilistic constant needs to be estimated. Start with a 6-layer small Transformer on Long Range Arena or a mesh-graph benchmark, using 1-4 relay tokens per layer. Compare full attention, naive relay-token deletion, and a learned low-rank approximation at matched training steps. Measure peak activation memory, attention FLOPs, wall-clock latency, validation accuracy, and long-context perplexity. The target is near-linear memory savings in the number of removed relay tokens, substantially smaller accuracy loss than deletion, and at least 1.3x inference speedup at iso-accuracy.
Verification
Stage 1 · Toy mechanism gate: Passed ✓
Stage 2 · Mechanism transferred to benchmark: Confirmed ✓
Stage 2 · Practical benchmark result: Inconclusive
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
Built and numerically verified star-delta hub elimination using a rank-one correction and row normalization. The mechanism predictions were confirmed: exact rank-one factorization and output preservation at machine precision, indirect mass scaling exactly with p (slope 1.0, R²=1.0), and correction Frobenius norm scaling linearly with p (R²=1.0). Naive deletion error increased with p, while the proposed reduction stayed near zero; however, one removed hub gives only a 1.6% theoretical attention-entry reduction and no measured NumPy wall-clock win.
- Agent confidence
- 9/10
- Baseline
- Naive hub deletion output RMSE increased from 0.00627 at p=0.02 to 0.26635 at p=0.85; at p=0.50 RMSE was 0.15667.
- Idea
- Star-delta reduction output RMSE was 6.4e-17 to 8.9e-17 across the sweep; indirect-mass slope was 1.000 with R²=1.000; correction-norm linear-fit R² was 1.000. Theoretical n=129 attention speedup was 1.0157x, while the noisy NumPy measured ratio was 0.766x.
Limitations: Only a controlled linear Markov/attention-value toy setting was tested. No trained Transformer, task accuracy, softmax-logit attention experiment, GPU memory measurement, multi-hub case, or realistic end-to-end latency benchmark was evaluated.
How to run: python3 experiment.py
Stage 2 — Benchmark vs. tuned baseline (arXiv:2606.31596)
Ahead of the baseline (−2.8%) but not statistically significant (p=0.3087, wins 5 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.4928
- Idea mean
- 0.4791
- Effect (Δ)
- -0.0136 (−2.8%; negative = idea better)
- Wins
- 5 / 8 paired seeds
- p-value
- 0.3087 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±5.7%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Tested only the built-in sequence track with one removed token, 400 training samples, 200 test samples, 10 epochs, and the tiny two-layer Transformer. Multi-hub elimination, long-context perplexity, peak activation memory, and reliable end-to-end latency were not tested.
How to run: python3 bench_star_delta.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
Star-delta hub elimination was implemented and tested on the structurally matching sequence forecasting track. The mechanism signature was confirmed to machine precision, but the modest MSE reduction was not statistically significant, so the system verdict is no significant win.
- Baseline mean mse
- 0.4927588701248169
- Delta mean
- -0.013626139611005783
- Idea mean mse
- 0.4791327305138111
- P value
- 0.3087
Benchmark result
Ahead of the baseline (−2.8%) but not statistically significant (p=0.3087, wins 5 / 8 paired seeds).
- Benchmark
- Sequence modeling
- Model
- transformer_tiny
- Paired seeds
- 8
- Baseline mean
- 0.4928
- Idea mean
- 0.4791
- Effect (Δ)
- -0.0136 (−2.8%; negative = idea better)
- Wins
- 5 / 8 paired seeds
- p-value
- 0.3087 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±5.7%
- Mechanism
- Confirmed ✓
- Practical effect
- Inconclusive
- Baseline tuning
- swept over 3 configs
Tested only the built-in sequence track with one removed token, 400 training samples, 200 test samples, 10 epochs, and the tiny two-layer Transformer. Multi-hub elimination, long-context perplexity, peak activation memory, and reliable end-to-end latency were not tested.
How to run: python3 bench_star_delta.py
Artifacts
- bench_report.json 3.0 KB View Raw
- bench_star_delta.py 4.9 KB View Raw
- experiment.py 5.1 KB View Raw
- report.md 1.5 KB View
- report_bench_2026-08-30T143421.md 4.3 KB View
- results.json 2.8 KB View Raw