컴백부터 K-뷰티까지 — K-컬쳐의 모든 것을 메일로 받아보세요메일로 받아보기

METAL MEDIA

정답 없이도 AI가 스스로 채점하고 스스로 가르쳐 수학 실력을 올린다

arXiv:2608.062962026-08-08

On-Policy Self-Distillation without Any Supervision

정답 없이도 AI가 스스로 채점하고 스스로 가르쳐 수학 실력을 올린다

언어모델을 후련(post-training)시킬 때 보통 정답이나 더 큰 모델의 도움이 필요했는데, 이 연구는 모델이 같은 문제를 여러 번 풀어본 뒤 다수결로 나온 답을 임시 정답처럼 쓰는 u-OPSD 방법을 제안한다. 다수결과 다른 답을 낸 풀이만 골라 다수결에 맞춘 스스로의 예측 분포를 가르치는 방식으로, 정답 없이도 수학 벤치마크에서 정답 기반 기존 방법과 비슷하거나 더 좋은 성적을 냈다. 코드는 공개되어 있다.

METAL MEDIA 해설 도표

u-OPSD: 정답 없이 스스로 교사가 되는 구조

증거 상태측정 결과가 보고됨

  1. 1. 다중 롤아웃 생성같은 문제에 대해 모델이 G=8번 독립적으로 답을 생성한다
  2. 2. 다수결 투표생성된 답들 중 가장 많이 나온 답을 임시 정답으로 삼고, 이 답과 일치/불일치하는 롤아웃을 나눈다(임계값 τ=0.5 미달 시 학습에서 제외)
  3. 3. 교사 조건화다수결과 일치하는 풀이 중 가장 긴 것을 정답 풀이 대신 사용해 교사 분포를 만든다
  4. 4. 불일치 롤아웃에 증류다수결과 다른 답을 낸 학생 풀이의 각 토큰 위치에서 교사의 다음 토큰 분포를 학생에게 가르쳐(순방향 KL) 스스로 틀린 지점을 고치게 한다
METAL MEDIA이 원문을 바탕으로 재구성한 해설 도표이며, 논문 저자의 원문 figure가 아닙니다.

무엇을 했나

  1. 기존 온폴리시 자기증류(OPSD) 방법은 모델이 스스로 만든 답을 학습에 쓰지만, 여전히 정답 풀이나 외부 피드백, 더 큰 모델의 도움 같은 외부 정보가 필요했다.
  2. u-OPSD는 한 문제에 대해 모델이 여러 번(G=8) 답을 생성한 뒤, 답들이 일정 비율(임계값 τ=0.5) 이상 일치하면 다수결로 나온 답을 정답 대신 쓴다.
  3. 다수결과 일치하는 풀이 중 가장 긴 것을 교사 역할로 삼아, 다수결과 다른 답을 낸 학생 풀이에 그 교사의 다음 토큰 확률 분포를 가르쳐(distill) 학생이 스스로 틀린 지점을 고치게 한다.
  4. 수학 경시 벤치마크 5종(AIME24, AIME25, HMMT25, MATH500, AMC23)에서 Qwen3 4B/8B 비추론(non-thinking) 모드 기준 기본 모델보다 8.5~10.7점 올랐고, 정답을 쓰는 기존 OPSD보다도 2.3~3.2점 더 높았다.
  5. 추론(thinking) 모드에서는 정답 기반 OPSD와 비슷한 수준(4B에서 0.9점 우위, 8B에서 동률)이었고, 정답 기반 GRPO보다는 0.7~1.1점 높았다.
Figure 1: Comparison between OPSD / SDFT with ground-truth solution or ICLs (left), SDPO with rich feedback from the environment (middle) and our u-OPSD without any supervision.
Figure 1: Comparison between OPSD / SDFT with ground-truth solution or ICLs (left), SDPO with rich feedback from the environment (middle) and our u-OPSD without any supervision.
Table 1: Performance comparison on math reasoning benchmarks for Qwen3 models with non-thinking mode.
MethodAIME24AIME25HMMT25MATH500AMC23Avg.
Qwen3-4B
Base25.8317.7810.8384.1066.2540.96
w/ GT.+ SFT26.6719.7213.0684.8569.3842.73
+ GRPO25.0022.5015.0086.2080.6245.86
+ OPSD32.2220.8316.3985.7576.2546.29
[1.2pt/1.4pt] w/o GT.+ TTRL25.0020.8311.9483.6067.5041.77
+ RENT22.2220.2811.3984.0074.3842.45
+ Intuitor23.8920.0011.6783.7070.6241.98
+ u-OPSD37.5027.7814.4486.5081.2549.49
Qwen3-8B
Base27.5023.3313.6184.0569.3843.57
w/ GT.+ SFT26.9421.6711.9484.1072.5043.43
+ GRPO30.5621.9413.0687.8573.7545.43
+ OPSD41.6728.0618.3387.1585.0052.04
[1.2pt/1.4pt] w/o GT.+ TTRL27.2221.1113.0684.4571.8843.54
+ RENT28.3321.6710.8384.0070.0042.97
+ Intuitor26.1122.5011.6784.2073.1243.52
+ u-OPSD45.5634.7218.6189.5583.1254.31
Figure 2: Overview of Unsupervised On-policy Self-Distillation (u-OPSD), which replaces ground-truth supervision in On-Policy Self-distillation with pseudo-labels generated from the model’s own majority-vote consensus.
Figure 2: Overview of Unsupervised On-policy Self-Distillation (u-OPSD), which replaces ground-truth supervision in On-Policy Self-distillation with pseudo-labels generated from the model’s own majority-vote consensus.
Table 2: Thinking mode, per benchmark at step 150, under the protocol of Table 1. Shading and bold as in that table.
MethodAIME24AIME25HMMT25MATH500AMC23Avg.
Qwen3-4B
Base74.1764.7245.5694.8095.0074.85
w/ GT.+ SFT73.0669.1741.6795.4094.3874.73
+ GRPO73.8969.4443.6195.4599.3876.35
+ OPSD75.2868.0643.0695.2099.3876.20
[1.2pt/1.4pt] w/o GT.+ TTRL72.7868.3345.5695.9096.2575.76
+ RENT74.7265.8343.0695.2599.3875.65
+ Intuitor76.3968.3342.7895.3597.5076.07
+ u-OPSD76.3968.0646.9495.7598.1277.05
Qwen3-8B
Base75.5666.6745.0096.3596.8876.09
w/ GT.+ SFT76.3969.7243.8995.8095.0076.16
+ GRPO76.9469.1747.7895.7095.0076.92
+ OPSD80.8369.7246.6795.7596.8877.97
[1.2pt/1.4pt] w/o GT.+ TTRL77.2268.6146.9495.7596.2576.95
+ RENT77.5070.2845.8395.9596.2577.16
+ Intuitor76.9470.2844.1796.2095.6276.64
+ u-OPSD76.9471.3947.5096.0098.1277.99
Figure 3: Training curves of the Qwen3-4B runs quoted in Tables 1 and 2, on AIME24, AIME25 and MATH500 over the first four checkpoints, in thinking (left three columns) and non-thinking mode (right three); the dashed line is the base model. Top: u-OPSD against the supervised arms. Bottom: u-OPSD against the label-free ones. The axis counts checkpoints: u-OPSD saves every 25 steps and GRPO every 50, so the GRPO points span its steps 50–200. The two modes separate at a glance—in non-thinking mode u-OPSD breaks away from every other arm, while in thinking mode all methods stay within about two points of base.
Figure 3: Training curves of the Qwen3-4B runs quoted in Tables 1 and 2, on AIME24, AIME25 and MATH500 over the first four checkpoints, in thinking (left three columns) and non-thinking mode (right three); the dashed line is the base model. Top: u-OPSD against the supervised arms. Bottom: u-OPSD against the label-free ones. The axis counts checkpoints: u-OPSD saves every 25 steps and GRPO every 50, so the GRPO points span its steps 50–200. The two modes separate at a glance—in non-thinking mode u-OPSD breaks away from every other arm, while in thinking mode all methods stay within about two points of base.
Table 3: Qwen3-30B-A3B-Instruct-2507, non-thinking, scored as pass@1 rather than average@n: generation i of every problem forms one single-sample run, and i=1,2,3 give three estimates, reported as mean ± population standard deviation. Each arm is shown at the checkpoint with the best five-benchmark mean under this metric. Because the metric differs from Tables 1 and 2, the two are not comparable and the numbers are kept apart. Bold marks the best value in each column.
ModelMethodAIME24AIME25HMMT25MATH500AMC23Avg.
Qwen3-30B-A3B -Instruct-2507Base80.00±0.0063.33±2.7243.33±2.7296.33±0.2595.83±1.1875.77±0.85
OPSD78.89±3.1461.11±3.1447.78±1.5797.20±0.4396.67±2.3676.33±1.58
u-OPSD75.56±5.6765.56±4.1650.00±0.0097.00±0.4399.17±1.1877.46±2.09
Qwen3-4B -Instruct-2507Base66.67±7.2053.33±5.4427.78±4.1693.87±0.3493.33±1.1867.00±2.25
OPSD62.22±5.0952.22±5.0931.11±5.0994.93±0.6495.00±0.0067.10±1.13
u-OPSD68.89±5.6757.78±1.5728.89±3.1494.20±0.8694.17±1.1868.78±1.70
Table 4: Performance under combinations of the teacher-reference selection (rows) and the distillation-target selection (columns), on Qwen3-8B non-thinking, G=8, τ=0.5, k=1. Each cell is the five-benchmark average at the best checkpoint in 25–150, with the step given in parentheses. “label-only” strips the teacher’s reference down to the boxed pseudo-label.
Teacher ref. \ distilllongestrandomshortest
label-only43.40 (125)43.00 (100)41.55 (25)
shortest (default)57.10 (125)54.37 (75)55.23 (150)
random57.96 (75)56.93 (75)55.77 (75)
longest59.00 (75)57.90 (100)55.01 (125)
Table 5: Ablation of the disagreeing-rollout selection policy (matched decay schedule). Each cell shows step 150 / best checkpoint in 25–150. All variants use G=8, τ=0.5.
VariantAIME24AIME25HMMT25MATH500AMC23
OPSD (supervised)27.50 / 27.5020.00 / 23.6112.50 / 13.3383.80 / 84.6571.88 / 72.50
disagree-1 (default)33.89 / 35.2827.78 / 27.7814.72 / 16.1187.60 / 87.6079.38 / 79.38
disagree-229.72 / 31.1123.33 / 27.5015.28 / 16.6786.20 / 86.3075.62 / 80.00
disagree-327.78 / 34.7225.83 / 31.1113.33 / 18.3385.45 / 87.1072.50 / 76.25
disagree-all (no cap)29.44 / 30.8322.50 / 25.8310.00 / 13.3385.25 / 86.0073.75 / 74.38
longest-133.06 / 34.7223.33 / 26.6714.72 / 18.3385.50 / 86.8073.75 / 78.12
Table 6: Comparison of divergence computation strategy: Full vocabulary is logit distillation over every token (2); sampled token evaluates the two policies only at the token the student drew (26); top-k rows truncate the teacher to its k largest entries. We report on Qwen3-8B non-thinking at the best checkpoint.
VariantAIME24AIME25HMMT25MATH500AMC23Avg.
Student token29.4419.7212.7884.7070.6243.45
top-k2050.0037.7819.4490.2088.1256.94
5048.0637.5024.1789.4585.6255.85
10053.8943.3320.5691.6585.6259.01
20051.6740.0022.5090.1588.7558.11
Full-vocabulary53.8937.5020.2889.9085.6257.10
Table 7: Divergence family Dβ under u-OPSD, on Qwen3-8B non-thinking, longest-1, G=8, τ=0.5. Each arm is shown at its best checkpoint in 25–150. Objective names follow 59, who report the same comparison under gold supervision.
ObjectiveAIME24AIME25HMMT25MATH500AMC23Avg.
Base27.5023.3313.6184.0569.3843.57
Forward KL KL(πT∥πS), β=0 (default)53.8937.5020.0089.7584.3857.10
Reverse KL KL(πS∥πT), β=1training diverges (not scored)
JSD (β=0.5)29.4421.9412.2283.7069.3843.34
Table 8: Reproduction of published OPSD results with the released code and hyperparameters (avg@12, temperature 1.0). “pub.” denotes the numbers published in the official OPSD repository; “ours” our rerun. Checkpoint columns are steps 50/75/100/150, matching the checkpoints published for this configuration.
ConfigBenchbasecheckpoints
Qwen3-4B (non-thinking)AIME24 pub.23.120.327.531.132.8
AIME24 ours22.223.127.232.231.9
AIME25 pub.21.421.420.821.121.9
AIME25 ours17.820.823.120.821.1
HMMT25 pub.10.811.113.116.414.4
HMMT25 ours12.210.612.816.411.7
Table 9: GRPO under matched and mismatched reasoning modes. Each row is the best of ten checkpoints (steps 50–500) by five-benchmark mean, scored at temperature 1.0 against the base model in the evaluation mode of that row. The two “thinking → non-thinking” rows reuse the checkpoints of the rows above them; only the evaluation prompt differs.
TrainEvalStepAIME24AIME25HMMT25MATH500AMC23Avg.Δ base
Qwen3-4B
thinkingthinking30073.8969.4443.6195.4599.3876.35+1.79
thinkingnon-thinking50025.0021.3913.8983.7576.2544.06+2.62
non-thinkingnon-thinking20025.0022.5015.0086.2080.6245.86+4.43
Qwen3-8B
thinkingthinking25076.9469.1747.7895.7095.0076.92+0.66
thinkingnon-thinking10027.2222.5013.3385.2067.5043.15+1.20
non-thinkingnon-thinking50030.5621.9413.0687.8573.7545.43+3.48
Table 10: Self-consistency threshold τ on Qwen3-8B non-thinking, longest-1, G=8. Each cell shows step 150 / best checkpoint in 25–150. τ is the fraction of valid rollouts that must agree for a prompt to receive a pseudo-label; τ=0 accepts every prompt.
τAIME24AIME25HMMT25MATH500AMC23Avg.
053.33 / 57.5041.39 / 46.6722.50 / 27.2290.70 / 92.1588.12 / 89.3859.21 / 61.83
0.350.83 / 53.3334.44 / 41.6720.28 / 23.0689.85 / 90.6585.00 / 88.1256.08 / 58.59
0.543.06 / 53.8935.83 / 37.5020.28 / 20.2888.00 / 89.9082.50 / 85.6253.93 / 57.10
0.732.78 / 37.5023.61 / 26.6713.06 / 14.4485.95 / 87.7074.38 / 76.2545.96 / 48.18
0.932.22 / 32.2220.83 / 22.5013.61 / 13.6184.70 / 84.8570.62 / 72.5044.40 / 44.40
Table 11: Number of rollouts per prompt G on Qwen3-8B non-thinking, longest-1, τ=0.5. Each cell shows step 150 / best checkpoint in 25–150.
GAIME24AIME25HMMT25MATH500AMC23Avg.
447.22 / 50.2835.83 / 37.7818.06 / 21.1188.15 / 90.2084.38 / 87.5054.73 / 56.99
843.06 / 53.8935.83 / 37.5020.28 / 20.2888.00 / 89.9082.50 / 85.6253.93 / 57.10
1253.61 / 57.2240.83 / 43.3320.56 / 26.3991.00 / 92.0086.25 / 90.0058.45 / 61.79
1651.67 / 55.2841.67 / 44.1722.22 / 22.2289.90 / 89.9087.50 / 87.5058.59 / 59.37
Table 12: How the teacher is updated, on Qwen3-8B non-thinking, longest-1, G=8, τ=0.5. Each cell shows step 150 / best checkpoint in 25–150. “fixed” freezes the teacher at the initial policy (the base model with the adapter disabled), which is OPSD’s own setting and ours everywhere else; the EMA rows let the teacher track the student at the given decay.
TeacherAIME24AIME25HMMT25MATH500AMC23Avg.
fixed43.06 / 53.8935.83 / 37.5020.28 / 20.2888.00 / 89.9082.50 / 85.6253.93 / 57.10
EMA 0.99943.61 / 52.7836.67 / 41.3918.33 / 23.3388.40 / 90.5087.50 / 87.5054.90 / 58.82
EMA 0.9952.78 / 54.1739.72 / 43.8923.33 / 23.3389.85 / 91.0083.75 / 86.8857.89 / 58.88
EMA 0.99554.17 / 55.0040.28 / 43.3321.94 / 22.5090.10 / 90.7583.75 / 89.3858.05 / 59.47
Table 13: Learning-rate schedule ablation: best AIME24 checkpoint (avg@12) per method under an effectively-constant LR (30-epoch horizon, ≈5×10−6 throughout) vs. the matched 150-step linear decay used throughout Tables 1 and 2. The constant-LR column is OPSD’s native configuration, so the released figure is directly comparable there; no released run exists under the decayed schedule.
MethodConstant LRLinear decay
OPSD (supervised)32.2227.50
as released32.8
disagree-137.5035.28
disagree-232.7831.11
disagree-333.6134.72
disagree-all33.6130.83
longest-131.9434.72

실제로 확인된 결과

  • 수학 벤치마크 5종 평균에서 Qwen3-4B/8B 비추론 모드 기준 기본 모델보다 각각 8.5점, 10.7점 향상되었고, 정답 기반 OPSD보다 각각 3.2점, 2.3점 더 높았다.
  • 추론 모드에서는 4B에서 OPSD보다 0.9점, 8B에서 동률이었으며, GRPO보다는 각각 0.7점, 1.1점 높았다.
  • 명령어 튜닝 모델 Qwen3-30B-A3B-Instruct-2507과 Qwen3-4B-Instruct-2507에서도 각각 75.77→77.46, 67.00→68.78로 향상되어 OPSD를 1.1점, 1.7점 앞섰다.
  • 훈련 프롬프트 64개를 분석한 결과 롤아웃의 96.3%에서 답을 추출할 수 있었고, 94.0%의 문제가 자기일치 임계값을 넘겼으며, 그 중 86.7%가 실제 정답과 일치했다.
  • 교사가 짧은 답(정답 라벨만)만 보고 학습하면 11.4~15.6점 성능이 떨어졌고, 순방향 KL 대신 대칭 발산(JSD)을 쓰면 13.8점 떨어져 훈련 전 모델 수준에 머물렀다.

어디에 쓸 수 있나

  • 정답 채점 라벨을 구하기 힘든 영역에서, 정답 대신 모델 스스로의 다수결 합의를 임시 정답으로 활용해 후련시키는 파이프라인 설계
  • 경시 수학처럼 정답이 명확히 추출 가능한 도메인에서 정답 없이 모델 성능을 끌어올리는 훈련 루프 구축
  • 기존 OPSD/GRPO 같은 정답 기반 방법과 비교 실험을 할 때 라벨 없는 기준선으로 참고

한계와 남은 검증

  • 실험은 Qwen3 계열(4B, 8B)과 경시 수학 도메인 한 곳에만 한정되어 검증되었고, 답을 명확히 추출·정규화할 수 있는 문제에만 다수결 투표가 적용된다.
  • 비추론 모드에서는 이득이 크지만 추론 모드에서는 이득이 작아, 기본 모델이 이미 강할수록 개선 여지가 줄어드는 경향이 있다.
  • 다수결이 만드는 임시 정답은 기본 모델이 가장 많이 내는 답을 그대로 반영하므로, 기본 모델의 정확도에 성능이 묶여 있고 훈련 중 오답 다수결(13.3% 사례에서 확인)이 섞일 위험이 있다.
  • 체크포인트별 12개 샘플과 5개 벤치마크로 분산을 줄였지만, 시드를 여러 번 반복한 오차범위(seed-replicated error bar)는 아직 보고되지 않아 후속 개정에서 추가될 예정이다.
  • GRPO를 훈련한 추론 모드 그대로 비추론 평가로 옮기는 실험은 했지만, 반대 방향(비추론 훈련 후 추론 평가)은 아직 수행되지 않았다.

왜 중요한가

이 방법이 통하면, 정답이 없거나 채점하기 어려운 문제(코딩, 서술형 등)에서도 모델이 스스로 훈련 데이터를 만들어 실력을 키울 수 있는 길이 열린다. 정답 라벨링에 드는 비용과 정답 없는 영역에서의 확장성 문제를 동시에 줄일 잠재력이 있다는 점에서 중요하다.

이 논문의 용어

  • 온폴리시 자기증류(OPSD) · 모델이 자기 자신이 만든 답변으로 스스로를 다시 가르치는 학습 방식
  • 다수결 투표(majority vote) · 같은 문제를 여러 번 풀게 한 뒤 가장 많이 나온 답을 정답처럼 취급하는 방법
  • 자기일치 임계값(τ) · 여러 번 푼 답 중 얼마나 많은 비율이 일치해야 그 답을 믿고 학습에 쓸지 정하는 기준값
  • GRPO · 정답과 비교한 보상값을 이용해 정책을 업데이트하는 강화학습 기법
  • 순방향 KL 발산(forward KL) · 교사 모델과 학생 모델의 예측 분포 차이를 계산해 학생을 교사 쪽으로 맞추는 척도

저자 · Yijiang Li

arXiv에서 원문 보기

최신 논문

논문 전체 보기 →

METAL MEDIA 최신 기사

그림 출처: Yijiang Li et al., arXiv:2608.06296, CC BY 4.0