자체 호스팅 Gemma 2 9B와 프론티어(Frontier) API 벤치마킹: NVIDIA L4에서의 FP8 양자화(Quantization)

상용 클라우드 API에서 프로덕션 LLM 워크로드를 이전하는 것을 평가할 때, 논의는 대개 품질과 인프라 비용 사이의 트레이드오프(trade-off)로 지나치게 단순화되곤 합니다. 깔끔하고 고립된 평균치를 넘어 살펴보기 위해, 저는 실제 워크로드인 제 이력서 생성 플랫폼을 위한 콜드 아웃리치(cold outreach) 및 문맥적 프로필 재설계(contextual profile re-engineering)를 사용하여 반복 가능한 평가 매트릭스를 구축했습니다.

저는 단일 범용 NVIDIA L4 GPU에서 vLLM을 통해 서빙되는 최적화된 FP8 변형 모델과 양자화되지 않은(unquantized) Gemma 2 9B를 벤치마킹했습니다.
데이터셋은 다양한 수신자 페르소나, 다양한 복잡도 범주(짧은 문맥에서 긴 문맥까지), 그리고 엄격한 정수 격식(integer formality) 파라미터를 가로지르는 동적 텍스트 생성을 평가합니다. 저는 FP8 압축이 런타임 실태를 어떻게 변화시키는지 감사하기 위해 클라이언트 측 및 서버 측 텔레메트리(telemetry)를 캡처했습니다.
기본 평가 세트는 rsher60/resume-gen-benchmark에 공개되어 있습니다. 제가 발견한 원시 텔레메트리와 인프라 트레이드오프(trade-off)는 다음과 같습니다.

1. 첫 번째 토큰 생성 시간 (TTFT): 양자화(Quantization)의 숨겨진 Prefill 비용
   지배적인 오픈 소스 내러티브는 FP8 양자화(Quantization)가 모든 것을 더 빠르게 만든다는 것입니다. 하지만 여러분의 애플리케이션이 매우 상호작용적이고 UI로 스트리밍되는 방식이라면, TTFT는 사용자가 체감하는 속도를 결정하는 유일한 지표입니다.
   제 텔레메트리는 전형적인 하드웨어-소프트웨어 트레이드오프(trade-off)를 드러냈습니다:

Prefill 페널티: 높은 복잡도의 페르소나를 대상으로 하는 복잡하고 긴 문맥의 프롬프트에 대해, 양자화되지 않은 모델은 서버에 866.93ms 만에 토큰을 반환했습니다. FP8 변형 모델은 1372.12ms로 급증했으며, 이는 초기 prefill 단계에서 58%의 지연 시간(latency) 페널티가 발생했음을 의미합니다.
이러한 현상이 발생하는 이유: 양자화(Quantization)는 토큰 생성(디코딩 단계) 중 메모리 대역폭 병목 현상을 줄여줍니다. 그러나 연산 집약적인(compute-bound) heavy prefill 단계 동안 발생하는 행렬 곱셈 역양자화(matrix-multiplication de-quantization) 오버헤드는 L4와 같은 연산 집약적인 범용 하드웨어에서 실행될 때 긴 입력 토큰에 대해 눈에 띄는 비용(tax)을 발생시킵니다.
프로덕션 엣지 케이스(Edge Cases): 저는 짧은 문맥 실행 중 FP8 모델에서 1,740.34ms에 달하는 거대한 TTFT 급증을 포착했습니다.

이는 vLLM 스케줄링 하에서의 실제 인프라 현실—예를 들어 콜드 프리필(cold prefill)이나 컨텍스트 블록 스와핑(context block swapping) 등—을 반영합니다. 이는 순수하게 깨끗하고 격리된 평균값만으로는 아키텍처를 평가할 수 없음을 증명합니다.

2. 엔드 투 엔드 지연 시간(End-to-End Latency): FP8이 생성(Generation) 전쟁에서 승리하는 지점
   FP8은 프리필(prefill) 단계에서 일종의 세금(tax)을 지불하게 만들지만, LLM이 메모리 대역폭 제한(memory-bandwidth bound)을 심하게 받는 정상 상태의 디코딩 루프(decoding loop) 동안에는 그 가치를 공격적으로 증명합니다.

가중치 정밀도(weight precision)를 8비트 정수(8-bit integers)로 낮춤으로써, GPU 메모리 버스를 통해 이동하는 데이터의 양이 대략 절반으로 줄어듭니다. 중간 길이의 생성 시퀀스의 경우, 평균 클라이언트 총 시간(average client total time)이 12,290.2ms에서 11,526.2ms로 감소했습니다. 만약 귀하의 애플리케이션이 중간에서 짧은 컨텍스트 크기를 처리하거나 완전히 비동기/배치(asynchronous/batch) 작업을 수행한다면, FP8은 명확하고 결정론적인 인프라 측면의 이점을 제공합니다.

3. 품질 장부(The Quality Ledger): 9B 파라미터가 성능을 유지했는가?
   저는 양자화되지 않은 원본 실행 결과와 FP8 모델 출력 결과(rsher60/resume-gen-benchmark-results)를 비교하여 생성된 출력물을 검증했습니다.

스키마 및 페르소나 준수(Schema & Persona Adherence): 고정된 개인 프로필을 기반으로 텍스트를 맞춤화하는 것과 같은 타겟팅된 단일 턴(single-turn) 작업의 경우, 정교하게 설계된 시스템 프롬프트(system prompt)를 통해 9B 아키텍처가 프론티어(frontier) 모델과 거의 동일한 포맷팅 및 페르소나 충실도로 실행됨을 보장합니다.
의미론적 드리프트(Semantic Drift): 좁고 도메인 특화된 작업의 경우, FP8 양자화(quantization)는 사실상 무시할 수 있는 수준의 의미론적 드리프트를 유발했습니다. 모델은 훨씬 낮은 메모리 점유율(memory footprint) 내에서 실행되면서도, 엔지니어를 대상으로 한 콜드 아웃리치(cold outreach)와 공식 지원서 사이의 톤을 맞추는 것과 같이 복잡한 컨텍스트 키(context keys)를 성공적으로 유지했습니다.

전략적 아키텍처 시사점

대화형/저배치/긴 입력(Interactive/Low-Batching/Long Inputs): TTFT를 보호하고 사용자 UI 마찰을 방지하려면 양자화되지 않은 가중치(unquantized weights)나 매우 공격적인 비청크(unchunked) 프리필 전략이 필요할 수 있습니다.
비동기/스트리밍/중단기 컨텍스트(Asynchronous/Streaming/Short-to-Medium Context): FP8은 절대적인 필수 사항입니다.

L4에서 FP8을 사용하는 진짜 이유는 총 지연 시간(total latency)을 몇 백 밀리초 절약하는 것만이 아닙니다. 바로 VRAM 확보입니다. 모델의 크기(footprint)를 줄임으로써 KV Cache에 엄청난 양의 메모리를 확보할 수 있고, 이를 통해 Out-Of-Memory (OOM) 예외 없이 동시성(concurrency)을 확장할 수 있습니다.

저는 vLLM 설정과 캐시 할당 전략을 포함하여 전체 분석을 정리했습니다. 이 전략들을 사용하여 높은 동시 부하(heavy concurrent load) 하에서도 92.7%의 KV Cache 활용률을 유지하는 방법을 자세한 글에 담았습니다:
https://billionars.substack.com/p/benchmarking-my-self-hosted-gemma
Hugging Face (HF) 데이터셋은 여기입니다:

submitted by /u/Ok_Waltz_5145 to r/MachineLearning [link] [comments]