无主之地2配置高吗|看真人裸体BBBBB|秋草莓丝瓜黄瓜榴莲色多多|真人強奷112分钟|精品一卡2卡3卡四卡新区|日本成人深夜苍井空|八十年代动画片

網易首頁 > 網易號 > 正文 申請入駐

超越TurboQuant,長上下文推理真2-bit KV Quantization算法問世

0
分享至



本文作者 Zhongzhu Zhou 是 TogetherAI 的 Senior Research Scientist,悉尼大學博士,研究方向為高效機器學習系統,方向覆蓋 模型訓推算法與系統協同設計,LLM 壓縮與量化。團隊成員均來自 TogetherAI,悉尼大學以及伊利諾伊大學厄巴納 — 香檳分校。

Together AI 于 2022 年 6 月創立,由蘋果前高管 Vipul Ved Prakash、斯坦福大模型研究中心主任 Percy Liang、芝加哥大學副教授 Ce Zhang、Flash Attention 作者 Tri Dao 聯合創辦。

超越 TurboQuant,內存有救了!TogetherAI 最新論文OSCAR直面沖擊 TurboQuant,提出一個面向長上下文推理服務的真正的 2-bit KV Cache 系統,開盒即用。



  • 論文標題:OSCAR: Offline Spectral Covariance-Aware Rotation for 2-bit KV Cache Quantization
  • 論文鏈接:https://arxiv.org/abs/2605.17757
  • 項目主頁:https://oscar-quantize.github.io/
  • 代碼:https://github.com/FutureMLS-Lab/OSCAR;
  • RotationZoo:https://huggingface.co/Zhongzhu/OSCAR-RotationZoo
  • 作者:Zhongzhu Zhou, Donglin Zhuang, Jisen Li, Ziyan Chen, Shuaiwen Leon Song, Ben Athiwaratkun, Xiaoxia Wu

長上下文模型越來越強,但服務時的瓶頸往往不是算力,而是 KV Cache:每生成一個 token,都要從顯存中讀取越來越長的歷史 key,value。上下文越長、batch 越大,KV Cache 越吃顯存,也越吃帶寬。把歷史 KV 壓到 2-bit,理論上可以讓歷史段顯存減少約8 倍;但真正難的是,壓完之后推理能力不能崩,系統也必須能在真實 serving 框架里跑起來。

為什么 2-bit KV Cache 這么難?INT2 只有 4 個量化等級,而 KV activation 中常常有少數幅值極大的 outlier channel。如果這些 outlier 主導量化尺度,大多數正常值會被擠到很少的有效等級里,注意力分布很快漂移。普通 Hadamard 旋轉能把 outlier 攤平,但它不知道模型在 attention 里真正讀哪些方向。OSCAR 的核心就是把旋轉目標從「重建原始 K/V 向量」改成「保留 attention 消費 KV 的方式」。

相比之前量化的工作,比如 TurboQuant 壓縮的是向量,但忽略了真正影響模型的是 attention 的質量,OSCAR 保留的是 attention 真正會讀的方向。樸素 INT2 和全模型層的 3-bit K/V TurboQuant 都會在困難推理任務上明顯掉分;OSCAR 在約2.28 effective bits per KV element下仍能接近 BF16,并在 Qwen3-4B-Thinking 上相對 3-bit K/V TurboQuant 最高提升40.1 分。

OSCAR 的動機



圖 1:為什么只看 K/V 重建誤差會誤導判斷

圖 1 對比了 naive INT2、Hadamard-only、clip-only 和 OSCAR 在量化誤差傳播鏈路上的差異。關鍵點是,原始 K/V 的重建誤差并不能完全解釋模型最終表現;真正影響推理質量的是 attention-score KL、attention-block output MSE 以及后續 hidden-state error。OSCAR 的優勢不只是讓向量數值更平滑,而是把量化誤差壓到 attention 不敏感的方向上。

OSCAR 的設計

具體來說,對 key 來說,量化誤差會進入 attention logits,也就是 QK?,因此 OSCAR 用 query covariance(Q?Q)構造 key 的旋轉目標;對 value 來說,誤差經過注意力權重進入輸出,因此 OSCAR 使用 score-weighted value covariance(V?S?SV)。離線校準階段,OSCAR 從少量校準樣本中估計這些 attention-aware covariance,為每層、每個 head 生成固定旋轉和 clipping 閾值。最終旋轉寫作R = U?Hadamard?bit-reversal:U 對準 attention 相關方向,Hadamard 分散 outlier,bit-reversal 平衡 INT2 分組,避免某個 group 被少數通道支配。

更重要的是,OSCAR 不是以往的量化論文,離線跑量化得到指標,而是已經接入SGLang,做到開箱即用的 2-bit KV serving。OSCAR 在 SGLANG 中維護一個 token 池:

BF16 sink (64 tokens) | INT2 history (~2.28 BPE) | BF16 recent (256 tokens)

其中 sink token 和 recent window 保持 BF16,用來保護 attention sink 與短期局部上下文;中間最長的歷史段存成旋轉后的 INT2。新 token 先寫入 recent window,隨著解碼推進,最老的 recent token 再由融合 Triton kernel 執行 rotate /clip/quantize/pack,并 demote 到 INT2 history。每 4 個 2-bit 值打包進 1 個 byte。decode 階段,OSCAR 在 GPU 上把緩存分成 BF16 段和 INT2 段:INT2 kernel 負責 unpack、scale/zero point 還原和浮點累加,BF16 kernel 處理 sink/recent,最后用 online softmax merge 合并結果。它同時兼容 paged KV、radix prefix cache 和 SGLang 的 fused kernel pipeline,因此可以直接用于長上下文 workload,而不是停留在論文圖表里。



圖 2:OSCAR 整體流程圖

圖 2 展示 OSCAR 從離線校準到在線 serving 的完整路徑。左側是離線階段:OSCAR 從少量校準樣本中估計 attention-aware rotation 和 clipping threshold,讓 KV activation 在進入 INT2 前變得更適合量化。右側是在線階段:sink/recent token 繼續保持 BF16,中間最長的 history KV 進入旋轉后的 INT2 cache,并在 SGLang paged KV 中完成真實 serving。因此 OSCAR 不是單一量化技巧,而是一整套 2-bit KV Cache pipeline。

評估結果

OSCAR 在 Qwen3-4B-Thinking、Qwen3-8B、Qwen3-32B 和 GLM-4.7-FP8 上測試,任務覆蓋 GPQA、HumanEval、LiveCodeBench v6、AIME25、MATH500,生成長度最高 32K,每個設置運行 5 次取均。

OSCAR 在2.28 BPE下,Qwen3-4B-Thinking 距 BF16 僅3.78分,Qwen3-8B 距 BF16 僅1.42 分,Qwen3-32B 與 GLM-4.7-FP8 基本與 BF16 持平。相比之下,QuaRot-INT2 和 naive INT2 在這些 reasoning /coding 任務上大多直接崩潰;TurboQuant 在全層 3-bit K/V、無 mixed-precision 保護的公平設置下,也在小模型推理任務上掉分明顯。

OSCAR還在128K長上下文設置下對中 / 大規模模型做了 RULER-NIAH 測試:OSCAR 在 Qwen3-8B 和 GLM-4.7-FP8 上都保持了明顯更穩定的檢索性能,說明這種 attention-aware 旋轉不僅能撐住短評測,也能抵抗超長歷史中 KV 誤差的累積。換句話說,OSCAR 是少數能在真近 2-bit 設置下仍保持現代 reasoning model 質量的方法。

系統收益也非常直接:相對 BF16 history storage,OSCAR 可減少約8×KV Cache memory;在 100k context、batch-size-1、full prefix-cache hit 設置下,decode 最高約3×加速;在大 batch、同顯存預算下,job-level throughput 最高約7×。prefix cache 命中率越高,OSCAR 越能利用更小的 KV footprint 提升并發吞吐,這對共享系統提示、多輪 Agent、工具調用循環等長前綴復用場景尤其重要。

精度損失



圖 3:完整主結果表,多種 KV 量化方法同場對比



圖 4:AIME25 32K 生成,和 KIVI / Kitty 的專項對比

圖 3 是論文主結果表,包含 BF16、Saw-INT4、TurboQuant、QuaRot-INT2、Naive INT2 和 OSCAR 在四個模型、五個任務上的完整對比。BF16 是精度上界;Saw-INT4 是強 4-bit 參考,BPE 為4.25;TurboQuant 在這里使用無 mixed-precision 保護的全層3-bit K/V設置,BPE 為3.25;QuaRot-INT2 和 Naive INT2 是接近 2-bit 的旋轉 / 樸素基線,BPE 約2.25;OSCAR 則在2.28 BPE下運行。

這張表的重點不是單一模型,而是「低比特能不能穩定」。在 Qwen3-4B-Thinking 上,TurboQuant mean 為31.74,QuaRot-INT2 只有1.40,Naive INT2 為0.00;OSCAR 達到71.86,距離 BF16 只差3.78,并相對 TurboQuant 提升40.1 分。在 Qwen3-8B 上,OSCAR mean 為69.42,距離 BF16 只差1.42,而 TurboQuant 為56.88。到 Qwen3-32B 和 GLM-4.7-FP8,OSCAR 基本與 BF16 持平。換句話說,在接近 2-bit 的 KV 預算下,OSCAR 是表中唯一能在多模型、多任務上穩定貼近 BF16 的 INT2 方法。

圖 4 單獨看 AIME25 這個高難數學推理任務,并對比 KIVI-KV2、Kitty 和 OSCAR。但由于 KIVI, KITTY 沒有 framework 支持,無法進行 long context run,所以選取了他們方法唯一在 32K 匯報的結果 - AIME25。在 Qwen3-8B 上,OSCAR 以2.38 BPE達到66.67,基本追平 BF16 的66.00,明顯高于 KIVI-KV2 和 Kitty;在 Qwen3-32B 上,OSCAR 達到74.00,甚至略高于 BF16 的72.59,也超過 Kitty 的69.26。這說明 OSCAR 不只是相對 TurboQuant 有優勢,在已有 KV-cache 量化方法中,也能在接近 2-bit 的預算下保住困難數學推理能力。

系統加速



圖 5:100k 長上下文下的 decode /batch throughput

圖 5 展示 100k 上下文下的系統性能。OSCAR 在 batch-size-1、full prefix-cache hit 的純 decode 場景下最高約3×加速;在固定顯存預算下,batch size 增大時,INT2 history 帶來的 KV footprint 降低可以顯著提高 job-level throughput,最高約7×。這說明 OSCAR 不只是精度能保住,也能實打實降低顯存帶寬壓力。



圖 6:prefix cache 命中率越高,吞吐前沿越往外推

圖 6 展示 prefix-cache hit ratio 對端到端 serving throughput 的影響。橫軸是單用戶吞吐,縱軸是單 GPU 吞吐;從 cache disabled 到 normal cache,再到接近 100% warmup replay,吞吐前沿逐步外擴。OSCAR 保持標準 paged KV /prefix cache 抽象,因此共享系統提示、多輪 Agent、工具調用循環等長前綴復用場景可以直接受益。

這些結果的一個重要含義是,OSCAR 并沒有依賴「挑選少數層保留高精度」來保住分數。很多低比特方法在真正部署時會借助混合精度:第一層、最后一層或若干敏感層仍然保留較高 bit,這會讓平均 bit 數上升,也會讓 kernel 和 cache layout 變復雜。OSCAR 的對比更嚴格:歷史 KV 主體保持統一的 INT2 表示,只在 sink 和 recent 兩個很小窗口保留 BF16。這樣做的好處是,系統工程上更容易接入 paged cache、prefix cache 和批量調度,也更接近真實服務場景中的顯存預算。

總結

另一個值得強調的點是,OSCAR 的收益不是只在小模型或短上下文上成立。論文同時測試了 4B、8B、32B 以及 GLM-4.7-FP8 這樣的大模型;既看了數學、代碼、知識問答等 32K 推理生成任務,也看了 128K RULER-NIAH 長上下文檢索。短評測里,OSCAR 能接近 BF16;長上下文里,它也能讓 attention 分布隨上下文增長更穩定。這說明 attention-aware rotation 不是只在某個 benchmark 上調參有效,而是在緩解 KV 誤差隨歷史長度累積這個根本問題。

從應用角度看,這對長上下文 Agent 特別關鍵。真實 Agent 往往包含很長的系統提示、工具說明、歷史對話和檢索內容,并且不同請求之間存在大量共享前綴。如果 KV Cache 只能用 BF16 存,系統很快會被顯存卡?。蝗绻苯幼鰳闼?INT2,又可能讓推理鏈條失真。OSCAR 的設計剛好夾在兩者之間:長歷史用 INT2 降顯存和帶寬,關鍵 sink/recent 用 BF16 兜住穩定性,再讓 prefix cache 復用共享前綴。換句話說,它把「能壓到 2-bit」和「能上線 serving」 放在同一個系統里考慮。

TurboQuant 是很強的通用 online vector quantization 方法;OSCAR 針對的是 attention-aware 2-bit KV serving。二者不是簡單替代關系,例如OSCAR 的 最新codebase中已經在attention-aware rotation 引入了更強的 Lloyd Max Codebook,將壓縮推向極致。OSCAR 帶來了一個獨特的觀點:2-bit KV Cache 要能上線,旋轉不只是「有沒有」,而是必須對準 attention,并且要有真實 serving 系統支撐。

特別聲明:以上內容(如有圖片或視頻亦包括在內)為自媒體平臺“網易號”用戶上傳并發布,本平臺僅提供信息存儲服務。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相關推薦
熱點推薦
蘇州一對情侶,談了7年,女子提了18次分手,分手后在街頭痛哭!

蘇州一對情侶,談了7年,女子提了18次分手,分手后在街頭痛哭!

川渝視覺
2026-04-17 22:13:14
夫妻一吃自己做的飯就拉肚子,吃外賣卻沒事,一年半還沒找到原因

夫妻一吃自己做的飯就拉肚子,吃外賣卻沒事,一年半還沒找到原因

夜深愛雜談
2026-05-29 07:56:08
雙管齊下!曼聯1.5億鎖中場雙星,紐卡飛翼+2英冠球員進引援名單

雙管齊下!曼聯1.5億鎖中場雙星,紐卡飛翼+2英冠球員進引援名單

體壇鑒春秋
2026-05-29 12:35:53
今天,你爺爺的爺爺都漲停了

今天,你爺爺的爺爺都漲停了

趨勢巡航
2026-05-29 14:51:54
譚松韻被曝離婚了!本人回應:的確是結了,但現在又離了,后來和好了

譚松韻被曝離婚了!本人回應:的確是結了,但現在又離了,后來和好了

八卦王者
2026-05-29 11:25:44
扎心!“小馬云”女友發聲,我就是想蹭流量,每次叫我姐姐發早安

扎心!“小馬云”女友發聲,我就是想蹭流量,每次叫我姐姐發早安

火山詩話
2026-05-27 07:40:38
湖北省供銷合作總社黨組書記、主任調整

湖北省供銷合作總社黨組書記、主任調整

娛樂圈的筆娛君
2026-05-29 17:58:26
莫言:如果你混到沒人找你吃飯,沒人喊你聚會,連電話也沒幾個,那慶祝了,你不是人緣變差,而是真正覺醒了

莫言:如果你混到沒人找你吃飯,沒人喊你聚會,連電話也沒幾個,那慶祝了,你不是人緣變差,而是真正覺醒了

品讀時刻
2026-05-27 09:00:58
國防部發言人回應網友“手搓”殲-35和殲-6亮相“東北超”

國防部發言人回應網友“手搓”殲-35和殲-6亮相“東北超”

大象新聞
2026-05-29 08:24:07
王晶點評《給阿嬤的情書》票房一飛沖天原因,內行果然一語中的!

王晶點評《給阿嬤的情書》票房一飛沖天原因,內行果然一語中的!

八卦南風
2026-05-29 11:14:02
她花50萬買英國精子生三胞胎,8年后三個娃長這樣,全網沉默

她花50萬買英國精子生三胞胎,8年后三個娃長這樣,全網沉默

吳蒂旅行ing
2026-05-29 06:40:39
馬斯克改口:Anthropic 租用 SpaceX 算力僅半年,必要時將收回

馬斯克改口:Anthropic 租用 SpaceX 算力僅半年,必要時將收回

環球網資訊
2026-05-29 10:33:14
歐洲多國使館力挺烏克蘭,拒絕撤離基輔,但澤連斯基想要更多

歐洲多國使館力挺烏克蘭,拒絕撤離基輔,但澤連斯基想要更多

共工之錨
2026-05-29 20:31:49
殯葬師提醒:守靈那一夜,家屬千萬別只顧著熬,一定守好這3件事

殯葬師提醒:守靈那一夜,家屬千萬別只顧著熬,一定守好這3件事

千秋歷史
2026-05-26 19:31:00
隨著廣東廣州豹0-0,寧波1-1,陜西0-1,中甲最新積分榜出爐

隨著廣東廣州豹0-0,寧波1-1,陜西0-1,中甲最新積分榜出爐

側身凌空斬
2026-05-29 21:27:35
有人預測:再過一兩個月,中國二手房或將面對這3個結局

有人預測:再過一兩個月,中國二手房或將面對這3個結局

童童聊娛樂啊
2026-05-29 19:12:35
“特變電工”潛力巨大!

“特變電工”潛力巨大!

木禾黑貓
2026-05-29 21:37:41
中方J-16戰機攜帶實彈驅趕荷艦機,30秒內可解決

中方J-16戰機攜帶實彈驅趕荷艦機,30秒內可解決

無意爭春
2026-05-29 18:53:23
第一批逃離高溫的人,已經在吉林住下了

第一批逃離高溫的人,已經在吉林住下了

聞旅派
2026-05-28 20:04:13
當你接觸的人多了,你會明白如果一個人還在穿5年前的衣服,只說明這幾個問題

當你接觸的人多了,你會明白如果一個人還在穿5年前的衣服,只說明這幾個問題

LULU生活家
2026-05-29 19:07:23
2026-05-29 22:35:00
機器之心Pro incentive-icons
機器之心Pro
專業的人工智能媒體
13115文章數 142655關注度
往期回顧 全部

科技要聞

Claude Opus 4.8凌晨突發上線

頭條要聞

疑第三方幫拉客宣傳"存1萬返100還報銷路費" 銀行回應

頭條要聞

疑第三方幫拉客宣傳"存1萬返100還報銷路費" 銀行回應

體育要聞

即使是文班亞馬,也做不到這件事

娛樂要聞

奚夢瑤何猷君將于6月在法國舉行婚禮

財經要聞

馬云密友"錢多多",深陷"殺豬盤"質疑

汽車要聞

900V+3.2秒破百 領克10+&領克10上市16.99萬元起

態度原創

教育
數碼
本地
健康
公開課

教育要聞

噓!中高考來了,全城禁噪聲

數碼要聞

vivo頭戴降噪耳機發布:238g重量、40mm動圈單元,499元

本地新聞

用剪紙的方式,打開江蘇揚州

嘗試干細胞療法如何避免踩坑?

公開課

李玫瑾:為什么性格比能力更重要?

無障礙瀏覽 進入關懷版