arXiv paper· machine learning (transformer architectures)· empirical study

Abstention and Noise Filtering: Two Missing Primitives of Softmax Attention

Richard Zhe Wang

cs.LG · cs.CL

Quick verdict

认为价值门控(value gating) = 放弃(abstention) + 噪声过滤(noise filtering);配对的 10M–350M 预训练支持这种划分,但 350M 是单种子(single-seed),且影响微小。

Summary

该论文认为,对注意力(attention)的价值通路(value pathway)进行门控(gating)可以改善语言模型的预训练,因为它提供了两种可分离的原始机制,而这正是 softmax 注意力所缺失的:放弃(abstention,即头部可以不输出任何内容,而不必分配总和为 1 的注意力质量)和噪声过滤(noise filtering,即价值通路可以抑制来自叠加特征的干扰)。放弃(Abstention)是通过每个头学习的接收端 logit(sink logit,即一个值固定为零的虚拟键)来实现的,过滤则是通过两种门控形式(范数门控 norm gate 和学习的投影门控 projection gate)来实现的,并在 FineWeb-Edu 上使用配对的单种子(per-seed)比较,对 10M 到 350M 参数的匹配 GPT-2 风格模型进行了训练。放弃(Abstention)的好处从 10M 时的 0.0185 nats 下降到 350M 时的 0.0013,而过滤的好处却在增长,结合了两种原始机制的模型在每个规模下都达到了最低的验证损失。注入实验证实了门控确实消除了移植读取干扰(transplanted-read interference)——结构化噪声对基线的损害是高斯噪声的十四倍——并暴露了每种门控的盲点:范数门控会通过大的垃圾数据,而投影门控会通过与学习方向对齐的垃圾数据。局限性:350M 的结果基于单种子(single seeds),且没有单门控分支(solo-gate arms),其影响仅为几千分之一 nats,处于一种脆弱的迁移机制中,仅测试了一个语料库和一种架构,且没有报告代码或吞吐量测量结果。

Review complete

Assessment at a glance

machine learning (transformer architectures)

5

risk flags identified

● high 1

● medium 3

● low 1

Reproducibility readinessmedium
Code & data availablenot stated
Review confidencehigh
AI text signal0/20 passages flagged
Deep research viableCode-reproducibleNeeds GPU

5 strengths · 5 weaknesses · 5 author questions

AI text signal = Binoculars zero-shot detector score over 20 sampled passages (Fast-DetectGPT co-reported) — absence does not prove human authorship, and a flag does not imply misconduct.

Levels reflect verification complexity — not a quality score. Evolve never rates your paper.

Key claims under review

  • Value-gating benefits decompose into two separable primitives — abstention (exact zero output escaping softmax's sum-to-one constraint) and noise filtering (content-based attenuation of value reads under superposition) — each supplied by a minimal dedicated mechanism (learned sink logit; norm or projection gate).
  • The abstention benefit declines with scale (0.0185 → 0.0013 nats from 10M to 350M) because larger baselines imitate a sink more effectively, while the filtering benefit grows, with the curves crossing between 50M and 124M.
  • The two benefits are additive up to the overlap that Lemma 3.1 predicts, and the sink-logit + projection-gate combination is the lowest-loss arm at every scale (4.1897/3.5266/3.3604/3.0210 nats at 10M/50M/124M/350M).
  • Causal injection experiments show gated models are far more robust to read interference, and every filter is blind along its own decision variable: norm gates pass magnitude-exceeding junk, projection gates pass junk aligned with their learned direction.

Strengths

  • Falsifiable decomposition anchored in a precise lemma

    Lemma 3.1 shows a per-key value gate equals a routing gate plus a per-query factor Z_i ≤ 1, predicting exactly which part of a solo gate's benefit should overlap with abstention; Table 2's overlap column (0.0024–0.0076 nats across tiers and gate forms) tests this prediction directly rather than leaving 'why gating works' as narrative.

  • Methodological care at tiny effect sizes

    Paired per-seed differences with standard errors throughout (Table 1), near-identity gate initialization (norm threshold −4/β so g_j ≥ 0.98; projection gate w=0, b=4), and principled weight-decay exclusions for gate/sink parameters (Appendix A) preempt the two most common artifacts in this literature.

  • Genuinely causal interference experiments

  • Evidence beyond the headline tables

  • Candid limitations section

Weaknesses

  • The scale crossover culminates at an underpowered tier

  • Effect sizes in the known non-transfer regime, without inferential statistics

  • Single corpus, single architecture, weak downstream signal

  • No code, checkpoints, or throughput numbers

  • Blind-spot asymmetry rests on an adversarial construction

Full Review Report

$1.99 · one-time

Unlock the complete preliminary analysis Evolve produced for this paper.

Report preview

Scope & Claims

The paper intervenes in a live disagreement: Qiu et al. (2025) attribute value/output-gating gains to nonlinearity and sparsity, Bu et al. (2025) to mitigation of extreme-token phenomena, Fesser et al. (2026) to eliminating no-operation sinks. The author's thesis is that these are partial views of two separable primitives: **abstention** (the zero option — a head can output nothing, escaping softmax's sum-to-one constraint) and **noise filtering** (content-based attenuation of value reads under superposition). Six contributions are claimed: the decomposition with dedicated arms, a scale study (Finding 1, §4), an additivity analysis (Finding 2, §5), causal injection experiments with a blind-spot principle (Finding 3, §6), evidence that interference exists in ordinary models (Appendices D–F), and a design corollary — sink logit plus projection gate (§7). Scope is pretraining-loss attribution on GPT-2-style models; no downstream-capability or throughput claims are made explicitly, though the corollary implies deployability.

The remaining sections cover Methodology Assessment, Statistical Evidence, Results & Reproducibility, Limitations & Risks, and Verdict & Recommendations.

  • All 5 strengths & 5 weaknesses, analyzed in detail
  • 5 risk flags with severity, section & reviewer questions
  • Methods & statistical evidence assessment
  • 5 questions a peer reviewer would ask
  • Full six-dimension evaluation report & recommendations

One-time purchase · this paper only

One-time purchase, this paper only. AI-generated preliminary analysis — not peer review.

Frequently asked questions

What are the two missing primitives of softmax attention according to this paper?

Abstention — the ability of an attention head to output nothing instead of being forced to distribute attention weights that must sum to one — and noise filtering — the ability of the value pathway to suppress interference from superposed features in the residual stream. The paper supplies abstention with a learned per-head sink logit (a phantom key whose value is fixed at zero) and filtering with a gate on each value read.

Does gating the values of attention actually improve language model pretraining?

Yes, but modestly, and for two distinguishable reasons. In matched GPT-2-style models from 10M to 350M parameters, combining a sink logit with a projection gate lowered validation loss by roughly 0.01–0.02 nats. The abstention portion helps most at small scale and nearly vanishes by 350M, while the filtering portion grows with model size.

What is an attention sink and why does it matter here?

An attention sink is a position (usually the first token) where a model parks unwanted attention mass because softmax weights must sum to one. The paper argues sinks are the baseline's imitation of abstention: at 124M the baseline places 6% of attention mass on position 0 with a value read driven to a norm ratio of 0.42, and providing an explicit sink logit drops that mass to 0.5% while the read recovers.

What are the blind spots of value gates found in the paper?

Every filter fails along its own decision variable. A norm gate passes injected junk large enough to exceed its threshold — at 124M it loses 4.4 nats versus the baseline's 2.6 at dose 0.8 — and a projection gate passes junk aligned with its learned gate direction, losing eleven to seventeen times more than under a random direction.

About this review

This is an AI-generated preliminary review by Evolve.Pub, produced from the openly available text of the paper. It is not peer review and does not represent the authors' views. The canonical record of the paper is at the publisher link above.

  • Figures 1–5 appear only as captions in the extraction; quantitative reading relies on the tables and numbers quoted in prose.
  • Minor text/table discrepancy: §6.2 quotes 0.035 nats for Gaussian damage where Table 7 lists 0.032 (124M, median cutoff, ε=0.4).
  • Several cited works are 2026 arXiv preprints whose content could not be verified offline; related-work characterizations are taken on trust.

Go deeper on this paper

Wrote a paper? Stress-test it before reviewers do.

Upload your manuscript for the same structured scrutiny — reasoning, statistics, evidence, figures, and reproducibility.

Scan your paper
Abstention and Noise Filtering: Two Missing Primitives of Softmax Attention | Evolve.Pub