Teaching Pretrained Language Models to Think Deeper with Retrofitted Recurrence
Created by
Haebom
Category
Empty
저자
Sean McLeish, Ang Li, John Kirchenbauer, Dayal Singh Kalra, Brian R. Bartoldson, Bhavya Kailkhura, Avi Schwarzschild, Jonas Geiping, Tom Goldstein, Micah Goldblum
개요
최근 딥-재귀 언어 모델의 발전은 재귀가 훈련 시간 계산과 매개변수 수를 테스트 시간 계산에서 분리할 수 있음을 보여줍니다. 본 연구에서는 기존의 사전 훈련된 비-재귀 언어 모델을 딥-재귀 모델로 변환하는 방법을 연구합니다. 훈련 과정에서 재귀의 커리큘럼을 사용하여 모델의 유효 깊이를 증가시키면 성능을 유지하면서 총 계산 비용을 줄일 수 있음을 발견했습니다. 수학 실험에서 사전 훈련된 모델을 재귀 모델로 변환하면 주어진 계산 예산에서 원래의 비-재귀 언어 모델을 단순히 사후 훈련하는 것보다 더 나은 성능을 얻을 수 있음을 확인했습니다.