Daily Arxiv

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

Pre-Attention Expert Prediction and Prefetching for Mixture-of-Experts Large Language Models

Created by
  • Haebom
Category
Empty

저자

Shien Zhu, Samuel Bohl, Robin Oester, Gustavo Alonso

개요

Mixture-of-Experts (MoE) Large Language Models (LLMs)의 효율적인 확장을 위해, 본 논문은 정확하고 가벼운 전문가 사전 페치를 위한 pre-attention expert prediction 기법을 제안한다. 기존 방법의 낮은 정확도와 첫 번째 레이어 최적화 부재 문제를 해결하기 위해, 간단한 선형 함수와 ranking-aware loss를 활용하여 attention 블록 이전의 활성화를 기반으로 전문가를 예측한다. 제안하는 기법은 DeepSeek V2 Lite, Qwen3-30B, Phi-mini-MoE 모델에서 93.03%, 94.69%, 97.62%의 정확도를 달성하며, 기존 최고 기법보다 약 15%의 절대 정확도 향상을 보였다.

시사점, 한계점

시사점:
MoE LLMs의 추론 속도 향상을 위한 효율적인 전문가 예측 기법 제시.
기존 방법 대비 높은 정확도를 달성하여 성능 개선.
계산 오버헤드 없이 가볍게 구현 가능.
첫 번째 레이어에서도 전문가 사전 페치 가능.
한계점:
논문에 구체적인 한계점에 대한 언급은 없음.
👍