[공지사항]을 빙자한 안부와 근황 
Show more

Daily Arxiv

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

Gradual Binary Search and Dimension Expansion : A general method for activation quantization in LLMs

Created by
  • Haebom

저자

Lucas Maisonnave, Cyril Moineau, Olivier Bichler, Fabrice Rastello

개요

본 논문은 대규모 언어 모델(LLM)의 효율적인 양자화를 위한 새로운 방법을 제시합니다. LLM의 크기가 크다는 점을 감안하여 메모리 사용량과 추론 시간을 줄이기 위해 Hadamard 행렬을 기반으로 한 양자화 기법을 제안합니다. Hadamard 행렬은 활성화 값의 이상치를 줄이는 데 효과적이며, 이를 통해 기존 방법들보다 성능이 향상된 3-bit 양자화를 가능하게 합니다. 특히, 가중치, 활성화 값, 그리고 key-value(KV) 캐시에 모두 3-bit 양자화를 적용하여 Mistral, LLaMA, Qwen과 같은 다양한 모델에서 기존 최고 성능(SoTA) 방법 대비 40% 향상된 정확도를 달성했습니다. 또한, Paley 알고리즘을 이용하여 2의 거듭제곱이 아닌 임베딩 차원도 지원하도록 확장했습니다.

시사점, 한계점

시사점:
Hadamard 행렬 기반 양자화를 통해 LLM의 3-bit 양자화를 실현하여 메모리 사용량과 추론 시간을 효과적으로 감소시켰습니다.
기존 SoTA 방법 대비 40% 향상된 정확도를 여러 모델에서 달성했습니다.
2의 거듭제곱이 아닌 임베딩 차원을 지원하는 확장성을 보여주었습니다.
활성화 값의 이상치 감소에 Hadamard 행렬의 이점을 이론적으로 증명했습니다.
한계점:
제시된 방법의 효과는 특정 모델과 벤치마크에 국한될 수 있습니다. 다양한 모델과 벤치마크에 대한 추가적인 실험이 필요합니다.
3-bit 양자화 이하의 더 낮은 bit 양자화에 대한 성능은 추가 연구가 필요합니다.
Paley 알고리즘을 이용한 비 2의 거듭제곱 차원 지원의 효율성에 대한 더 자세한 분석이 필요할 수 있습니다.
👍