TL;DR
기존의 토큰별 정보량 관점으로는 LLM에서의 정책기울기 기반 RL이 왜 샘플 효율적인지 설명하기 어렵다. 본문은 핵심 원인으로 신호대잡음비(SNR)를 제시하며, RL의 보상 신호는 적은 비트이지만 목표 과제에 대한 순수한 신호로 집중되어 있어 사전학습의 방대한 토큰별 그래디언트에서 요구되는 목표 신호가 묻히는 상황을 피한다고 본다. 따라서 RL은 이미 충분히 능력을 갖춘 사전학습 기반 모델에서 짧은 파라미터 이동으로 특정 과제를 급격히 개선할 수 있으며, 이 과정에서는 미드트레이닝으로 출발점을 조정하고 편향-분산 균형을 신중히 관리하는 것이 중요하다.
섹션별 상세
용어 해설
- 사전학습(pretraining)
- — 대규모 텍스트 코퍼스에서 다음 토큰을 예측하는 목적(next-token prediction)으로 모델을 학습해 일반적 언어능력과 광범위한 표현을 획득시키는 단계로, 각 토큰마다 손실을 계산해 많은 양의 그래디언트를 생성한다.
- 정책 기울기 방법(policy gradients)
- — 에이전트의 행동 정책을 직접 매개변수화하고 샘플된 에피소드의 보상을 바탕으로 확률적 정책의 파라미터를 기울기법으로 갱신하는 강화학습 계열로, LLM 적응에서는 보상 롤아웃의 스칼라 신호를 활용한다.
- 지도적 파인튜닝(SFT)
- — 사전학습된 모델을 인간 레이블이 달린 질의-응답 등 지도 데이터로 추가 학습시켜 특정 생성 스타일이나 행동을 강화하는 단계로, 롤아웃 기반 RL과는 목표 신호의 밀도와 성격이 다르다.
- 신호대잡음비(SNR)
- — 학습 신호(목표 손실로부터 얻는 유효한 그래디언트 정보)와 배경 잡음(목표와 무관한 예측 손실로 인한 그래디언트)의 비율을 의미하며, 동일한 샘플에서도 목적에 맞는 신호를 좁게 쥐어주는 방법이 SNR을 높인다.
- 중간학습(mid-training)
- — 사전학습과 최종 RL 사이에 위치시키는 추가적 데이터/목표 단계로, RL 시작 시점에 요구되는 최소 성공률을 확보하고 목표 행동을 모델 파라미터 상에서 더 가까운 거리로 압축해 RL의 비트 제약을 완화한다.
근거 모음
- 정책기울기 계열의 RL은 수백에서 수천 스텝 만에 특정 과제를 거의 완벽하게 수행하도록 모델을 개선할 수 있다. — 본문의 반복적 관찰 부분: 'RL can achieve extremely rapid gains... in only a few hundreds to thousands of steps' 및 'strong gains can be seen... in as few as ten steps'.
- 사전학습 기반의 다음 토큰 예측(NTP)은 샘플당 정보량은 크지만 목표 과제의 신호대잡음비가 낮아 RL보다 특정 과제에서는 덜 효과적일 수 있다. — 본문의 SNR 논의: NTP는 모든 토큰에 손실을 계산해 많은 그래디언트를 생성하지만 그중 상당수가 목표와 무관한 잡음임을 지적한 부분.
- LLM RL이 고전 RL과 달리 오프폴리시성 및 수치적 불일치에 매우 민감하기 때문에 리플레이 버퍼나 값추정 도입이 성능을 저해할 수 있다. — 본문에서 'tiny numerical differences... suddenly become decisive'와 'even moderate degrees of off-policyness appear highly damaging'라는 관찰을 근거로 함.
기술
- pretraining
- SFT
- policy gradients
- next-token prediction (NTP)
- SGD
활용 사례
- 사전학습된 LLM을 특정 과제에서 빠르게 성능 향상시키기 위한 RL 미세조정
- 미드트레이닝으로 RL 시작 조건을 개선해 성공률을 확보하는 파이프라인
AI 요약 · 북마크 · 개인 피드 설정 — 무료
출처 · 인용 안내
인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.
