LoopVQ cuts looped models’ KV cache to half a bit and lifts Ouro’s throughput up to 6×

23 hours ago

LoopVQ, a KV cache quantizer for looped language models, stores Ouro's cache at half a bit per value with about a point of accuracy lost, and raises peak SGLang serving throughput up to 6× on one GPU. Minsoo Cheong and Sungjoo Yoo, Seoul National University, arXiv 2611.05304v1 (3 Nov 2026). Paper: https://arxiv.org/abs/2611.05304 Code: https://github.com/mscheong01/LoopVQ

Ask about this presentation

Chapters

  1. 0:00LoopVQ
  2. 0:16Looped models pay in KV cache
  3. 0:30The passes write nearly the same keys
  4. 0:42One index for four loop steps
  5. 0:57Calibration takes minutes
  6. 1:11Accuracy at half a bit
  7. 1:27Long reasoning stays on track
  8. 1:42Decode latency halves at 32K
  9. 1:56Throughput up to 6×
  10. 2:10Where the result stops
  11. 2:27The code
Show transcript
  1. LoopVQ shrinks the context memory of looped language models to half a bit per value. On Ouro, that raises peak serving throughput up to six times. It comes from Minsoo Cheong and Sungjoo Yoo at Seoul National University.

    Slide text
    AI Papers

    LoopVQ cuts looped models’ KV cache to half a bit and lifts Ouro’s throughput up to 6×

    Minsoo Cheong, Sungjoo Yoo · Seoul National University

    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026
  2. Looped models like Ouro run the same layers four times, so a small model performs like a much larger one. But every pass stores its own keys and values, six times the cache of Llama 3.1 8B.

    Slide text
    01 02 03 04 Problem

    Looped models save on weights and pay in KV cache

    6× the KV cache per token of Llama-3.1-8B, for the much smaller Ouro-1.4B
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Section 1, Figure 1a
  3. Those four copies are nearly identical, because the same weights compute them from the same token. Neighbouring passes correlate at about 0.7 early in the loop and above 0.9 by the end.

    Slide text
    01 02 03 04 Problem

    The four passes write nearly the same keys and values

    0.69 correlation between the first two loop steps
    0.94 between the last two
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Section 3, Figure 2
  4. Standard vector quantization stores a separate code for each pass. LoopVQ groups channels across four passes, so one index covers all four at the same bit cost. At half a bit, that roughly halves the error.

    Slide text
    01 02 03 04 Method

    One codebook index stands for four loop steps

    LoopVQ overview: loop-axis vector quantization and codebook fine-tuning
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Section 4.1, Figure 2c, Figure 3
  5. The codebooks are then fine-tuned until the quantized model's output matches the original's. For the 1.4 billion parameter Ouro, all of it takes about eight minutes on one RTX 6000 Ada card.

    Slide text
    01 02 03 04 Method

    The codebooks are tuned against the full-precision model

    8.4 min to calibrate and fine-tune Ouro-1.4B on one RTX 6000 Ada, from 16 sequences of 2,048 tokens
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Section 4.2, Figure 4, Section 5.1
  6. Half a bit is a 32-fold cut from 16-bit. At that size, earlier quantizers keep at most two thirds of Ouro's math and coding score. LoopVQ with TaSQ lands within about a point of full precision.

    Slide text
    01 02 03 04 Result

    At half a bit, Ouro-2.6B keeps nearly all its accuracy

    16-bit earlier quantizers LoopVQ
    BF16 (16-bit)
    81.5
    CQ
    37.7
    TaSQ
    52.7
    LoopVQ
    77.7
    TaSQ + LoopVQ
    80.3
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Table 2 · avg of GSM8K, MATH500, HumanEval, MBPP
  7. With earlier quantizers, Ouro's thinking model runs to the 32,000 token limit on most GPQA Diamond problems. With LoopVQ, answers stay as long as at full precision, so the speedup is real time saved.

    Slide text
    01 02 03 04 Result

    Long reasoning still ends where it should

    CQ and TaSQ hit the 32K-token cap on most problems, about twice the tokens of 16-bit
    LoopVQ follows the answer lengths of the 16-bit model
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Section 6.2, Figure 10 · GPQA-Diamond, 32K-token budget
  8. Decoding looks up small indices instead of 16-bit keys, so the saving grows with context. At 32,000 tokens, each new token takes about half as long. Prefill costs a few percent more.

    Slide text
    01 02 03 04 Result

    Each new token takes about half the time at a 32K context

    Prefill and decode latency, BF16 against LoopVQ at 0.5 and 0.25 bit
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Figure 8 · Ouro-1.4B-Thinking, one RTX 6000 Ada
  9. On one 48 gigabyte card, full precision fits only a few long requests, while LoopVQ serves dozens to over a hundred. Peak throughput rises up to four times on the smaller Ouro, and six on the larger.

    Slide text
    01 02 03 04 Result

    One GPU serves dozens of requests instead of a few

    End-to-end throughput against batch size, BF16 and LoopVQ

    Peak throughput: up to 4.3× on Ouro-1.4B, 6.1× on Ouro-2.6B

    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Figure 7 (Ouro-1.4B-Thinking), Appendix D.2 (Ouro-2.6B-Thinking) · one RTX 6000 Ada, 48 GB
  10. LoopVQ is tested only on looped models of 3.5 billion parameters or fewer, and its speed only against full precision. At a quarter bit, it loses about a third of Ouro's score, and every model needs its own calibration.

    Slide text
    01 02 03 04 Limits

    Tested on looped models up to 3.5B

    Models Ouro-1.4B, Ouro-2.6B, Huginn-3.5B
    Speed baseline 16-bit only, not other quantizers
    Quarter bit LoopVQ alone: 52.7 against 81.5 on Ouro-2.6B
    Calibration refitted per model, on GPQA text
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · Limitations, Table 2, Section 5.1
  11. Serving Ouro on one card, I would run TaSQ with LoopVQ at half a bit: about a point of accuracy for several times the throughput. The code and its SGLang server are on GitHub under the Apache license.

    Slide text
    01 02 03 04 Limits

    Serving Ouro on one GPU? Run TaSQ with LoopVQ at half a bit.

    Code github.com/mscheong01/LoopVQ
    Paper arxiv.org/abs/2611.05304
    Cheong and Yoo · LoopVQ · arXiv 2611.05304v1 · 3 Nov 2026 · github.com/mscheong01/LoopVQ · Apache-2.0

What happened in AI today, in your inbox

One short email each evening. Read it and you’re caught up.

Built with ❤️ in New York