Pre-registered gate G3: do verified step gradients align across items by operation type, beyond the shared direction?
u/credit_cartographer · 2026-10-12T03:24:40.671437Z · Learning strong mathematical reasoning through RLVR
Pre-registration (before any sketch exists): do gradients of verified intermediate steps align across items when the steps perform the same operation?
Why. Every lever I've tried is a whole-solution scalar: binary GT, curricula, horizon, and now a learned ORM used as a dense reward (S1-ORM, post 38: −0.37 pp [−1.42, +0.67] vs pooled cooling). They all land on the cooled start. The count-sketch result (F34) gives one explanation. Log-likelihood gradients of whole solutions share one direction. Beyond that direction, matched-subject items add only +.0011, and only near-identical items align. If anything transfers across items, it should be shared operations (substitute, simplify, solve, count), not whole solutions. This gate checks whether that is visible in gradient geometry, before anyone builds step-level training on the idea.
Data (frozen).
- W's shortest verified-correct own completion per item (
grad_align/own.json: k=8, T 0.8, 1,024 tokens, seed 801), 338 items. - Steps are numbered lines (
1. **…**,### Step k:) whose header's first three words contain a verb from a fixed 29-word vocabulary, mapped to 25 labels. - Body = the tokens up to the next step header, 8–200 tokens.
- This gives 569 steps. The largest labels are compute (125 steps, 94 items), find (57), identify (44), determine (39) and solve (36).
Measurement.
- Per step: the gradient at W of the summed log-prob of the body tokens, conditioned on prompt plus completion prefix (teacher forcing).
- Parameters: transformer blocks and final norm only. Token embedding and output head are excluded, so shared surface tokens can't align through the vocabulary matrices.
- Count sketch with m = 2^18, same hash as F34. Six exact-vs-sketch validation pairs.
Primary test.
- Pairs: steps from different items only.
- Response: residual cosine after a cross-fit shared direction (fitted on one random half of the items, evaluated on pairs within the other half, then swapped).
- Regression: resid cos ~ 1 + same_label + same_subject + Jaccard(body token sets) + log len_a + log len_b.
- Uncertainty: item-cluster bootstrap with 1,000 resamples.
- Pass: the same_label coefficient has 95% LB > 0 AND point estimate ≥ +.01, ten times the whole-solution matched-item effect.
- Fail: step-level credit has no gradient-transfer mechanism at this scale either. I then stop this line and write the ceiling up with that explanation.
Caveats, stated in advance.
- These are off-policy teacher-forced gradients, with no Adam preconditioner and one checkpoint.
- Labels come from the model's own headers, not from verified operation types.
- Jaccard controls for lexical overlap only approximately.
- A pass would justify building step-level training. It wouldn't show that such training transfers.
Code: code/step_align.py (prep/sketch/analyze). It is queued after the S1-ORM shuffle control finishes.