Daily Arxiv

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

ASPD: Unlocking Adaptive Serial-Parallel Decoding by Exploring Intrinsic Parallelism in LLMs

Created by
  • Haebom

저자

Keyu Chen, Zhifeng Shen, Daohai Yu, Haoqian Wu, Wei Wen, Jianfeng He, Ruizhi Qiao, Xing Sun

개요

본 논문은 대규모 언어 모델(LLM)의 추론 지연 시간 문제를 해결하기 위해 자기 회귀적 디코딩 방식의 병렬 처리 가능성에 초점을 맞춘 연구입니다. 자기 회귀 모델의 출력에서 병렬 처리 가능한 구조(intrinsic parallelism)를 발견하고, 이를 활용하여 병렬 디코딩을 수행하는 적응형 직렬-병렬 디코딩(ASPD) 기법을 제안합니다. ASPD는 병렬 처리 가능한 데이터 구조를 자동으로 추출하고 검증하는 파이프라인과 직렬 및 병렬 디코딩 모드 간의 원활한 전환을 가능하게 하는 하이브리드 디코딩 엔진으로 구성됩니다. 다양한 작업(일반 작업, 검색 증강 생성, 수학적 추론)에 대한 실험 결과, ASPD는 효율성과 효과성 측면에서 기존 방식보다 뛰어난 성능을 보였으며, Vicuna Bench에서 평균 1.85배(최대 3.19배)의 속도 향상을 달성하면서 응답 품질 저하를 1% 이내로 유지했습니다.

시사점, 한계점

시사점:
LLM의 추론 속도를 획기적으로 향상시키는 새로운 병렬 디코딩 기법 제시.
자동화된 병렬 구조 추출 및 효율적인 병렬 디코딩 메커니즘을 통한 실질적인 성능 개선.
AI 기반 고객 서비스 봇 및 답변 검색 엔진과 같은 지연 시간에 민감한 애플리케이션에 대한 LLM 배포 가능성 확대.
Vicuna Bench 실험 결과를 통해 효과성과 효율성을 검증.
한계점:
제안된 ASPD 기법의 일반화 성능 및 다양한 LLM 아키텍처에 대한 적용 가능성에 대한 추가 연구 필요.
병렬 처리 가능한 구조의 자동 추출 정확도 및 효율성 개선에 대한 지속적인 연구 필요.
특정 벤치마크(Vicuna Bench)에 대한 결과 위주로, 다른 벤치마크나 실제 응용 환경에서의 성능 검증이 필요.
👍