△ Mechanism confirmed, baseline not beaten 2026

Gradient-Side Error-Feedback SignMuon

Implementation & benchmark of arXiv:2607.29674 — Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedback

Usefulness8/10
Difficulty5/10
Novelty6/10

Source paper: Sign compression for Muon: SignMuon, MuonSign, and the Limits of Error Feedback arXiv:2607.29674 ⓘ · analyzed Aug 31, 2026

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

Idea description

Compress the matrix gradient or momentum before applying Muon's polar LMO, and maintain an error residual in the uncompressed gradient space. The residual prevents systematic sign quantization bias from accumulating, unlike error feedback applied after the nonlinear polar/sign operation. This is suitable for distributed training because workers communicate one sign bit per matrix entry while the server still applies a matrix-aware Muon step.

Formulas

$$H_t = G_t + R_t,\qquad Q(H_t)_{ij}=s_t\,\operatorname{sign}(H_{t,ij}),\qquad s_t=\frac{1}{mn}\sum_{i=1}^{m}\sum_{j=1}^{n}|H_{t,ij}|,$$
$$R_{t+1}=H_t-Q(H_t),\qquad W_{t+1}=W_t-\eta_t\,\mathcal{L}(Q(H_t)),$$
$$\mathcal{L}(A)=UV^\top\quad\text{for }A=U\Sigma V^\top,\qquad \mathcal{L}(A)\in\arg\max_{\|X\|_2\le 1}\langle A,X\rangle_F,$$
$$\frac{1}{T}\sum_{t=0}^{T-1}\mathbb{E}\bigl\|\nabla F(W_t)\bigr\|_F^2=\mathcal{O}(T^{-1/2}).$$

Mathematical statement

For a parameter matrix W_t, let G_t be the stochastic gradient or momentum matrix and R_t the accumulated compression residual, initialized as R_0=0. The compensated matrix is H_t=G_t+R_t. Define the one-bit compressor Q(H) entrywise by Q(H)_{ij}=s sign(H_{ij}), where s=(1/(mn)) sum_{ij}|H_{ij}| is an optional magnitude scale for an m x n matrix and sign(0)=0. The residual update is R_{t+1}=H_t-Q(H_t). Let L(A)=UV^T be the polar LMO, where A=U Sigma V^T is a thin singular-value decomposition; equivalently, L(A) maximizes the Frobenius inner product with A over matrices whose spectral norm is at most one. The optimizer update is W_{t+1}=W_t-eta_t L(Q(H_t)). The transferable convergence target is the standard smooth nonconvex rate (1/T) sum_t E[||grad F(W_t)||_F^2]=O(T^{-1/2}) under smoothness, bounded stochastic variance, bounded compressor moments, and suitable stepsizes when feedback is applied before the LMO. Applying residual feedback after L(G_t) does not have this universal guarantee.

Implementation notes

Integrate this into the Muon optimizer at the gradient-to-update boundary for every 2-D weight matrix; leave biases, layer-normalization parameters, and 1-D tensors on AdamW or ordinary SGD. On each optimizer step, first form the usual Muon momentum matrix G from the current minibatch gradient. Maintain a full-precision residual R of identical shape locally on each communicating worker or centrally on a parameter server. Use: H = G + R; scale = mean(abs(H)); bits = sign(H); transmit(bits, scale); Hq = scale * bits; R = H - Hq; U,S,Vh = svd(Hq); D = U @ Vh; W -= lr * D. For distributed training, test both local error-feedback followed by sign aggregation and signing the averaged compensated matrix, because these operations do not commute. The mathematical components are the residual recursion, entrywise compressor, and polar LMO; estimate the scale from the current tensor and compute the polar factor with stable SVD or Newton-Schulz iteration. Keep Muon's momentum decay and learning-rate schedule unchanged initially. First run a ResNet-18 on CIFAR-10 and a small Transformer on a language-modeling subset. Compare full Muon, SignMuon without residuals, post-LMO sign compression, and this method at equal communicated bits and equal FLOPs. Log loss and gradient norm versus optimizer steps, final validation accuracy, residual norm ||R||_F/||G||_F, and communication volume. Success means slower residual growth and faster loss descent than sign-only Muon, with accuracy approaching full Muon at one bit per matrix entry.

Verification

Mechanism confirmed, baseline not beaten

Stage 1 · Toy mechanism gate: Passed ✓

Stage 2 · Mechanism transferred to benchmark: Confirmed ✓

Stage 2 · Practical benchmark result: Significantly worse ✗

Methodology: 8 paired seeds, permutation test (p<0.05) against a learning-rate-tuned baseline. How verification works

Stage 1 — Mechanism check agent confidence 8/10

Built experiment.py implementing mean-absolute-scale one-bit compression, pre-LMO residual feedback, polar LMO, and a deterministic quadratic comparison. The mechanism checks passed: telescoping identity error was 1.90e-10, EF average error had tail log-log slope -0.996 versus predicted -1, no-feedback stayed at slope 0, and the EF residual bound ratio never exceeded 1.001. This demonstrates the predicted error-feedback mechanism, although it was not an optimization win in this toy run: EF had lower best loss than sign-only but worse final loss and slower early descent.

Agent confidence
8/10
Baseline
No-residual SignMuon: loss@50=4.9212, loss@200=0.0006067, final=0.0006067; post-LMO quantization final=0.0004893
Idea
Gradient-side EF SignMuon: loss@50=6.8500, loss@200=0.0005507, final=0.0016576, best=0.0002591; exact cumulative-error identity max error=1.90e-10, asymptotic error slope=-0.9959

Limitations: Only a small deterministic fixed-gradient sweep and 3x5 quadratic were tested. No CIFAR-10, Transformer, stochastic gradients, distributed aggregation, communication timing, large-scale memory measurement, hyperparameter tuning, or convergence-rate estimate was performed.

How to run: python3 experiment.py

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

Significantly WORSE than the tuned baseline (+1553.3%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
11.304
Idea mean
186.8938
Effect (Δ)
175.5898 (+1553.3%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±46.4%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular/Friedman regression track was tested. Distributed aggregation variants, communication timing, full-precision Muon comparison, and other built-in tracks were not tested. The baseline sweep covered learning rate and beta=0.9 only; no separate Muon baseline was available in the harness.

How to run: python3 bench_signmuon.py

Verdict computed by deterministic test code from paired-seed statistics — not by the language model.

Stage 2 — Benchmark latest Failed ✗

Agent confidence: 10/10

Gradient-side Error-Feedback SignMuon was implemented and evaluated against a matched Adam baseline on tabular/mlp_tiny. Error feedback reduced the measured cumulative quantization error substantially, but test MSE was significantly worse than baseline, so the idea did not win the benchmark.

Baseline
8-seed test MSE mean 11.3039829731, std 0.8817624131; tuned lr=0.003, beta=0.9
Idea
8-seed test MSE mean 186.8938159943, std 5.4758876435; delta +175.5898330212; permutation p=0.0081; 0/8 wins

Benchmark result

Significantly WORSE than the tuned baseline (+1553.3%, p=0.0081, wins 0 / 8 paired seeds).

Benchmark
Tabular regression (Friedman)
Model
mlp_tiny
Paired seeds
8
Baseline mean
11.304
Idea mean
186.8938
Effect (Δ)
175.5898 (+1553.3%; negative = idea better)
Wins
0 / 8 paired seeds
p-value
0.0081 (permutation test, 20 000 shuffles)
Smallest detectable effect
±46.4%
Mechanism
Confirmed ✓
Practical effect
Hurts
Baseline tuning
swept over 3 configs
Limitations:

Only the structurally matched tabular/Friedman regression track was tested. Distributed aggregation variants, communication timing, full-precision Muon comparison, and other built-in tracks were not tested. The baseline sweep covered learning rate and beta=0.9 only; no separate Muon baseline was available in the harness.

How to run: python3 bench_signmuon.py

Artifacts

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