본문으로 건너뛰기
Zamba2-VL 해설: 엣지 VLM은 모델 크기보다 첫 토큰 지연·KV 캐시·시각 토큰 예산을 먼저 설계해야 하는 이유
← 블로그로 돌아가기

Zamba2-VL 해설: 엣지 VLM은 모델 크기보다 첫 토큰 지연·KV 캐시·시각 토큰 예산을 먼저 설계해야 하는 이유

개발정보·10분

AI타임스의 Zamba2-VL 보도를 출발점으로, 하이브리드 SSM-트랜스포머 VLM이 왜 엣지·온디바이스 추론에서 TTFT, KV 캐시, 시각 토큰 예산 문제를 다시 보게 만드는지 정리했습니다.

Zamba2-VL 하이브리드 SSM 트랜스포머 엣지 VLM 추론 대표 이미지
Zamba2-VL의 핵심은 정확도 경쟁보다 긴 시각 입력에서 첫 토큰 지연, 메모리, 시각 토큰 예산을 다시 설계하게 만든다는 점입니다.

1. 한 줄 문제 정의

핵심 요약: 이미지를 이해하는 AI를 제품에 넣을 때 병목은 정확도 하나가 아니라 첫 응답 지연, 메모리, 시각 토큰 수입니다.

AI타임스는 2026년 6월 16일 지프라(Zyphra)가 맘바2 상태공간 레이어와 트랜스포머 블록을 결합한 오픈소스 비전-언어모델 Zamba2-VL을 공개했다고 보도했습니다. VLM은 Vision-Language Model의 줄임말입니다. 쉽게 말하면 사진, 문서, 차트 같은 이미지를 텍스트 질문과 함께 읽고 답하는 모델입니다.

이 글의 문제는 "Zamba2-VL이 모든 VLM을 대체하는가"가 아닙니다. 진짜 문제는 고해상도 이미지, PDF, 차트, 다중 이미지를 넣을수록 트랜스포머 기반 VLM의 프리필(prefill)과 KV 캐시 비용이 빠르게 커진다는 점입니다. 반대로 서버 GPU가 넉넉하고 최고 정확도만 중요한 연구 실험이라면 이 글의 판단 기준이 우선순위가 아닐 수 있습니다.

2. 먼저 결론

핵심 요약: Zamba2-VL은 리더보드 1등 모델이 아니라, 긴 시각 입력을 빠르게 읽어야 하는 제품형 VLM 후보로 봐야 합니다.

도입 검토 대상은 온디바이스 AI, 산업용 엣지 장비, 영수증·청구서 OCR, 재고 카운팅, 현장 점검 앱처럼 이미지를 넣고 바로 응답해야 하는 팀입니다. 특히 1.2B와 2.7B 크기는 모바일·엣지 배포에서 "정확도 조금 더"보다 "첫 응답이 언제 나오는가"가 더 중요한 경우 의미가 있습니다.

비추천 대상도 분명합니다. 복잡한 수학 추론, 범용 지식 추론, 최고급 멀티모달 reasoning이 핵심이면 더 큰 트랜스포머 VLM이나 상용 API가 나을 수 있습니다. Hugging Face 모델 카드의 벤치마크에서도 Zamba2-VL-2.7B는 DocVQA와 카운팅은 강하지만 MathVista, MMMU, BLINK 같은 추론 벤치마크에서는 더 큰 경쟁 모델에 밀리는 항목이 있습니다.

3. 핵심 구조 분해

핵심 요약: Zamba2-VL의 구조는 비전 인코더, MLP 어댑터, 하이브리드 SSM-트랜스포머 언어 백본 세 층으로 나뉩니다.

Zamba2-VL은 낯선 구조처럼 보이지만, 큰 흐름은 LLaVA 계열 VLM과 비슷합니다. 이미지를 먼저 작은 특징 조각으로 바꾸고, 그 특징을 언어모델이 읽을 수 있는 임베딩 공간으로 옮긴 뒤, 텍스트 질문과 함께 처리합니다.

  • 비전 인코더: 이미지에서 패치 특징을 추출합니다. AI타임스와 Hugging Face 모델 카드는 Qwen2.5-VL 계열 비전 인코더를 사용한다고 설명합니다.
  • 2계층 MLP 어댑터: 이미지 특징을 언어모델 입력 공간으로 투영합니다. 초보 개발자 기준으로는 "이미지 언어를 텍스트 모델이 읽는 언어로 번역하는 연결부"입니다.
  • Zamba2 백본: 대부분의 처리는 Mamba2 상태공간 레이어가 맡고, 일부 트랜스포머 어텐션 블록을 섞습니다.
  • 공유 어텐션 블록: 매 레이어마다 큰 어텐션을 새로 두지 않고 일부 블록을 공유해 매개변수와 메모리 부담을 줄입니다.

핵심은 "트랜스포머를 완전히 버린다"가 아닙니다. 순수 상태공간 모델은 긴 입력에서 특정 정보를 다시 집어내는 능력이 약할 수 있습니다. Zamba2-VL은 이 약점을 보완하려고 필요한 지점에 어텐션을 남깁니다.

4. 설계 의도 해설

핵심 요약: 설계 의도는 긴 시각 입력의 프리필 비용을 낮추면서, 어텐션의 검색 능력은 최소한으로 보존하는 것입니다.

트랜스포머는 입력 토큰이 길어질수록 어텐션 계산과 KV 캐시가 커집니다. 텍스트만 볼 때도 문제지만, VLM에서는 이미지 한 장이 수천 개의 시각 토큰으로 바뀔 수 있습니다. PDF 여러 장, 차트, 고해상도 사진을 넣으면 첫 토큰이 나오기 전 모델이 입력을 읽는 시간이 길어집니다.

Zamba2-VL의 선택은 대담하지만 현실적입니다. Mamba2 계열 상태공간 레이어로 대부분의 토큰 처리를 선형 시간에 가깝게 만들고, 트랜스포머 어텐션은 일부 공유 블록으로 제한합니다. 얻는 것은 첫 토큰 지연과 메모리 효율입니다. 잃는 것은 순수 대형 트랜스포머가 가진 범용 추론 여유와 생태계 호환성 일부입니다.

그래서 이 모델은 "더 작은 모델인데 모든 점수가 더 높다"가 아니라 "긴 입력을 읽는 제품에서 비용 구조가 달라진다"는 관점으로 봐야 합니다. 저는 이 차이가 엣지 VLM에서는 꽤 중요하다고 봅니다.

5. 근거 및 비교

핵심 요약: 비교 기준은 평균 점수만이 아니라 TTFT, KV 캐시, 문서/OCR 강점, 추론 약점까지 함께 봐야 합니다.

접근 강점 약점 맞는 사용처
순수 트랜스포머 VLM 성숙한 생태계, 강한 범용 추론, 다양한 배포 옵션 긴 시각 토큰에서 프리필과 KV 캐시 비용 증가 서버 GPU 기반 고정밀 분석, 복잡한 reasoning
순수 SSM VLM 긴 입력 처리 비용을 낮추기 쉬움 문맥 내 검색과 세부 정보 회수에서 약해질 수 있음 정해진 패턴의 장문·영상 입력 실험
Zamba2-VL 하이브리드 SSM 효율과 일부 어텐션 검색 능력을 절충 최고 추론 점수와 표준 런타임 호환성은 아직 확인 필요 엣지 OCR, 카운팅, 문서·차트 질의응답

AI타임스 보도에 따르면 Zamba2-VL은 1.2B, 2.7B, 7B 세 크기로 공개됐고, 3만2000 토큰 입력 환경에서 동급 트랜스포머 기반 모델 대비 최소 10배 낮은 첫 토큰 생성 시간(TTFT)을 기록했다고 소개됐습니다. arXiv 기술 보고서도 같은 방향으로, Zamba2-VL이 동급 오픈 VLM과 경쟁 가능한 정확도를 유지하면서 TTFT를 대략 한 자릿수 배율로 낮춘다고 설명합니다.

Hugging Face의 Zamba2-VL-2.7B 모델 카드 기준으로는 DocVQA 90.9, CountBenchQA 87.5, PixMoCount 82.5를 기록했습니다. 반면 MathVista 51.0, MMMU 37.7은 더 큰 모델과 비교할 때 약점으로 남습니다. 즉 이 모델의 강점은 "모든 문제를 더 잘 푸는 것"이 아니라 "문서·카운팅·OCR 계열에서 빠른 응답 비용을 노려볼 수 있는 것"입니다.

6. 실제 동작 흐름 / 단계별 실행 방법

핵심 요약: 바로 운영에 넣지 말고, 동일 이미지 세트에서 TTFT, 메모리, 실패 유형을 먼저 측정해야 합니다.

Hugging Face 모델 카드의 빠른 시작은 Zyphra의 transformers 브랜치와 Qwen VLM 유틸, Mamba2 커널 의존성을 요구합니다. 테스트 환경은 운영 서버와 분리한 CUDA 머신에서 시작하는 편이 안전합니다.

python -m venv .venv
source .venv/bin/activate

pip install qwen-vl-utils==0.0.2 flash_attn

# Zyphra 모델 카드의 Quick start 기준으로
# transformers zamba2-vl 브랜치, causal-conv1d, mamba-ssm 커널을 설치합니다.
# 운영 문서에는 실제 커밋 해시와 wheel 호환성을 고정해 두는 것이 좋습니다.

최소 검증 코드는 다음 순서로 잡습니다. 실제 제품 검토에서는 모델 출력 내용보다 먼저 응답 시간과 메모리부터 기록해야 합니다.

from time import perf_counter
import torch
from PIL import Image
from transformers import Zamba2_VLForConditionalGeneration, Zamba2_VLProcessor

device = "cuda"
model_id = "Zyphra/Zamba2-VL-2.7B"
processor = Zamba2_VLProcessor.from_pretrained(model_id, temporal_patch_size=1)
model = Zamba2_VLForConditionalGeneration.from_pretrained(
    model_id,
    device_map=device,
    torch_dtype=torch.bfloat16,
    attn_implementation="flash_attention_2",
)

image = Image.open("invoice-sample.jpg")
question = "이 이미지에서 청구 금액, 공급자명, 날짜를 표로 정리해줘."

conversation = [{
    "role": "user",
    "content": [
        {"type": "image", "image": image, "max_pixels": 3400 * 28 * 28, "min_pixels": 10 * 28 * 28},
        {"type": "text", "text": question},
    ],
}]

prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
inputs = processor(text=prompt, images=[image], add_special_tokens=True, return_tensors="pt")
inputs = {k: v.to(device) for k, v in inputs.items()}

t0 = perf_counter()
with torch.inference_mode():
    output = model.generate(**inputs, max_new_tokens=160)
torch.cuda.synchronize()
print("latency_sec", round(perf_counter() - t0, 3))
print(processor.tokenizer.decode(output[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True))

운영 후보를 판단할 때는 같은 이미지 100장으로 세 가지를 남기면 됩니다. 첫째, 첫 응답까지 걸린 시간. 둘째, GPU 메모리 피크. 셋째, 사람이 다시 봐야 하는 오류 유형입니다. 이 세 값이 없으면 "빠르다"는 말은 제품 의사결정에 쓸 수 없습니다.

7. 실수/함정(Pitfalls)

핵심 요약: Zamba2-VL 도입 실패는 모델 성능보다 커널, 이미지 토큰 예산, 평가셋 착시에서 자주 납니다.

  • 함정 1: 최적화 커널 없이 속도를 비교하기. 모델 카드는 최적화 Mamba2 커널 없이 실행하면 지연과 메모리 사용이 크게 나빠진다고 경고합니다. 예방책은 설치 로그와 실제 커널 로딩 여부를 벤치마크 기록에 포함하는 것입니다.
  • 함정 2: 이미지 해상도를 무제한으로 올리기. VLM은 이미지가 커질수록 시각 토큰이 늘어납니다. 예방책은 max_pixels를 업무별로 고정하고, OCR·카운팅·차트 분석마다 따로 허용치를 잡는 것입니다.
  • 함정 3: 평균 점수만 보고 도입하기. DocVQA와 PixMoCount가 좋아도 MathVista와 MMMU가 낮을 수 있습니다. 복구법은 제품 입력과 비슷한 100~300개 샘플로 자체 평가셋을 만드는 것입니다.
  • 함정 4: Apache 2.0만 보고 상용 배포를 확정하기. 모델 라이선스와 별개로 학습 데이터, 출력물 검수, 개인정보 이미지 처리 정책은 따로 봐야 합니다. 특히 영수증·문서에는 개인식별정보가 들어갈 수 있습니다.

8. 강점과 한계

핵심 요약: 강점은 긴 시각 입력의 지연·메모리 비용이고, 한계는 범용 추론과 런타임 성숙도입니다.

강점은 분명합니다. 첫째, 1.2B·2.7B·7B 크기로 공개돼 작은 배포 실험부터 시작할 수 있습니다. 둘째, Hugging Face 기준 Apache-2.0 라이선스로 공개되어 연구와 제품 검토가 쉽습니다. 셋째, 문서 이해와 카운팅 계열 벤치마크에서 작은 모델 대비 꽤 설득력 있는 수치를 보입니다.

한계도 같이 봐야 합니다. Zamba2-VL은 Zyphra의 특수 transformers 브랜치와 Mamba2 커널에 의존합니다. 또한 2026년 6월 현재 주요 추론 서버에서 얼마나 안정적으로 지원되는지는 별도 검증이 필요합니다. 무엇보다 이 모델은 추론 문제에서 가장 강한 범용 VLM이 아니라, 엣지·문서·카운팅 같은 특정 조건에서 비용 대비 성능을 노리는 후보입니다.

9. 더 깊게 공부할 포인트

핵심 요약: Zamba2-VL을 제대로 보려면 VLM 구조, SSM, KV 캐시, 자체 평가셋을 함께 공부해야 합니다.

  • VLM 기본 구조: 비전 인코더, 어댑터, 언어 백본의 역할을 먼저 이해해야 합니다.
  • SSM과 Mamba2: 상태공간 모델이 왜 긴 입력에서 효율적인지, 대신 무엇을 포기하는지 봐야 합니다.
  • KV 캐시와 TTFT: 사용자가 체감하는 속도는 총 토큰 처리량보다 첫 토큰 지연에 더 민감할 수 있습니다.
  • OCR·카운팅 평가셋: DocVQA, CountBenchQA, PixMoCount 같은 지표가 실제 업무와 맞는지 따져야 합니다.
  • 런타임 호환성: vLLM, SGLang, Transformers fallback 등 운영 경로가 준비됐는지 확인해야 합니다.

10. 실행 체크리스트 + 작성자 관점

핵심 요약: 도입 판단은 "오픈소스라서"가 아니라 "내 이미지 입력에서 지연·비용·오류가 줄었는가"로 해야 합니다.

  • 제품 이미지 100장 이상으로 자체 평가셋을 만들었다.
  • 각 이미지 유형별 max_pixels와 최대 입력 토큰 기준을 정했다.
  • TTFT, 전체 지연, GPU 메모리 피크, 실패 유형을 함께 기록했다.
  • 동급 트랜스포머 VLM과 같은 하드웨어, 같은 프롬프트, 같은 이미지로 비교했다.
  • OCR·카운팅·문서 이해와 범용 추론을 분리해 점수를 봤다.
  • 개인정보 이미지 저장, 로그 마스킹, 재처리 정책을 모델 도입 전에 정했다.
  • 최적화 커널이 없는 fallback 경로에서 성능이 얼마나 떨어지는지 확인했다.

Definition of Done: Zamba2-VL 도입 실험은 동일 샘플에서 기존 VLM 대비 TTFT 30% 이상 개선, 메모리 피크 감소, 업무 핵심 오류율 허용 범위 충족, 개인정보 처리 정책 문서화가 모두 확인될 때 완료로 봅니다.

제 판단은 조건부 추천입니다. 엣지 장비나 문서 자동화처럼 첫 응답과 메모리가 제품 경험을 좌우한다면 Zamba2-VL은 반드시 벤치마크해볼 가치가 있습니다. 하지만 "멀티모달이면 전부 잘해야 한다"는 요구라면 아직 더 큰 트랜스포머 VLM이나 상용 API를 기준선으로 두는 편이 현실적입니다.

참고자료

공유하기

관련 글

AQ 테스트 해보기

지금 내 AI 활용 능력이 어느 수준인지 3분 안에 확인해보세요. 인지력, 활용력, 검증력, 통합력, 윤리감을 한 번에 진단하고 맞춤형 인사이트를 받아볼 수 있습니다.

무료 AQ 테스트 시작하기