PALS: LLM 가지치기를 위한 분위수 인식 계층별 희소성
요약
본 논문은 LLM의 효율적인 경량화를 위해 PALS (Percentile-Aware Layerwise Sparsity)라는 새로운 가지치기 기법을 제안합니다. PALS는 활성화 크기의 분위수를 기반으로 계층별 희소성을 조정하여, 기존 방법들보다 더 나은 성능과 낮은 퍼플렉서티를 달성함을 보여줍니다. 또한, 기울기 기반 할당 방식의 한계점도 제시했습니다.
핵심 포인트
- PALS는 활성화 분위수 기반으로 계층별 희소성을 조정합니다.
- LLaMA-2-7B에서 기존 방법 대비 퍼플렉서티 개선을 입증했습니다.
- 가지치기는 미세 조정 없이 적용 가능하며 비용이 적습니다.
- 기울기 기반 할당은 이산적 가중치 제거의 영향을 예측하기 어렵습니다.
Wanda나 SparseGPT 같은 원샷(one-shot) 가지치기 방법들은 트랜스포머의 모든 계층에 동일한 희소성 비율을 적용하며, 계층 중요도의 알려진 변화를 무시합니다. 우리는 PALS (Percentile-Aware Layerwise Sparsity)를 제안하는데, 이는 활성화 크기의 99번째 분위수(percentile)를 기반으로 계층별 희소성을 조정하며, 목표 비율 주변 $\pm 5%$로 제한됩니다. 50%의 희소성에서 LLaMA-2-7B에 적용했을 때, PALS는 균일한 Wanda (9회 평균, $p < 0.001$)가 달성하는 12.92 대비 10.96의 WikiText-2 퍼플렉서티를 달성합니다. 이 이점은 아키텍처에 따라 달라지는데, LLaMA-3-8B에서는 미미한 개선을 보였고 Mistral-7B에서는 아무런 개선도 보여주지 않았습니다. 또한 우리는 기울기 기반 할당(gradient-based allocation) — 겉보기에 더 원칙적인 접근 방식 — 이 무작위보다 나쁜 결과를 초래한다는 것을 발견했으며, 이는 기울기 크기가 이산적인 가중치 제거의 영향을 예측하지 못함을 시사합니다. PALS는 가지치기 파이프라인에 미미한 비용을 추가하며 미세 조정(fine-tuning)을 필요로 하지 않습니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 arXiv cs.CL (NLP)의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기