Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Breaking Memory Limits: Gradient Wavelet Transform Enhances LLMs Training

Created by
  • Haebom

저자

Ziqing Wen, Ping Luo, Jiahuan Wang, Xiaoge Deng, Jinping Zou, Kun Yuan, Tao Sun, Dongsheng Li

개요

본 논문은 대규모 언어 모델(LLM) 학습 시 메모리 효율적인 최적화를 위한 새로운 방법인 Gradient Wavelet Transform (GWT)를 제안합니다. 기존의 메모리 효율적인 알고리즘들은 특이값 분해 투영이나 가중치 고정과 같은 방법에 의존하여 성능 저하를 야기하지만, GWT는 웨이블릿 변환을 기울기(gradient)에 적용하여 메모리 요구량을 크게 줄이는 동시에 성능 저하 없이 효율적인 학습을 가능하게 합니다. 전처리 및 미세 조정 작업에 대한 광범위한 실험을 통해 GWT가 메모리 사용량과 학습 성능 측면에서 기존의 고급 메모리 효율적인 최적화 기법 및 전랭크 접근 방식보다 최첨단 성능을 달성함을 보여줍니다.

시사점, 한계점

시사점:
대규모 언어 모델 학습 시 메모리 제약을 극복하는 새로운 효율적인 방법 제시
기존의 저랭크 훈련 방식보다 우수한 성능과 메모리 효율성 달성
GWT는 Adam과 같은 메모리 집약적인 최적화 기법과 원활하게 통합 가능
전처리 및 미세 조정 작업 모두에서 우수한 성능 검증
한계점:
GWT의 일반화 성능에 대한 추가적인 연구 필요
다양한 LLM 아키텍처 및 데이터셋에 대한 추가적인 실험 필요
GWT의 계산 복잡도 및 구현의 어려움에 대한 평가 필요
👍