Attention-Path Fragility as an Uncertainty Signal in Large Language Models
Source: arXiv:2608.11138 (cs.CL)
Authors: Minsoo Kim, Sungyoung Ji, Kisung Moon, Ilyong Yoon
Published: 2026-08-11
URL: https://arxiv.org/abs/2608.11138
PDF: https://arxiv.org/pdf/2608.11138
Abstract (English)
We propose that a model's uncertainty about a token is reflected not only in the breadth of its output distribution but also in whether a confident prediction is \emph{fragile} under perturbation of its attention pathways. We instantiate this as ASMI (Attention-Subnetwork Mutual Information), a training-free estimator that masks attention heads and measures the BALD mutual information among the resulting subnetworks, with a semantic-agreement kernel to discount surface-form disagreement. The signal is not a restatement of output confidence: on grounded QA an out-of-fold test shows it adds error-predictive information beyond single-pass confidence and entropy, concentrated in \emph{confident-but-fragile} predictions, where acting on it roughly halves the retained error of a confidence filter. The distinctness is regime-graded, so ASMI predicts its own domain of applicability, strong where answers are routed through provided context and bounded by design where they are recalled from parametric knowledge. Sem-ASMI reads the signal from a single greedy response, without the stochastic generations the strongest baselines require, and ties or beats Semantic Entropy on ten of the twelve grounded benchmark-backbone settings. Across the same twelve settings, the best ASMI variant, typically the adaptive one reusing the ten samples already drawn for the baselines, ties or leads the strongest baseline in eight, significantly in three under a paired test. On parametric QA all variants revert to or below the zero-cost MSP baseline, exactly as predicted, and the estimates are near-deterministic across reruns. A head-level analysis shows that what tracks this boundary is not the presence of head-level fragility but whether that fragility couples to errors.
Entry Metadata
- id:
ed0ad287 - source platform: arXiv
- paper id: 2608.11138
- primary category: cs.CL
- authors: Minsoo Kim, Sungyoung Ji, Kisung Moon, Ilyong Yoon
- published: 2026-08-11
- quality_score: 4
- one_liner_author: openclaw
- added_at: None
- tags: ["arxiv", "external-scan", "research", "cs.cl"]
- rationale:
Notes
- Grounded by
opencli arxiv paper 2608.11138 -f json. - summary_en is derived from the arXiv abstract verbatim (truncated to 1500 chars when long).
- External scan mode: discovery via
opencli arxiv recent cs.AI -f json; per-paper fetch via opencli; dedupe via pipeline_utils.normalized_url_key.