PMI-shift · 매치드-베이스 인과 설계

좋은 메타인지 행동을 보상하는 자기증류

모델이 <|meta|> 블록을 지나며 정답을 향해 얼마나 마음을 바꾸는지(gold−decoy 마진의 이동)를 측정해 그 스팬에 보상한다. 자기 신뢰도를 맞추는 보정(calibration)은 필요조건일 뿐, 행동을 바꾸는 통제는 별개의 문제다.
의도. 메타인지는 목적이 아니라 정확도를 끌어올리는 수단이다. 우리는 신뢰도 숫자를 흉내내게 하지 않고, 메타 블록을 여는 순간(ctx_open)과 닫는 순간(ctx_close) 사이에서 gold와 decoy 사이 마진이 실제로 얼마나 벌어졌는지를 보상 신호로 삼는다. 그 이동이 곧 "유용한 메타인지"의 증거다.
checkpoint pmishift · gs300 eval pmishift_1030_v2 held-out 1030 (GSM8K 500 · MATH500 · AIME 30) PRELIMINARY · single seed
방법 · PMI-shift

메타 블록을 지나며 믿음이 이동하는 만큼 보상한다

한 rollout이 <|meta|>…<|/meta|> 블록을 담으면, 얼어붙은 참조 모델로 teacher-forcing 하여 블록 직전직후 두 지점에서 gold−decoy 마진을 잰다. 두 마진의 차이(SHIFT)가 신호다. 핵심 세 가지: 스팬-이동(한 지점이 아니라 구간의 변화) · 디코이 대조(정답 확률이 아니라 gold−decoy 판별 마진) · 메타 지역화(보상을 메타-콘텐츠 토큰에만 라우팅).

gold(정답) 방향 decoy(오답) 방향 <|meta|> 블록 일반 추론
실제 예시 · 부호 실수 → 메타에서 자각 → gold로 복귀 ↪ 커서를 올리면 애니메이션이 멈춥니다
ROLLOUT (실제 텍스트) body · 부호 실수 2−5=−3, 3·(−3)=−9, 7−9 = −2 <|meta|> · 자각/검산 "잠깐 — 7−(−9)=7+9=16, 부호 오류였다." continuation "따라서 답은 16." ctx_open ctx_close 믿음 축 · logp(gold) − logp(decoy) decoy −2 gold 16 0 (동률) PMI_open ≈ −0.8 PMI_close ≈ +1.5 SHIFT = +2.3 (clip ±2 → +2.0) +보상 → 메타-콘텐츠 토큰에만 reversal bonus: decoy→gold SAVE +1.0 gold→decoy DERAIL −2.0
계산. 얼어붙은 참조 모델로 teacher-forcing: PMI_open = logp(gold|ctx_open) − logp(decoy|ctx_open), 닫은 뒤 같은 계산이 PMI_close, SHIFT = PMI_close − PMI_open. 보상 = scale·clip(SHIFT, ±2) + reversal bonus, 메타-콘텐츠 토큰에 라우팅. PMI 값은 예시(illustrative)이고, 텍스트·정답·decoy·리전 경계는 실제 rollout이다.
왜 이동(shift)인가. 한 지점의 gold 확률만 높으면 "원래부터 답을 알던" 경우와 "메타에서 고쳐낸" 경우를 구별하지 못한다. 두 지점의 차이를 보상하면, 메타 블록이 인과적으로 믿음을 옮겼을 때에만 점수가 붙는다. gold와 decoy가 함께 움직이면 SHIFT ≈ 0 — 장식적 메타는 걸러진다.
위치 짓기 · 선행연구 vs 우리

필드는 "정답-우도 신호"로 수렴했다 — 우리는 스팬-이동·디코이·메타 구석에 선다

자기증류로 추론을 강화하려는 최근 흐름은 대부분 참조 정답의 로그확률을 신호로 삼는다. 우리의 신호도 그 이웃에 있지만, 좌표가 다르다. 각 카드는 그들이 한 일우리가 다르게 가는 지점을 짝짓는다.

AntiSD
arXiv 2605.11609 · 가장 가까운 선행
토큰별 조건 PMI u_t = log[π(y|x,c,y<t)/π(y|x,y<t)] so = Wait Maybe ↑ 부호 반전
검증된 해답 c를 조건으로 한 자기증류가 구조·지름길 토큰을 부풀리고 deliberation 토큰("Wait","Maybe")을 깎는다고 진단 → JS divergence를 상승(부호 반전). 우리: 토큰별 모방-부호반전이 아니라, 명시적 메타 스팬에서 gold−decoy 판별 마진의 스팬-이동을 보상 + 매치드-베이스 트윈. 지름길/보일러플레이트 위험은 공유(우리의 verify-boilerplate = 그들의 shortcut bias).
RLPR · CRM
Universal / Conditional Reward Modeling
reward ∝ logp(정답 | 추론) reasoning logp(gold) 디코이 대조 없음 · 전역 스칼라
추론을 참조 정답의 로그확률로 직접 보상(process reward 합 = 정답 로그확률). 필드가 수렴한 "정답-우도" 신호. 우리: 더 좁은 novelty — decoy 대조 + 스팬 이동 + 메타 지역화. 전역 우도가 아니라 두 지점 사이 판별 마진의 변화.
RLSD (self-distillation)
gold-conditioned teacher → 부작용
gold 조건 증류 → hedging 억제 Wait… Maybe 망설임 토큰 체계적 억제 → 어려운 문제 경로 단축
학생이 스스로의 교사 — 정답 조건 분포를 증류하며 RLVR과 결합. 그러나 gold 조건은 망설임 토큰을 억제해 hard 문제에서 추론 경로를 줄인다. 우리: 증류 위에 행동-통제 층을 얹되 priming하지 않는다. 신호는 모델 자신의 rollout 대조에서만 나온다.
OURS · PMI-shift
span-shift + decoy + meta + twin
decoy gold open close SHIFT (구간의 변화)
gold−decoy 판별 마진을 메타 스팬의 열림/닫힘 두 지점에서 재고, 그 이동을 보상. 메타-콘텐츠 토큰에 지역화. 메타만 뺀 매치드-베이스 트윈으로 인과 분리. 왜 decoy + 분해가 필요한가: 지름길/보일러플레이트 위험을 AntiSD와 공유하므로, RQ2 분해 + placebo 검정이 필수다.
RQ1 · 실측

메타 arm이 6/6 벤치에서 매치드-베이스를 넘는다

PMI-shift(gs300)와 메타 기제만 제거한 매치드-베이스(gs300)를 문제-쌍 부트스트랩으로 비교. 이득은 문제가 어려울수록 커진다.

pmishift gs300 vs matched-base gs300 avg@8 · AIME 16k avg@16 problem-paired bootstrap
before / after · matched-base(회색) → pmishift(초록) ↪ hover to pause
GSM8Kavg@8
base
89.9
meta
93.9
+4.0
MATH500avg@8
base
62.7
meta
81.5
+18.8
AIME@4k avg@8
base
5.0
meta
19.2
+14.2
AIME@16k avg@16
base
4.8
meta
18.5
+13.8
GSM8K+4.0ppp<.001
MATH500+18.8ppp<.001
AIME @4k+14.2ppbootstrap p<.001
AIME @16k+13.8ppMcNemar p≈.06 (n=30)
4k→16k는 두 arm 모두 평탄(토큰을 더 줘도 이득 없음). 그러나 base는 모든 예산에서 더 많이 잘린다(MATH500 ~19% vs 6%, AIME ~73% vs 50%). 즉 16k 비종료 degeneration은 base/데이터 성질이고, 메타 기제는 이를 유발이 아니라 완화한다.
정직한 caveat — 이건 "패키지" 효과다. 발사된 pmishift arm은 5-head 패키지다: correctness + meta/pmi-shift + cal 0.3 + emit 0.1 + format 0.35 + len_cost 0.08 + cf_group rollout split. 따라서 RQ1은 패키지 vs vanilla GRPO 효과이지 PMI-shift 단독이 아니다. PMI-shift만 떼어내는 분해는 RQ2(PENDING)에서 다룬다.
RQ2 · 아직 안 돌림

분해 사다리 — PMI-shift를 단독으로 분리하기 PENDING · not yet run

RQ1이 패키지 효과이므로, PMI-shift의 순기여를 격리하려면 아래 arm들을 같은 데이터·하이퍼파라미터·grader로 돌려야 한다. 아직 실행 전이며, 각 단계에 결정 규칙을 미리 못박아 둔다.

B

매치드-베이스 트윈

arm: base (meta 기제 제거, 5-head 중 pmi-shift·emit·cal·meta OFF)

모든 비교의 기준선. RQ1에서 이미 확보된 gs300.

G

Gandhi-arm (form-only)

arm: 메타 형식은 보상하되 PMI-shift 신호는 OFF

메타를 쓰는 것(form)만으로 오르는가, 아니면 신호가 필요한가를 가른다.

규칙: Gandhi ≈ base 면 → form만으로는 부족(신호가 진짜 레버). Gandhi ≫ base 면 → 이득의 상당부가 형식/emit에서 옴 → PMI-shift 주장 약화.
S

shift-only arm

arm: PMI-shift만 ON (cal·emit·len_cost 등 보조 head OFF)

PMI-shift 단독의 순효과. 이것이 우리가 주장하고 싶은 핵심 레버.

규칙: shift-only > Gandhi 면 → PMI-shift가 form을 넘는 순기여(주장 성립). shift-only ≈ Gandhi 면 → 이득은 신호가 아니라 형식/보일러플레이트(→ AntiSD 긴장 현실화).
P

placebo · decoy-shuffle

arm: SHIFT를 무작위/뒤섞은 decoy로 계산

SHIFT 보상이 진짜 gold−decoy 구조를 잡는지, 아니면 아무 스칼라나 밀어도 오르는지 검정.

규칙: placebo ≈ base 면 → SHIFT의 구조가 인과적(clean). placebo ≈ shift-only 면 → SHIFT는 장식, 다른 head가 진짜 레버.
RQ3 · 실측

이득은 어려운 문제에 집중된다 (Q4 +34.8pp) — 그런데 메타 발화는 비선택적

held-out 문제를 난이도 사분위(Q1 쉬움 → Q4 어려움)로 나눠 본 층화 분석. 쉬운 두 사분위는 이미 포화(100%), 이득은 뒤로 갈수록 커진다.

held-out 1030 · 난이도 층화 eval pmishift_1030_v2
난이도 사분위별 정확도 · base(회색) vs pmishift(초록) ↪ hover to pause
Q1가장 쉬움
base
100
meta
100
포화
Q2
base
100
meta
100
포화
Q3
base
84.6
meta
95.7
+11.1
Q4가장 어려움
base
12.1
meta
46.9
+34.8
이득 = 난이도 의존적. Q1/Q2는 포화, Q3 +11.1pp, Q4에서 +34.8pp로 최대. 그런데 메타 발화율은 사분위 전반 88–96%로 거의 일정 — 즉 발화는 비선택적인데 보상(payoff)만 난이도에 따라 달라진다.
비선택적 발화의 의미. 모델은 쉬운 문제에서도 거의 항상 메타를 쓴다(always-on). 이득이 어려운 곳에 몰리는 건 메타를 언제 쓰는지 골라서가 아니라, 어려운 문제에서만 메타가 실제로 믿음을 옮길 여지가 크기 때문이다. "유용할 때만 메타를 쓰는" 선택성은 아직 학습되지 않았다.
RQ4 · 실측

보정은 대체로 좋지만 AIME에서 과신 — 그리고 base는 신뢰도를 아예 안 낸다

메타 arm의 verbalized confidence를 ECE(15-bin)로 측정. 쉬운 벤치는 잘 맞지만 AIME에서는 크게 과신한다. base arm은 0% 메타 → 신뢰도 자체가 없어 비교 불가라는 구조적 비대칭을 정직하게 밝힌다.

meta arm only · ECE(15) eval pmishift_1030_v2 base emits 0% meta → uncomparable
ECE(15-bin) · 낮을수록 잘 보정됨 ↪ hover to pause
잘 보정 ◀ ▶ 과신 0 0.2 0.4 GSM8K 0.082 MATH500 0.052 AIME 0.402 pooled 0.053 base 신뢰도 없음 · 비교 불가
AIME 과신. 정답률이 낮은데도 평균 신뢰도가 높다 — AIME에서 81.8%가 과신, ECE 0.402. GSM8K/MATH500은 잘 보정(0.082 / 0.052), pooled 0.053. 구조적 비대칭: base는 메타를 0% 내므로 신뢰도 축에서 비교 자체가 불가능하다.
핵심 인사이트 — 보정 ≠ 통제. pooled ECE 0.053으로 보정은 대체로 성립하지만, 그것이 곧 "행동을 옳게 바꾸는" 메타인지 통제를 뜻하지는 않는다. AIME에서 보정이 무너지는데도 RQ1의 이득이 존재한다는 사실은, 보정은 필요조건일 뿐 충분조건이 아님을 보여준다.
한계 · 정직하게 읽기

믿기 전에 읽을 것

단일 시드 · PRELIMINARY

모든 수치는 single seed다. avg@8 / AIME avg@16으로 rollout 분산은 줄였지만, 시드 간 재현은 아직 없다. 결론은 예비적이며 다중 시드 확증이 필요하다.

패키지 효과 (PMI-shift 단독 아님)

RQ1은 5-head 패키지 vs vanilla GRPO. PMI-shift의 순기여는 RQ2 분해(Gandhi-arm · shift-only · placebo)가 끝나야 주장할 수 있다. 아직 미실행.

보일러플레이트 위험 (AntiSD 긴장)

SHIFT 보상이 진짜 자각이 아니라 "verify처럼 보이는 상투구"를 밀어올릴 수 있다 — 이는 AntiSD가 진단한 shortcut/deliberation-억제와 같은 실패 클래스다. RQ3의 비선택적 발화(88–96%)가 이 위험의 방증. placebo 검정이 이래서 필수다.

AIME degeneration은 완화지 해결 아님

16k에서 base가 ~73% 잘리는 비종료 degeneration을 메타가 완화(→50%)하지만 없애지 못한다. 4k→16k 이득이 평탄하다는 건 더 긴 예산이 답이 아니라는 뜻이다.

SFT v2 · grader

수치는 robust math_verify 재채점 기준(이전 broken-grader 수치와 다름)이며 held-out 1030에서 측정. SFT 초기화 버전(v2)과 데이터 구성에 결과가 의존한다.

안정적으로 성립하는 것

매치드-베이스 대비 6/6 벤치 우위, 이득의 난이도 집중(Q4 +34.8pp), 그리고 base가 유발한 16k degeneration을 메타가 완화한다는 방향성 — 이 세 가지는 단일 시드 안에서 일관된다.