让大模型在长文本中高效准确地做集合统计,误差仅1.6%
Mergeable Model-Side Aggregation States for Long-Context Language Models
- 用哈希+超对数日志结构在模型内维护紧凑聚合状态
- 百万级数据去重误差仅1.6%,跨段合并结果完全一致
- 适合需要快速统计的长文本任务,如日志分析、多轮对话
长上下文语言模型在非加性集合聚合任务中性能随上下文增长而下降,如基数估计、集合关系判断和分组统计,这些常见于日志、程序输出、表格和多轮对话。为解决此问题,我们引入一种模型侧聚合接口,在冻结的语言模型旁维护紧凑的基于哈希的超对数日志(HLL)状态。模型处理上下文时,提取器将相关记录映射为规范标识符,经哈希后更新HLL状态。该状态可跨上下文片段合并或直接读出用于下游推理,避免额外生成-执行-返回循环。我们设定HLL状态大小为2 KiB(2,048个寄存器),不随上下文长度或集合基数增加。在包含一百万条记录的去重实验中,平均相对误差为1.6%。在256段分片合并测试中,结果与单次遍历完全一致。在来自174个源窗口的3,969个聚合-推理任务上,固定预算接口在Gemma 4 (31B, BF16)上达到99.2%准确率,相比精确聚合的100.0%仅差0.8个百分点(95%窗口聚类置信区间:0.5–1.3点)。在174个匹配项上,相比直接全上下文推理,分别提升63.2点(Qwen)和56.3点(Gemma);相比思维链(CoT)推理,提升分别为60.9和63.2点。在固定的1,200任务Oolong-Synth子集上,分别达到91.1%(Qwen)和99.3%(Gemma)。
原文摘要 · Abstract (English)
A known limitation of long-context language models is their increasingly unreliable performance in non-additive, set-based aggregation as context length grows. Examples include cardinality estimation, set relationships, and grouped statistics, which widely exist in logs, program outputs, tables, and multi-turn conversations. To provide the aggregation state required by these tasks, we introduce a model-side aggregation interface that maintains compact Hash-based HyperLogLog (HLL) sketch states alongside a frozen language model. While the model processes the context, an extractor maps each relevant record to a canonical identity. The identity is then hashed and updates the HLL state. These states can be merged across context segments and/or read out directly for downstream reasoning, avoiding an additional generate-execute-return cycle. We validate the proposed approach by setting the HLL state size as 2 KiB (2,048 registers), which does not increase with context length or set cardinality. In a distinct-count experiment involving one million records, the mean relative error was 1.6%. In a separate merge test, states built from as many as 256 segments produced exactly the same readout as a single pass over the same stream. On 3,969 aggregate-then-reason tasks from 174 source windows, the fixed-budget interface reached 99.2% accuracy on Gemma 4 (31B, BF16), compared with 100.0% under exact aggregation; the paired gap was 0.8 percentage points (95% window-cluster CI: 0.5-1.3 points). On a matched set of 174 items, our method improved over direct full-context reasoning by 63.2 points on Qwen and 56.3 points on Gemma. The corresponding gains over chain-of-thought (CoT) reasoning were 60.9 and 63.2 points, respectively. On a fixed 1,200-task Oolong-Synth subset, our method reached 91.1% on Qwen and 99.3% on Gemma. Code is available at https://github.com/songdc98/sketchops.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。