論文解説: Gated DeltaNet はなぜ4bit量子化に耐えるのか — ハイブリッド27BのNVFP4 W4A4
ハイブリッドLLMの再帰層(Gated DeltaNet)は4bit量子化に弱い、という通説を実測で否定した論文の解説。ゲートの対数パラメータ化とdelta則の上書きが、なぜ量子化ノイズを消してしまうのかを1から追う。
Why Gated DeltaNet Survives 4-Bit Quantization: NVFP4 W4A4 for the Recurrent Half of a Hybrid 27B LLM
一次資料 — この記事の根拠
論文の発表 2026-09-03→この解説の公開 2026-09-06同月
Why Gated DeltaNet Survives 4-Bit Quantization: NVFP4 W4A4 for the Recurrent Half of a Hybrid 27B LLMSergii Kozyrev, Davyd Maiboroda · 2026-09-03 · v1arXiv:2609.04098論文ページ·PDF原文の要旨(Abstract)を読む
Hybrid LLMs pair softmax attention with linear-attention layers such as Gated DeltaNet (GDN), whose recurrent state summarizes the context in fixed size. Early community 4-bit quantizations of Qwen3.8-27B (48 GDN layers, 16 attention layers) left the GDN block in 8- or 16-bit precision -- especially its decay and write-strength gates -- on the intuition that errors in a recurrence accumulate over long contexts. We test that intuition by building Minima: NVFP4 W4A4 on all 496 linear layers, GDN included. Across perplexity at 4K/32K, MMLU-Pro, GSM8K, AIME'25, GPQA-Diamond, LiveCodeBench, and RULER retrieval to 64K, Minima matches BF16 within seed noise (5-task average -0.52) while being the smallest (17.5 GiB) and fastest-prefill (+14-19%) recipe we compare, and its 32K perplexity gap shrinks with position. A four-part mechanism study explains why: (i) NVFP4's 16-element block scaling localizes the residual stream's extreme outliers, equalizing activation error across layer roles; (ii) the supposedly fragile gate projections are the least sensitive -- softplus/exponential and sigmoid parameterizations compress ~11% GEMM error to ~2% output error; (iii) the delta-rule recurrence holds injected noise at a flat plateau over 32K tokens and forgets a state impulse within hundreds of steps, because each write overwrites the state along the current key direction; (iv) the per-token quantization cost washes out with context instead of compounding. We also repair a global-scale mismatch that arises when per-module-calibrated NVFP4 checkpoints are served by kernels that fuse those modules into one GEMM, and show calibrated FP8 KV-cache scales are performance-free. The result: a practical recipe -- quantize everything, ship KV scales -- and a mechanistic account of why the recurrent half of a hybrid LLM is the easy half to quantize. Checkpoint: https://huggingface.co/minima-ai/mnma_qwen3.8_27b_nvfp4
「再帰は量子化に弱い」という直感を疑う
この記事で読む論文の原題は "Why Gated DeltaNet Survives 4-Bit Quantization: NVFP4 W4A4 for the Recurrent Half of a Hybrid 27B LLM"(Sergii Kozyrev, Davyd Maiboroda / arXiv:2609.04098, 2026-09-03)です。
要旨を日本語でまとめます。最近の大規模言語モデルは「ハイブリッド」— 大半の層をsoftmax注意ではなく、固定サイズの状態で文脈を要約する線形注意(Gated DeltaNet、以下GDN)に置き換えた構成 — になりつつあります。Qwen3.8-27B は64層のうち48層がGDN、16層だけが通常の注意です。ところがこのモデルの4bit量子化版はどれも、GDNブロック、とりわけ減衰ゲートと書き込み強度ゲートの射影だけは8bitや16bitのまま残していました。「再帰は誤差が溜まるから」という理由です。論文はこの直感を検証するため、496本の線形層すべてを NVFP4 の W4A4(重みも活性値も4bit)にした Minima を作り、4K/32Kのperplexity、MMLU-Pro、GSM8K、AIME'25、GPQA-Diamond、LiveCodeBench、64KまでのRULER検索で評価します。結果はBF16とシード誤差の範囲で一致(5タスク平均 )、かつ比較対象の中で最小(17.5 GiB)でprefillが最速(+14〜19%)。しかも32Kのperplexity差は文脈が進むほど縮む。論文はその理由を4段構えの機構分析で説明します。
比喩: 伝言ゲームか、上書きされるホワイトボードか
再帰が量子化に弱いと考えるとき、頭にあるのは伝言ゲームです。1人が少しずつ聞き間違えれば、100人後には原型を留めない。ならば状態 を毎ステップ更新する層に4bitの丸め誤差を混ぜれば、3万トークン後には壊れているはずだ、と。
論文が示すのは、GDNの状態はむしろ共用のホワイトボードに近いという事実です。新しいトークンが来るたび、そのトークンの「キー」に対応する欄が消されて書き直される。古い書き込みは薄れるのではなく消される。だから誤差も同じ速さで消えていきます。
前提1: 4bitに丸めるとは何をすることか
NVFP4 は値を E2M1(符号1・指数2・仮数1の計4bit)で持ち、16要素ごとに1つの E4M3 スケール(ブロック最大値÷6)と、テンソル全体で1つのFP32スケールを添える形式です。W4A4 とは重みと活性値の両方をこの粒度で量子化することで、行列積が4bitのテンソルコアで直接走ります(§2)。
ここで効く性質が2つあります。ひとつ、ブロックの最大値がそのブロックのスケールを決めるので、外れ値が壊すのは自分と隣の15要素だけです。ふたつ、ブロック内では なので、1つの値だけが突出した状態には上限があります。粗い目盛りに丸めると何が起きるかは、画像圧縮で体感するのが早いです。
前提2: Gated DeltaNet は何を計算しているか
GDN層は残差ストリームを4つの線形写像に通します。in_proj_qkv(因果畳み込みとSiLUを経て に分かれる)、出力ゲート in_proj_z、そしてヘッドごとにスカラーを出す2本のゲート射影 in_proj_a と in_proj_b です。ゲートは対数空間でパラメータ化されます(§2)。
読み下すと、 が射影の出力(生の数値)、それを softplus と指数に通した がどれだけ覚えておくか(1に近いほど忘れない)、 をシグモイド に通した がどれだけ強く書き込むかです。量子化ノイズが乗るのは そのものではなく、その手前の — これが後で効きます。
状態 (ヘッドごとに の行列)は、 正規化されたキーとクエリの上で次のように更新されます。
括弧の中身 が肝です。これは「いまの状態がキー に対して答える内容」と「本当に書きたい値 」の差です。つまりGDNは を足し込むのではなく、キー の方向にあった内容を で置き換える。これがdelta則(差分則)と呼ばれる理由です。
量子化の候補になるのはGDN層あたり5本の重み行列で、コミュニティの通例は と を完全に守り、残りを8bitに留めるというものでした。対して Minima は496本 — GDN 240、注意 64、MLP 192 — を量子化し、BF16のまま残すのは embedding・lm_head・GDNの conv1d・各種norm・/dt_bias だけ。校正は128サンプル×32Kトークンの凍結セット、配信はNVFP4をネイティブ対応するRTX PRO 6000 1枚でテキスト専用(§3)です。
主結果: 全部4bitにしても壊れない
比較は同一条件(vLLM 0.27.1、TP=1、RTX PRO 6000 1枚、FP8 KVキャッシュ、同一ハーネス)で4本(§4)。BF16、Minima、そして公開版のUnsloth Dynamic v3 と RadixArk(どちらもGDNと注意はFP8 W8A8、/ はBF16、NVFP4はMLPのみ)。読み取れることは3つあり、論文の議論が本当に始まるのは3つ目からです。
コメント
コメントにはログインが必要です