Minimum-eigenvalue spectral pruning
Implementation & benchmark of arXiv:2608.27024 — Discrete eigenvalue optimization from entropic smoothing and first-order methods
Source paper: Discrete eigenvalue optimization from entropic smoothing and first-order methods arXiv:2608.27024 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace magnitude-based channel or expert pruning with a subset-selection objective that maximizes the weakest direction in the candidates' activation span. Relax the binary mask to continuous gates, optimize an entropic soft minimum eigenvalue, and round the gates to retain a fixed number of channels or experts. This should preserve diverse representations and reduce redundant feature directions.
Formulas
Mathematical statement
The paper considers maximizing the minimum eigenvalue of M(x)=C+sum_{i=1}^m x_i m_i m_i^T over binary x in X subset of {0,1}^m, where C is an n-by-n positive-semidefinite matrix, m_i in R^n is candidate i's vector, and lambda_min denotes the smallest eigenvalue. For neural pruning, collect a calibration matrix A in R^{B-by-d} for a layer with d candidate channels over B examples; define m_i=A_{:,i}/(||A_{:,i}||_2+epsilon), so n=B, and use C=delta I_B with delta>0. For differentiable optimization, use the entropic spectral soft minimum phi_tau(M)=-tau log tr(exp(-M/tau)), where tau>0 is temperature and exp is the matrix exponential. Its gradient with respect to M is S_tau(M)=exp(-M/tau)/tr(exp(-M/tau)), so for M(x)=C+sum_i x_i m_i m_i^T, partial phi_tau/partial x_i=m_i^T S_tau(M)m_i. Enforce a cardinality budget sum_i x_i=k and 0<=x_i<=1. Add binary-pushing negative entropy H(x)=-sum_i[x_i log x_i+(1-x_i)log(1-x_i)], and maximize F(x)=phi_tau(M(x))-mu H(x), where mu>=0. The gradient is g_i=m_i^T S_tau(M)m_i-mu log((1-x_i)/x_i). As tau approaches zero, phi_tau approaches lambda_min(M) up to a temperature-dependent log-trace error; the ridge delta prevents singular numerical matrices.
Implementation notes
Integrate this as a post-training structured-pruning procedure for a linear, convolutional, or MLP layer; the selected objects are output channels, input channels, attention heads, or MoE experts. First run B calibration examples through the frozen model and record each candidate's activation signature. For channel i, set m_i=A[:,i]/(||A[:,i]||_2+epsilon), and initialize C=delta I_B. Initialize x_i=k/d, choose tau after normalizing the signatures, and repeat 50-200 projected ascent steps. At each step form M=C+sum_i x_i m_i m_i^T, compute S=expm(-M/tau)/trace(expm(-M/tau)), compute q_i=m_i^T S m_i, compute the entropy contribution h_i=log((1-x_i+epsilon)/(x_i+epsilon)), set g_i=q_i-mu h_i, update y=x+eta g, and project y onto the capped simplex {z:0<=z_i<=1, sum_i z_i=k} with sorting-based water filling. Anneal tau and optionally mu, then retain the k largest x_i and fine-tune. The paper-derived quantities are the affine Gram matrix and soft minimum-eigenvalue gradient; estimate the spectrum with Lanczos for diagnostics. Test on CIFAR-10 ResNet-20 or a small Transformer, pruning 50% of channels against magnitude, random, and norm-greedy baselines at equal fine-tuning FLOPs. Measure accuracy, activation Gram minimum eigenvalue, and redundancy. Success means equal or better accuracy at fixed sparsity and a larger post-pruning minimum eigenvalue. For MoE, compare against load-balanced top-k routing at identical active-expert FLOPs and measure perplexity.
Verification
Stage 1 · Toy mechanism gate: Failed ✗
Stage 2 · Mechanism transferred to benchmark: Not tested
Stage 2 · Practical benchmark result: Not run
Methodology: Toy-system gate first; the benchmark stage runs only after a pass. How verification works
Stage 1 — Mechanism check agent confidence 9/10
Built a differentiable minimum-eigenvalue spectral pruning MVP with stable eigendecomposition, capped-simplex projected ascent, entropy regularization, top-k rounding, and baselines. The finite-difference gradient error was 1.36e-11 and the soft-minimum log-trace bound passed. Spectral selection substantially improved the nontrivial selected-span minimum eigenvalue, but reconstruction MSE was slightly worse than magnitude pruning, so the claimed end-to-end win was not observed.
- Agent confidence
- 9/10
- Baseline
- Magnitude: MSE 9.4861 +/- 5.9793; selected Gram minimum eigenvalue 0.02316 +/- 0.01967. Random: MSE 9.6321 +/- 5.8274; eigenvalue 0.03046 +/- 0.03394. Norm-greedy: MSE 9.6536 +/- 5.8651; eigenvalue 0.06676 +/- 0.04424.
- Idea
- Spectral: MSE 9.5698 +/- 5.5495; selected Gram minimum eigenvalue 0.08314 +/- 0.05444. Gradient maximum absolute error 1.36e-11; soft-minimum bound passed.
Limitations: Only a small synthetic linear calibration/reconstruction benchmark was tested; no CIFAR/ResNet, convolutional or attention-layer pruning, fine-tuning, wall-clock/FLOP measurements, or large-scale model evaluation was performed.
How to run: python3 spectral_pruning_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- report.md 1.4 KB View
- results.json 1.0 KB View Raw
- spectral_pruning_experiment.py 6.4 KB View Raw