TriRoute: 조건부 계산을 통한 언어 모델 최적화
TriRoute: Unified Learned Routing for Joint Adaptive Attention, Experts, and KV-Cache Allocation
TriRoute는 언어 모델의 품질과 토큰당 추론 비용을 분리할 수 있는 조건부 계산 기법을 제안한다. 이 시스템은 주의 모드, 희소 FFN 전문가 집합, KV-cache 비트 폭을 조정하는 단일 경량 컨트롤러를 사용하여 각 토큰에 대해 조정된 정책을 생성한다. TriRoute는 160M에서 1.3B 파라미터의 디코더 전용 모델에서 MoD, MoE, KV-양자화의 조합보다 더 나은 성능을 보이며, 희귀 개체와 코드, 산술 문제에 대한 강인성을 유지한다.
TriRoute는 Lagrangian 예산 제약 하에 평균 계산 및 메모리 비용을 조절할 수 있는 기능을 제공한다.
원문 출처
arXiv cs.LG (머신러닝)