Daily Arxiv

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

ROMA: a Read-Only-Memory-based Accelerator for QLoRA-based On-Device LLM

Created by
  • Haebom
Category
Empty

저자

Wenqiang Wang, Yijia Zhang, Zikai Zhang, Guanting Huo, Hao Liang, Shijie Cao, Ningyi Xu

개요

본 논문은 에지 디바이스 상에서 대규모 언어 모델(LLM)을 배포하기 위한 QLoRA 가속기인 ROMA를 제안합니다. ROMA는 양자화된 기본 모델을 ROM에 저장하고, LoRA 가중치와 KV 캐시를 SRAM에 저장하는 하이브리드 저장 아키텍처를 사용합니다. 양자화된 기본 모델의 안정성을 고려하여 ROM에 저장하고, LoRA 모듈은 새로운 데이터에 대한 적응성을 제공합니다. 또한, ROM 면적 비용을 줄이기 위해 새로운 B-ROM 설계를 도입하고 컴퓨팅 유닛과 통합하여 효율적인 칩 자원 사용을 가능하게 합니다. ROMA는 4-bit 3B 및 2-bit 8B LLaMA 모델을 온칩으로 저장하고 20,000 토큰/초를 넘는 생성 속도를 달성합니다.

시사점, 한계점

시사점:
에지 디바이스에서의 LLM 배포를 위한 효율적인 하드웨어 가속기 설계를 제시.
ROM과 SRAM의 하이브리드 저장 아키텍처를 통해 메모리 사용량 및 전력 소모 감소.
B-ROM 설계를 통해 ROM 면적 비용 감소 및 성능 향상.
높은 토큰 생성 속도 달성 (20,000 토큰/초 이상).
한계점:
제안된 아키텍처의 구체적인 에너지 효율성에 대한 분석 부족.
다양한 LLM 모델 및 크기에 대한 적용 가능성에 대한 추가적인 실험 필요.
B-ROM 설계의 일반화 가능성 및 다른 아키텍처와의 비교 분석 필요.
👍