Daily Arxiv

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

DLP: Dynamic Layerwise Pruning in Large Language Models

Created by
  • Haebom

저자

Yuli Chen, Bo Cheng, Jiale Han, Yingying Zhang, Yingting Li, Shuhao Zhang

개요

본 논문은 대규모 언어 모델(LLM)의 매개변수 규모를 줄이고 추론 효율을 높이기 위한 새로운 가지치기 방법인 동적 계층별 가지치기(Dynamic Layerwise Pruning, DLP)를 제안합니다. 기존의 균일한 계층별 가지치기 방법은 높은 스파스성 수준에서 성능 저하를 초래하는 반면, DLP는 모델 가중치와 입력 활성화 정보를 통합하여 각 계층의 상대적 중요도를 적응적으로 결정하고, 그에 따라 가지치기 비율을 할당합니다. 실험 결과, DLP는 여러 LLM에서 높은 스파스성 수준에서도 모델 성능을 효과적으로 유지하는 것으로 나타났습니다. 특히, 70% 스파스성에서 LLaMA2-7B의 perplexity를 7.79 감소시키고 평균 정확도를 2.7% 향상시켰으며, 기존의 다양한 LLM 압축 기술과 호환되고 PEFT에도 통합 가능합니다. 소스 코드는 공개되었습니다.

시사점, 한계점

시사점:
기존의 균일한 계층별 가지치기의 한계를 극복하고 높은 스파스성에서도 성능 저하를 최소화하는 효과적인 LLM 가지치기 방법 제시.
LLaMA2-7B를 포함한 여러 LLM에서 우수한 성능 향상을 실험적으로 검증.
기존 LLM 압축 기술 및 PEFT와의 호환성을 확보하여 실용성 증대.
소스 코드 공개를 통한 향후 연구 활성화 기여.
한계점:
DLP의 성능 향상이 특정 LLM 및 데이터셋에 국한될 가능성.
다른 가지치기 방법들과의 비교 분석이 더욱 심도 있게 필요.
DLP의 계산 비용 및 메모리 사용량에 대한 자세한 분석 부족.
👍