StreamDQ: 맞춤형 HBM에서의 근접 메모리 가중치 디양자화(Weight DeQuantization)를 통한 확장 가능한 AI 추론 가속
요약
본 논문은 대규모 언어 모델(LLMs)의 메모리 및 연산 요구 증가에 따른 병목 현상을 해결하기 위한 아키텍처 개선 방안인 StreamDQ를 제안합니다. StreamDQ는 HBM 베이스 다이에 DeQuantization Blocks (DQBs)를 통합하여, 온디바이스 디양자화를 인라인으로 수행함으로써 고처리량 LLM 추론 가속을 목표로 합니다.
핵심 포인트
- LLMs의 메모리 및 연산 요구 증가가 주요 병목 현상을 초래함.
- 기존 GPU 기반 디양자화는 명령어 오버헤드와 트래픽 문제를 야기함.
- StreamDQ는 HBM 베이스 다이에 DQBs를 통합하여 온디바이스 디양자화를 구현함.
- 이를 통해 고처리량 대규모 배치 LLM 추론이 가능해짐.
대규모 언어 모델(LLMs)이 규모를 키우면서, 이들의 메모리 및 연산 요구 사항은 크게 증가했습니다. 이에 따라 가중치만 양자화하는 기법은 정확도 손실을 최소화하면서 모델 크기를 줄이는 널리 채택된 기술이 되었습니다. 하지만 현재 GPU에서 CUDA 코어 기반의 디양자화(dequantization)는 상당한 명령어 오버헤드, 온칩 트래픽, 파이프라인 정지(pipeline stalls)를 유발하여, 고처리량 클라우드 규모 LLM 서비스에 주요 병목 현상을 초래합니다. 이러한 한계를 해결하기 위해, 우리는 StreamDQ라는 경량 아키텍처 개선 방안을 제안합니다. 이는 메모리 서브시스템에서 온디바이스(on-the-fly) 디양자화를 가능하게 하여 고처리량 대규모 배치 LLM 추론을 수행할 수 있게 합니다. StreamDQ는 고대역폭 메모리(HBM)의 베이스 다이(base die)에 컴팩트한 DeQuantization Blocks (DQBs)를 통합하고 인라인으로 수행합니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 arXiv cs.AR의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기