Daily Arxiv

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

Block: Balancing Load in LLM Serving with Context, Knowledge and Predictive Scheduling

Created by
  • Haebom

저자

Wei Da, Evangelia Kalyvianaki

개요

본 논문은 대규모 언어 모델 서빙 프레임워크에서 인스턴스 간의 부하 분산 및 자동 프로비저닝을 최적화하기 위해 수신 요청의 상황 정보를 활용하는 분산형 스케줄링 프레임워크인 Block을 제시합니다. 기존의 단일적이고 휴리스틱한 작업 스케줄러에 의존하는 모델 서빙 시스템과 달리, Block은 완전히 분산되고 상태 비저장이며 예측적인 스케줄링 시스템으로 작동하여 오버헤드가 낮고 안정적이며 확장 가능합니다. 호스트 구성, 응답 길이, 하드웨어 성능과 같은 LLM 추론의 결정적이고 예측 가능한 특성을 활용하여 정확하게 예측된 메트릭을 기반으로 스케줄링 결정을 내립니다. 12개의 GPU 클러스터에 대한 평가 결과, Block은 휴리스틱 스케줄러보다 훨씬 뛰어난 성능을 보여주며, 서빙 용량을 최대 16.7% 향상시키고 P99 지연 시간을 최대 49.5% 단축시킵니다. 이러한 성능 향상은 다양한 모델, 작업량 및 구성에서 일관되게 유지됩니다. 코드와 데이터는 오픈소스로 공개됩니다.

시사점, 한계점

시사점:
대규모 언어 모델 서빙 시스템의 성능을 크게 향상시킬 수 있는 새로운 분산형 스케줄링 프레임워크를 제시합니다.
부하 분산 및 자동 프로비저닝을 효율적으로 수행하여 서빙 용량을 증가시키고 지연 시간을 감소시킵니다.
LLM 추론의 특성을 활용하여 정확한 예측 기반 스케줄링을 가능하게 합니다.
오픈소스로 공개되어 다른 연구자들이 활용할 수 있습니다.
한계점:
현재 12개의 GPU 클러스터에서만 평가되었으므로, 더 큰 규모의 클러스터에서는 성능이 어떻게 변화할지 추가적인 연구가 필요합니다.
다양한 모델과 작업량에 대한 평가를 수행했지만, 모든 유형의 LLM과 작업량에 대해 일반화할 수 있는지 추가적인 검증이 필요합니다.
실제 운영 환경에서의 장기적인 안정성과 확장성에 대한 추가적인 평가가 필요합니다.
👍