Daily Arxiv

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

Key and Value Weights Are Probably All You Need: On the Necessity of the Query, Key, Value weight Triplet in Decoder-Only Transformers

Created by
  • Haebom

저자

Marko Karbevski, Antonij Mijoski

개요

본 논문은 최첨단 LLM의 어텐션 메커니즘을 구성하는 쿼리, 키, 값 가중치 삼중항의 축소 가능성을 이론적으로 연구합니다. 단순화된 가정을 통해 쿼리 가중치가 불필요하다는 것을 증명하여 임베딩/lm-head를 제외한 파라미터 수를 8% 이상 줄일 수 있음을 보였습니다. GPT-3 small 아키텍처(레이어 정규화, 스킵 연결 및 가중치 감쇠 포함)를 처음부터 학습하여 이론을 검증했으며, 축소된 모델이 표준 기준선과 유사한 검증 손실을 달성함을 확인했습니다.

시사점, 한계점

쿼리 가중치 불필요성 이론 제시 및 검증.
모델 파라미터 수 감소 가능성 제시 (8% 이상).
GPT-3 small 아키텍처에서 검증 완료.
대규모 모델에서의 쿼리 가중치 불필요성 연구 필요.
단순화된 가정 하에 이론 증명.
GPT-3 small 아키텍처에 한정된 검증.
👍