vLLM에서 양방향 KV 전송으로 다중 턴 작업의 TTFT 개선
Improve TTFT for multi-turn agentic workloads with bi-directional KV transfer in vLLM
다중 턴 에이전트 작업은 누적된 컨텍스트를 기반으로 하며, 각 턴마다 클라이언트는 대화 기록을 다시 전송한다. 기존의 분산 서비스는 KV 캐시를 한 방향으로만 전송하여 비효율이 발생한다. 양방향 KV 전송을 통해 프리필 노드와 디코드 노드가 이전에 계산된 컨텍스트를 재사용할 수 있어, TTFT를 Qwen3-32B에서 최대 1.6배, zai-org/GLM-5-FP8에서 최대 3.3배 감소시킨다. 이 기능은 AWS Trainium 인스턴스 클러스터에서 검증되었으며, AWS EFA를 통해 GPU 간 KV 블록을 직접 이동할 수 있다.
양방향 KV 전송은 프리필 노드가 이전 턴에서 생성된 KV를 재사용할 수 있게 하여, 계산 중복을 줄이고 TTFT를 개선한다.
원문 출처
AWS Machine Learning Blog