Fathom: Per-Query Read Depth for Sparse Decoding over Offloaded KV Caches

Authors: Vivek Kalyanarangan Affiliation:(文内独立署名) arXiv: 2609.17652(2026-09-15,cs.LG;Thu 9/17 cs.DC 列表) PDF: arXiv PDF

面向 百万 token agent 会话、多会话常驻、KV/索引在 host 的 regime:top-k 前的全量 key scan 主导 PCIe 流量。Fathom 让每个 query 决定每个 key 通道读多少 bit。

动机

  • Dense attention 读整份 KV;top-k 把 winner 行压到 ~200 MB/step,但排名用的 scan 随 n 线性涨(Qwen3-8B @1M 约 5.1 GB/step 量级的 136-bit 扫)。
  • Loki / Double Sparsity / SparQ 等固定每通道读深;重要通道的第 1 bit 远比第 4 bit 值钱。

方案

  1. Bit-plane K store:4-bit K 按 channel-major 存为平面;前 t 个平面即该通道 t-bit mid-rise 量化器。
  2. Per-query reverse water-filling:按方差加权边际收益分配 bit 预算。
  3. 评测:A100;对照 Loki / Double Sparsity / SparQ / thumbnail / landmark;含 RULER 与真实 coding-agent 会话。

效果(仅论文数字)

设定数字
Qwen3-8B @1M,host KV+indexGPU time vs 136-bit 扫 1.67×;vs landmark 2.50×
同模型 @256kvs 136-bit 1.37×
同 GPU time vs SparQ r=16少读 18% 字节;7 设定中 6 个更低 attention error
Coding-agent 会话达最准 136-bit 扫的 step agreement 仅需 92 bits
Index 在 HBM不更快(扫核本身不赢)

口径: 算法/系统测量在 A100;加速来自 host 卸荷扫索引,非新 ASIC。

与 wiki 的关系

开放问题

  1. 与 PagedAttention / 分层 KV 索引的正交组合。
  2. 写路径(K 更新)对 bit-plane 布局的开销。
  3. 多租户并发扫对 PCIe QoS 的影响。

Citations

[1] arXiv PDF — Kalyanarangan, arXiv:2609.17652 [2] fathom-sparse-decoding-offloaded-kv.md — 结构化摘录