Parity-block curvature preconditioner
Implementation & benchmark of arXiv:2609.01857 — The Hessian of Planar Central Configurations in Pair Space: Decomposition, Morse Index and Symmetry Reduction
Source paper: The Hessian of Planar Central Configurations in Pair Space: Decomposition, Morse Index and Symmetry Reduction arXiv:2609.01857 ⓘ · analyzed Sep 3, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
When a model, dataset, or parameterization has an involutive symmetry, transform gradients and curvature into symmetry sectors before applying adaptive updates. The Hessian and optimizer can then be handled as independent even and odd blocks, preventing curvature from one sector from forcing an unnecessarily conservative learning rate in the other.
Formulas
Mathematical statement
For a reflection operator $R$ satisfying $R^2=I$, the paper constructs bases $B_+$ and $B_-$ for the $+1$ and $-1$ eigenspaces and, in the orthogonal case, forms $Q=[B_+\mid B_-]\in O(6)$. Reflection-invariant quadratic forms become block diagonal in these coordinates: $Q^THQ=\operatorname{diag}(H_+,H_-)$, so generalized eigenvalues and degeneracies can be computed independently. The paper also emphasizes that a symmetry action may be non-orthogonal, in which case $Q$ is invertible but $Q^TQ\ne I$; the correct reduction is by congruence using metric matrices rather than by assuming Euclidean orthogonality. For neural networks, let $S\in\mathbb R^{d\times d}$ be an involutive parameter transformation preserving the loss locally or exactly. Define projectors $P_+=(I+S)/2$ and $P_-=(I-S)/2$. If $S^THS=H$, then $P_+HP_-=0$ and the local quadratic model separates into even and odd sectors. Sector-specific preconditioners $A_+$ and $A_-$ can use separate step sizes with stability ceilings $\eta_\pm<2/\lambda_{\max}(A_\pm^{1/2}H_\pm A_\pm^{1/2})$.
Implementation notes
Choose a concrete involution with a known network meaning. The cheapest test is a two-branch MLP or CNN with branch-swap symmetry: $S$ swaps the branch parameters, so $P_+$ gives shared or even parameters and $P_-$ gives antisymmetric or odd parameters. A second test is a transformer with paired attention heads, where $S$ exchanges each head pair. At every step compute $g_+=P_+g$ and $g_-=P_-g$, maintain separate RMS or diagonal curvature estimates $A_+$ and $A_-$, and update $\theta\leftarrow\theta-\eta_+A_+g_+-\eta_-A_-g_-$. Estimate each sector's top generalized curvature by one Hutchinson or Lanczos pass restricted to that sector. If the implemented symmetry is only a coordinate relabeling with a non-Euclidean parameter metric, construct a basis $Q=[B_+|B_-]$ by eigendecomposition of $S$ and solve $H_\pm v=\lambda G_\pm v$ with $G_\pm=B_\pm^TB_\pm$ rather than treating the basis as orthonormal. The paper supplies the parity-sector and non-orthogonal congruence reduction; curvature estimates and optimizer states are empirical. First experiment: train a two-branch CIFAR-10 CNN with tied initialization and branch-swap-invariant loss, comparing AdamW against sector-wise AdamW. Log $\lambda_{\max,+}$, $\lambda_{\max,-}$, gradient norms, and validation accuracy. The prediction is that the measured stable step-size boundary for independently scaled sectors follows $\eta_{\pm,c}=2/\lambda_{\max,\pm}$ within 20%, and that one sector can tolerate at least a 1.5-fold different learning rate whenever its top curvature differs by that factor.
Verification
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 a readable parity-sector quadratic preconditioner experiment with orthogonal and non-orthogonal involutions. The orthogonal check observed numerical block decoupling (cross-block norm 7.4e-15), a 15x sector curvature ratio, and stability boundaries matching 2/lambda_max: 0.99x was stable while 1.01x was unstable in both sectors. Sector-specific steps reduced loss substantially faster than the conservative global step, but this is a toy quadratic result rather than evidence from a neural-network dataset.
- Agent confidence
- 8/10
- Baseline
- Global step eta=0.04; loss at step 10=1.2138; final loss after 30 steps=0.1160.
- How to run
- /home/maxwelhelp/main/bin/python3 parity_experiment.py
- Idea
- Sector steps eta_plus=0.6 and eta_minus=0.04; loss at step 10=0.3001; final loss after 30 steps=3.99e-05. Non-orthogonal generalized-pencil residuals were 4.8e-16 and 2.5e-14, while naive Euclidean sector eigenvalues differed from generalized-pencil values.
Limitations: No CIFAR-10 CNN, AdamW implementation, stochastic training, wall-clock/FLOP measurement, validation accuracy, or learned curvature estimator was tested. The observed win depends on an exactly known reflection symmetry and a deliberately constructed 15x sector-curvature mismatch.
Stage 2 — Benchmark vs. tuned baseline (arXiv:2609.01857)
Significantly WORSE than the tuned baseline (+51.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- two_branch_mlp
- Paired seeds
- 8
- Baseline mean
- 0.2395
- Idea mean
- 0.3616
- Effect (Δ)
- 0.1221 (+51.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±29.3%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered tabular track was tested. The experiment used a small two-branch MLP and did not test vision, sequence, dynamics, wall-clock speed, FLOPs, or empirical sector-specific stability-boundary scans.
How to run: python3 official_stage2.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
The parity-block AdamW intervention was tested on the registered tabular track with the same branch-swap two-branch MLP as the baseline. It performed significantly worse: baseline test MSE was 0.2395 versus 0.3616 for the idea, with paired delta +0.1221 and p=0.0081. The mechanism signature observed separated even and odd gradient energy, but no task-metric win occurred.
- Baseline
- Best lr=0.01; full 8-seed test MSE mean 0.2395111453.
- Delta mean
- +0.1221111957; p=0.0081; idea wins 0/8.
- Idea
- Best tested lr=0.003; full 8-seed test MSE mean 0.3616223410.
Benchmark result
Significantly WORSE than the tuned baseline (+51.0%, p=0.0081, wins 0 / 8 paired seeds).
- Benchmark
- Tabular regression (Friedman)
- Model
- two_branch_mlp
- Paired seeds
- 8
- Baseline mean
- 0.2395
- Idea mean
- 0.3616
- Effect (Δ)
- 0.1221 (+51.0%; negative = idea better)
- Wins
- 0 / 8 paired seeds
- p-value
- 0.0081 (permutation test, 20 000 shuffles)
- Smallest detectable effect
- ±29.3%
- Mechanism
- Confirmed ✓
- Practical effect
- Hurts
- Baseline tuning
- swept over 3 configs
Only the registered tabular track was tested. The experiment used a small two-branch MLP and did not test vision, sequence, dynamics, wall-clock speed, FLOPs, or empirical sector-specific stability-boundary scans.
How to run: python3 official_stage2.py
Artifacts
- bench_report.json 3.5 KB View Raw
- official_stage2.py 5.1 KB View Raw
- parity_experiment.py 4.2 KB View Raw
- report.md 1.4 KB View
- report_bench_2026-09-03T122941.md 4.0 KB View
- stage2_bench.py 7.4 KB View Raw