Skip to content

Phase 4k: richer geometry (full Hessian spectrum + mode connectivity) - #7

Merged
dkjo8 merged 1 commit into
mainfrom
phase-4k-richer-geometry
Jul 29, 2026
Merged

Phase 4k: richer geometry (full Hessian spectrum + mode connectivity)#7
dkjo8 merged 1 commit into
mainfrom
phase-4k-richer-geometry

Conversation

@dkjo8

@dkjo8 dkjo8 commented Jul 28, 2026

Copy link
Copy Markdown
Owner

What & why

The Phase-4j headline caveat — on graph, geometry is redundant with softmax — rested on a thin curvature description (only λmax + tr(H)). This phase asks whether that redundancy is a task property or an artifact of feature poverty, by adding the two richer signals the paper names as future work and re-running the existing 4j complementarity test on all four (task, reasoner) cells.

  • geometry/spectrum.py — full d×d Hessian spectrum (exact eigh, since d is small): lmin, negative-eigenvalue fraction, effective rank (participation ratio), log-det — each mean-over-restarts and at the best restart.
  • geometry/connectivity.py — mode connectivity: energy barriers along the straight-line path from the best restart to the others (barrier mean/max, connected fraction).
  • New JAX (batched_spectrum, batched_path_energy) lives in curvature.py to keep invariant 1; the two feature modules are pure NumPy.
  • Opt-in [eval] richer_geometry; features appended after the base 14 so every prior result reproduces byte-identically. Aggregation is feature-set-aware (feature_set_of; canonical tables/figures default feature_set="base"), so richer never pollutes T1/T2/T4 or the pooled figures.
  • T5 table (base vs richer per cell).

Verdict — redundancy confirmed as a task property

task reasoner features ΔAURC geom adds over softmax seeds add
arithmetic IRED base +0.007 4/5
arithmetic IRED richer +0.007 4/5
graph IRED base +0.001 0/5
graph IRED richer −0.002 0/5

Richer geometry helps nowhere: the crux graph-IRED conditional signal stays −0.002 (0/5), it's a wash where geometry already wins (arith-IRED, unchanged), and slightly worse on the fixed-bowl basin-center cells (extra features add noise). So a well-calibrated softmax genuinely captures what even the richest descent geometry would add on graph. A decisive negative result that cements the headline caveat and closes the paper's stated "richer geometry" open question.

Tests

109 tests green (offline, CPU-only), ruff clean. New test_spectrum.py / test_connectivity.py verify exact spectra on a known quadratic and barriers on a double well; extended feature/evaluate/aggregate suites cover the richer path + feature-set aggregation. Paper rebuilds via tectonic.

🤖 Generated with Claude Code

Tests whether the Phase-4j "graph tie = redundancy" verdict is real or an
artifact of thin curvature features (only lambda_max + trace). Adds two opt-in
feature groups behind [eval] richer_geometry and re-runs the 4j complementarity
test on all four (task, reasoner) cells.

- geometry/spectrum.py: full d×d Hessian spectrum (exact eigh) -> lmin,
  negative-eigenvalue fraction, effective rank, log-det (mean + at best restart).
- geometry/connectivity.py: energy barriers along the path from the best restart
  to the others (mode connectivity) -> barrier mean/max, connected fraction.
- curvature.py: batched_spectrum + batched_path_energy (all new JAX stays in the
  core, invariant 1; the two feature modules are pure NumPy).
- evaluate/config/features: richer appended AFTER the base 14 (order-preserving),
  metrics["feature_set"], ablation guards empty groups.
- aggregate: feature_set_of + canonical tables/figures default feature_set="base"
  so richer never pollutes T1/T2/T4 or the pooled figures (e.g. S1's K=12 point).
- make_tables T5 (base vs richer per cell); make_figures excludes richer rows.

Verdict: richer geometry helps NOWHERE. Crux graph-IRED conditional signal over
softmax stays -0.002 (0/5); a wash where geometry already wins (arith-IRED
+0.007, 4/5, unchanged) and slightly worse on the fixed-bowl basin-center cells.
So the graph redundancy is a task property, not thin features. Base rows
reproduce the 4j numbers exactly (purely additive). 109 tests green, ruff clean.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@cursor

cursor Bot commented Jul 28, 2026

Copy link
Copy Markdown

Bugbot is not enabled for your account, so this pull request was not reviewed.

Enable Bugbot in the Cursor dashboard to get automatic reviews on future PRs.

@dkjo8
dkjo8 merged commit 67dc7a2 into main Jul 29, 2026
1 check passed
@dkjo8
dkjo8 deleted the phase-4k-richer-geometry branch July 29, 2026 07:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant