본문으로 건너뛰기

애플의 '당황스러울 정도로 간단한 자가 증류' 논문 리뷰: 코드 생성 성능 향상 기법

애플 연구진이 발표한 SSD 기법은 모델이 높은 온도로 생성한 데이터를 스스로 학습함으로써 복잡한 검증기나 강화학습 없이도 코드 생성 성능을 대폭 향상시킨다.

챕터별 상세

00:30

SSD(Simple Self-Distillation)의 핵심 아이디어

SSD는 모델이 높은 온도(High Temperature) 설정에서 생성한 데이터를 다시 자신의 학습 데이터로 사용하는 자가 증류 기법이다. 기존의 자가 증류가 더 뛰어난 '스승 모델'을 필요로 했던 것과 달리, SSD는 동일한 모델이 생성한 가공되지 않은(Unverified) 샘플을 그대로 사용한다. 실험 결과 Qwen2-30B 모델의 LiveCodeBench 성능이 42.4%에서 55.3%로 약 30% 이상 향상되었다. 이는 복잡한 보상 모델이나 검증 절차 없이도 모델 내부의 잠재 능력을 끌어낼 수 있음을 시사한다.

자가 증류는 모델의 지식을 더 효율적으로 압축하거나 정제하기 위해 모델 자신의 출력을 학습에 활용하는 방식이다.

01:38

탐색과 정밀도 사이의 갈등 해결

코드 생성 시 모델은 특정 위치에서 다양한 대안을 탐색(Exploration)해야 하는 상황과 정확한 토큰을 선택(Precision)해야 하는 상황에 동시에 직면한다. SSD는 학습 시 높은 온도를 사용하여 모델이 다양한 가능성을 경험하게 하고, 추론 시에는 낮은 온도를 사용하여 정밀도를 높이는 방식으로 이 갈등을 해결한다. 논문은 이를 'Fork(분기)'와 'Lock(고정)' 위치라는 개념으로 설명하며, SSD가 분기점에서의 탐색 능력을 유지하면서도 고정점에서의 확신을 강화한다고 분석했다. 결과적으로 모델은 더 깨끗한 로짓 분포를 가지게 되어 오류율이 줄어든다.

로짓 분포가 '피키(Peaky)'하다는 것은 모델이 특정 정답에 대해 매우 높은 확신을 가지고 있음을 의미한다.

11:40

강화학습(RL)과의 비교 및 효율성

전통적인 강화학습(RL)은 정답 여부에 따른 희소한 피드백(Sparse Feedback)을 기반으로 모델을 최적화하지만, SSD는 모든 토큰에 대해 SFT를 수행하는 밀집된 피드백(Dense Feedback) 방식이다. SSD는 외부 검증기(Verifier)나 보상 모델이 필요 없기 때문에 연산 비용이 매우 저렴하며 구현이 간단하다. 연구진은 SSD가 RLHF가 수행하는 분포 최적화 과정을 훨씬 단순한 형태로 모사하고 있다고 평가했다. 특히 데이터의 정답 여부를 가리지 않고 학습해도 성능이 오른다는 점이 가장 놀라운 발견 중 하나로 꼽혔다.

RLHF는 인간의 피드백을 통해 모델을 정렬하는 복잡하고 비용이 많이 드는 과정이다.

python
def apply_temperature(logits, temp):
    return logits / temp

# vLLM implementation of SSD decoding
# 1. Temperature scaling
# 2. Top-k / Top-p filtering
# 3. Truncation
# 4. Gumbel-max sampling

논문에서 사용된 SSD 디코딩의 핵심 로직을 요약한 의사 코드

17:00

데이터 반복 학습의 영향력 분석

영상의 토론자들은 SSD의 성공 요인 중 하나로 '데이터 반복(Data Repetition)' 효과를 언급했다. 모델이 이미 알고 있는 지식을 다른 분포로 반복 학습하면서 특정 패턴에 대한 이해도가 깊어지는 현상이 발생한다. 최근 연구들에 따르면 품질이 낮은 데이터라도 적절한 방식으로 반복 학습하면 추론 능력이 향상되는 경우가 보고되고 있다. SSD 역시 모델이 생성한 데이터를 통해 자신의 지식 체계를 재구성하고 강화하는 과정을 거치는 것으로 해석된다. 이는 단순히 새로운 정보를 배우는 것이 아니라 기존 지식의 인출 확률을 최적화하는 과정이다.
22:00

실무 적용 가능성과 한계점

SSD는 코딩뿐만 아니라 수학이나 일반적인 추론 도메인에서도 성능 저하 없이 적용 가능한 것으로 나타났다. 하지만 특정 벤치마크에 과적합(Overfitting)될 위험이 있으며, 학습 시 사용되는 온도의 설정값에 따라 결과가 민감하게 변할 수 있다. Hugging Face 팀은 이 기법이 매우 간단하므로 기존 SFT 파이프라인에 즉시 통합하여 성능 향상을 시도해볼 가치가 충분하다고 결론지었다. 특히 자원이 부족한 환경에서 고가의 RL 학습 대신 선택할 수 있는 훌륭한 대안이다.

용어 해설

자가 증류(Self-Distillation)
모델이 생성한 데이터를 다시 자신의 학습에 사용하는 기법이다. 별도의 거대 모델(Teacher) 없이도 모델 스스로의 출력을 정제하여 성능을 개선하는 데 활용된다.
지도 미세 조정(SFT)
특정 데이터셋에 대해 모델을 직접 학습시키는 과정이다. 이 논문에서는 모델이 높은 온도로 생성한 데이터를 정답지로 삼아 모델을 다시 학습시키는 용도로 사용된다.
온도 스케일링(Temperature Scaling)
LLM의 출력 확률 분포를 조절하는 하이퍼파라미터이다. 온도가 높으면 더 다양하고 창의적인(때로는 오류가 섞인) 결과가 나오고, 낮으면 가장 확률이 높은 토큰 위주로 보수적인 결과가 나온다.
로짓(Logit)
모델의 마지막 레이어에서 출력되는 정규화되지 않은 예측값이다. 소프트맥스 함수를 거쳐 확률로 변환되기 전의 값으로, 모델이 특정 토큰을 얼마나 선호하는지 나타내는 지표가 된다.
AI 분석 전체 내용 보기

AI 요약 · 북마크 · 개인 피드 설정 — 무료

출처 · 인용 안내

원문 발행 2026. 04. 16.수집 2026. 04. 16.출처 타입 YOUTUBE

인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.