개요
함께 초기 훈련을 안정화하고 최종 정확도를 향상시킵니다. 이것이 바로 거의 모든 현대 트랜스포머가 이런 방식으로 훈련되는 이유입니다.
심층 분석
훈련이 시작되면 모델 가중치가 무작위이고 기울기가 클 수 있으므로 큰 학습률로 바로 점프하면 손실 급증이나 발산이 발생하는 경우가 많습니다. 특히 첫 번째 단계에서 분산 추정치를 신뢰할 수 없는 Adam과 같은 적응형 최적화 프로그램의 경우 더욱 그렇습니다. Warmup은 수백 단계에서 수천 단계에 걸쳐 속도를 선형적으로 증가시켜 이 문제를 해결합니다. 모델이 안정된 기반에 있으면 코사인 어닐링이 대신되어 최고점의 0.5 * (1 + cos(pi * t / T)) 비율로 감소합니다. 코사인 형태는 빠른 진행을 위해 초기에 속도를 높게 유지한 다음 점차적으로 완화하여 최적화 프로그램이 튕기는 대신 적절한 최소값에 안착할 수 있도록 합니다.
실제 구현
GPT 스타일 및 BERT 스타일 언어 모델은 단계의 처음 ~1~2%에 걸쳐 선형 워밍업을 사용한 후 거의 0에 가까운 코사인 붕괴를 사용합니다.
ViT(Vision Transformer)는 ImageNet의 초기 발산을 방지하기 위해 코사인 어닐링과 짧은 워밍업을 통해 학습합니다.
Hugging Face Transformers는 작업 미세 조정을 위한 한 줄 스케줄러로 'get_cosine_schedule_with_warmup'을 제공합니다.
Stable Diffusion 및 기타 확산 모델은 사전 훈련된 가중치를 적용할 때 경사 폭발을 방지하기 위해 준비 작업을 통해 미세 조정됩니다.
워밍업 및 코사인 어닐링 일정의 미래
Warmup-plus-cosine은 대규모 언어 모델의 기본 방식으로 남아 있지만 변형이 확산되고 있습니다. WSD(Warmup-Stable-Decay)는 일정한 속도를 유지한 다음 마지막에 급격하게 감소하므로 고정 길이로 다시 커밋하지 않고도 실행을 쉽게 확장할 수 있습니다. 연구원들은 또한 워밍업이 작동하는 이유(이를 경사 노이즈 및 손실 환경 곡률과 연결)와 도구가 점점 더 워밍업 길이 및 최고 속도를 자동 조정하여 오늘날 지배적인 수동 시행착오를 줄이는 이유를 연구하고 있습니다.
실천에 옮기기
왜 중요한가요?
- 비용 및 예산. 아키텍처 결정은 수년 동안 성능과 운영 비용을 결정합니다.
- 더 명확한 결정들. 기술 교육은 팀이 최신 스택뿐만 아니라 올바른 스택을 선택하는 데 도움이 됩니다.
- 품질 관리. 더 나은 엔지니어링 선택은 생산 시 신뢰성 사고를 줄입니다.
조심하세요
- 하나의 벤치마크를 최적화하면 더 광범위한 시스템 약점을 숨길 수 있습니다.
- 인프라 및 유지 관리 비용은 종종 과소평가됩니다.
- 시스템이 더욱 복잡해짐에 따라 보안 및 관찰 가능성의 격차가 커질 수 있습니다.
다음 단계
- 구현하기 전에 지연 시간, 품질, 비용 목표를 정의하세요.
- 현실적인 로드 및 데이터 조건에서 벤치마킹합니다.
- 오류, 드리프트 및 사용자 영향에 대한 계측기 모니터링.
- 확장하기 전에 롤백 및 사고 대응 경로를 준비하세요.
계속 탐색하세요
무료 뉴스레터
일일 AI 브리핑 받기
매주 평일 아침마다 3개의 검증된 AI 스토리를 일반 영어로 작성합니다. 무료이며, 광고 없음.
매주 평일마다 한 번 이메일을 받습니다. 한 번 클릭으로 구독 해지. 주소를 판매하거나 공유하지 않습니다.
자신을 테스트해 보세요
Take the Warmup and Cosine Annealing Schedules quiz
모든 답변에 즉각적인 피드백과, 과정을 통과하면 검증 가능한 ID가 있는 공유 가능한 인증서를 제공합니다.
무료 AI 교육을 지원합니다. AI Understanding는 501(c)(3) 비영리 단체입니다 — 광고 없음, 유료 장벽 없음, 영원히. 기부하기
자주 묻는 질문
워밍업 및 코사인 어닐링 일정이란 무엇입니까?
워밍업은 훈련 전에 거의 0에서 학습률을 부드럽게 높인 다음, 코사인 어닐링을 통해 코사인 곡선을 따라 학습률을 다시 낮춥니다. 함께 초기 훈련을 안정화하고 최종 정확도를 향상시킵니다. 이것이 바로 거의 모든 현대 트랜스포머가 이런 방식으로 훈련되는 이유입니다.
훈련 시작 시 준비 단계의 주요 목적은 무엇입니까?
무작위 가중치에 대해 큰 학습 속도로 시작하면 손실 급증이나 발산이 발생할 수 있으므로 준비 단계에서는 속도를 부드럽게 높여 초기 단계를 안정적으로 유지합니다.
코사인 어닐링은 어떤 형태에 따라 학습률을 감소시킵니까?
속도는 0.5 * (1 + cos(pi * t / T))로 조정되어 초기에는 높은 상태를 유지하고 마지막에는 거의 0에 가깝게 미끄러지는 부드러운 언덕을 생성합니다.
분산 추정치가 초기에 신뢰할 수 없기 때문에 특히 준비 작업을 통해 이점을 얻는 최적화 프로그램은 무엇입니까?
Adam의 실행 분산 추정은 첫 번째 단계에서 매우 적은 수의 샘플을 기반으로 하므로 유효 단계 크기가 불규칙합니다. 워밍업은 이를 완화합니다.
코사인 공식에서 T는 무엇을 나타냅니까?
T는 금리가 최고점에서 거의 0으로 감소하는 지평선입니다. t는 해당 지평선 내의 현재 단계입니다.
In the guide’s warmup-plus-cosine schedule, what begins after warmup reaches its target rate?
The guide describes a short linear warmup followed by cosine annealing.