Spectral-Pole-Tuned Decentralized Optimizer
Implementation & benchmark of arXiv:2607.25463 — Optimal Parameter Design for DIGing on Minimizing Unweighted Sum of Squares
Source paper: Optimal Parameter Design for DIGing on Minimizing Unweighted Sum of Squares arXiv:2607.25463 ⓘ · analyzed Aug 29, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Choose the consensus gain and gradient-tracking gain in decentralized training from the communication Laplacian spectrum rather than tuning them independently. The gains minimize the worst asymptotic pole radius for the paper's exact quadratic model, providing a principled initialization and a conservative stability safeguard for neural-network optimization.
Formulas
Mathematical statement
Let \(\mathcal L\in\mathbb R^{N\times N}\) be a symmetric Laplacian of a connected communication graph, with eigenvalues \(0=\lambda_1<\lambda_2\leq\cdots\leq\lambda_N\), and define \(\kappa=\lambda_N/\lambda_2\). The modified DIGing dynamics are \(x_{k+1}=(I-\varepsilon\mathcal L)x_k-\alpha y_k\) and \(y_{k+1}=(I-\varepsilon\mathcal L)y_k+x_{k+1}-x_k\). Transforming with \(\mathcal L=U\operatorname{diag}(0,\lambda_2,\ldots,\lambda_N)U^T\) decouples every graph-frequency mode. The paper gives the optimal worst-case convergence factor \(\gamma^*=\frac{4-\kappa^2+\kappa\sqrt{\kappa^2+8\kappa}}{2(\kappa^2+2\kappa+2)}\), attained by \(\alpha^*=1-\gamma^*\) and \(\varepsilon^*=2(1-\gamma^*)/\lambda_2\). The feasible stability interval is \(\frac{2}{\lambda_2}(1-\gamma)\leq\varepsilon\leq\frac{1}{\lambda_N}(1+\gamma-\sqrt{1-\gamma^2})\), together with \(\alpha\in[1-\gamma,1+\gamma]\). Here \(\alpha\) is the tracker-step gain, \(\varepsilon\) is the graph-mixing gain, \(\lambda_2\) and \(\lambda_N\) are the extremal nonzero Laplacian eigenvalues, and \(\gamma\) is the worst pole modulus. Neural losses are not quadratic, so these are a gain proposal and safeguard rather than a global convergence theorem.
Implementation notes
Integrate this into the communication and optimizer step of decentralized neural-network training. Each worker stores parameters \(\theta_i\), a gradient tracker \(s_i\), and the current minibatch gradient \(g_i\). Build the symmetric communication Laplacian \(\mathcal L=D-W\), where \(W_{ij}\) is the edge weight and \(D_{ii}=\sum_jW_{ij}\). Estimate \(\lambda_N\) with 10-20 power iterations on \(\mathcal L\). Estimate \(\lambda_2\) by power iteration after projecting each vector orthogonally to the all-ones vector, or use a small Lanczos eigensolver on the graph matrix. Set \(\kappa=\lambda_N/\max(\lambda_2,10^{-6})\), compute \(\gamma^*\), \(\alpha^*\), and \(\varepsilon^*\) from the displayed equations, and optionally clip \(\varepsilon\) to the displayed feasible interval using a target \(\gamma\geq\gamma^*\). Initialize \(s_i^0=g_i^0\). At each step execute: theta_new = theta - eps * sum_j L[i,j] * theta_j - alpha * s; compute the new minibatch gradient g_new; then execute s_new = s - eps * sum_j L[i,j] * s_j + g_new - g. Use the plain tracker first; only afterward test Adam-style diagonal preconditioning of \(s_i\). The formulas and eigenvalue estimates are computed analytically/numerically from the graph; actual neural-loss stability is measured empirically. Test eight workers training a two-layer MLP on MNIST or CIFAR-10 under ring, grid, and random geometric graphs. Compare standard decentralized gradient tracking, the proposed spectral gains, and a grid-searched best gain pair at equal communication rounds. Record loss versus rounds and transmitted bytes, parameter disagreement \(N^{-1}\sum_i\|\theta_i-\bar\theta\|^2\), tracker disagreement, gradient norms, and divergence frequency. Success is faster loss descent or higher accuracy at equal communication cost without increased instability, especially on poorly connected graphs with large \(\lambda_N/\lambda_2\).
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 10/10
Implemented the exact modal dynamics, spectral gain formulas, stability sweeps, and a small quadratic decentralized-tracking comparison in spectral_pole_experiment.py. The paper-consistent convention κ=λ₂/λ_N reproduced the predicted worst pole radius to numerical precision and showed endpoint equalization, but the authoritative idea defines κ=λ_N/λ₂ while retaining the same formula; under that literal definition, measured radii were 2.63, 9.13, 20.09, 63.11, and 209.48 for condition ratios 2, 5, 10, 30, and 100, respectively, causing divergence. Therefore the proposed idea does not work as stated, despite the corrected paper convention exhibiting the claimed mechanism.
- Agent confidence
- 10/10
- Baseline
- Quadratic toy baseline (alpha=0.1, epsilon=0.1): final loss 0.2039, finite after 100 steps.
- Idea
- Literal spectral gains on an 8-node path: final loss exceeded 1.21e101 and the modal worst radius was 9.13 for graph condition ratio 25.27; paper-consistent gains gave radius 0.9711 for ratio 25.27 and final loss 0.2395.
Limitations: No neural-network, MNIST/CIFAR, communication-byte, GPU, or gain-grid experiment was run because the literal formula failed the required toy mechanism check. The implementation uses exact eigenspectrum values in the toy graph rather than noisy distributed eigenvalue estimation.
How to run: python3 spectral_pole_experiment.py
Verdict computed by deterministic test code from paired-seed statistics — not by the language model.
Artifacts
- report.md 1.6 KB View
- results.json 5.0 KB View Raw
- spectral_pole_experiment.py 3.9 KB View Raw