본문으로 건너뛰기

OCTOPUS: 최적화된 제곱 오차 양자화를 통한 Transformer KV 캐시 최적화

OCTOPUS는 팔면체 매개변수화와 최적화된 제곱 오차 양자화를 사용하여 Transformer의 KV 캐시를 효율적으로 압축하는 데이터 독립적 코덱이다.

섹션별 상세

기존 KV 캐시 압축 방식인 회전 사전 조건 코덱을 개선하여 좌표 삼중항을 팔면체 매개변수화로 매핑하고 비트 할당을 최적화한다. 각 삼중항의 방향을 정사각형으로 매핑한 뒤, 두 좌표와 삼중항 노름을 Lloyd-Max 양자화하여 비트 할당을 수행한다. 데이터 독립적이고 결정론적인 알고리즘으로, 텍스트·비디오·오디오 등 다양한 도메인에서 기존 코덱 대비 우수한 압축 성능을 입증했다.
Triton 기반의 커널 구현을 통해 압축 해제 과정에서 추가적인 대역폭이나 지연 시간 없이 키를 즉석에서 재구성하여 추론 효율을 높인다. 이 방식은 압축된 키를 메모리에서 읽어 즉시 복원하므로, 기존의 dequantization 과정에서 발생하는 오버헤드를 제거한다. 결과적으로 긴 컨텍스트를 처리하는 모델의 메모리 대역폭 병목을 효과적으로 해결한다.

기술

  • OCTOPUS
  • Transformers
  • Triton
  • Lloyd-Max Quantization

활용 사례

  • LLM 추론 최적화
  • 긴 컨텍스트 처리
  • 메모리 대역폭 절감
AI 분석 전체 내용 보기

AI 요약 · 북마크 · 개인 피드 설정 — 무료

출처 · 인용 안내

원문 발행 2026. 06. 04.수집 2026. 06. 04.출처 타입 RSS

인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.