cs.LGSep 28, 2026

Counterexamples to Local Reconstruction Gain as a Proxy for Final Fidelity in Residual Completion

Authors: Yasuto Hoshi, Daisuke Miyashita, Jun Deguchi

Organizations: Kioxia Corporation

Abstract

Residual completion augments query-aware sparse attention by estimating the contribution of tokens omitted from the exact sparse computation. We ask whether improving a layer's attention-output reconstruction on the same incoming Q/K/V and selected support necessarily improves the fidelity of the final model output. We study training-free RESA and learned Top-K+φφ with frozen backbone language models. A prespecified single-layer screen yields two Qwen3-0.6B/Multi-LexSum interventions for which direct-runtime measurements show positive prespecified request-aggregate local reconstruction gain but worse final KL fidelity than the corresponding all-abstain Exact Top-K baseline on both discovery and prompt-token-disjoint holdout requests. Exact restoration at the same layer instead improves final fidelity, showing that the reversal is specific to approximate completion in these cases. In complementary multi-layer experiments, a task-independent local diagnostic often repairs the tested completion estimators, although the repaired models do not consistently outperform Exact Top-K. Together, these results show that better local reconstruction need not translate into better final-model fidelity.

Explore similar work

CardsList
  1. Language Re-generation: An investigation into information locality effects on reconstruction

    Jul 11, 2026Amirhossein Mohammadi, Laurence E. Frank, Albert Gatt +1Language ModelingInductive Bias

  2. CEDAR: Error-Bounded Residual Routing for Efficient Long-Context Attention

    Sep 7, 2026Siyu Li, Dong Wang, Jie Zhou +3Dynamic Sparse AttentionEfficient Long-Context Inference