Attention Mechanism and Heuristic Approach: Context-Aware File Ranking Using Multi-Head Self-Attention
Not on the map. Why?
Abstract (arXiv)
The identification and ranking of impacted files within software reposi-tories is a key challenge in change impact analysis. Existing deterministic approaches that combine heuristic signals, semantic similarity measures, and graph-based centrality metrics have demonstrated effectiveness in nar-rowing candidate search spaces, yet their recall plateaus. This limitation stems from the treatment of features as linearly independent contributors, ignoring contextual dependencies and relationships between metrics that characterize expert reasoning patterns. To address this limitation, we propose the application of Multi-Head Self-Attention as a post-deterministic scoring refinement mechanism. Our approach learns contextual weighting between features, dynamically adjust-ing importance levels per file based on relational behavior exhibited across candidate file sets. The attention mechanism produces context-aware adjustments that are additively combined with deterministic scores, pre-serving interpretability while enabling reasoning similar to that performed by experts when reviewing change surfaces. We focus on recall rather than precision, as false negatives (missing impacted files) are far more costly than false positives (irrelevant files that can be quickly dismissed during review). Empirical evaluation on 200 test cases demonstrates that the introduc-tion of self-attention improves Top-50 recall from approximately 62-65% to between 78-82% depending on repository complexity and structure, achiev-ing 80% recall at Top-50 files. Expert validation yields improvement from 6.5/10 to 8.6/10 in subjective accuracy alignment. This transformation bridges the reasoning capability gap between deterministic automation and expert judgment, improving recall in repository-aware effort estimation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 4 of its 14 references that are in the map. First posted on arXiv 7 Jan 2026.
Builds on
4 of 14 references.
References · 4
- 2017Attention Is All You Needcited by 60,361
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
- 2025Scalable and Explainable Enterprise Knowledge Discovery Using Graph-Centric Hybrid Retrievalcited by 1
- 2025Keeping Code-Aware LLMs Fresh: Full Refresh, In-Context Deltas, and Incremental Fine-Tuningcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Breaking the Alphabet: Rethinking File Ordering in Code Review ↗
- 2025Rethinking Reasoning in Document Ranking: Why Chain-of-Thought Falls Short
- 2026AttnCompress: Dynamic Attention-Guided Trajectory Compression for Software Engineering Agents
- 2025From Illusion to Insight: Change-Aware File-Level Software Defect Prediction Using Agentic AI
- 2026Spotlights: Discovering Improvement Opportunities in Software Repositories
- 2026ContextBench: A Benchmark for Context Retrieval in Coding Agents
- 2026LENS: In-Context Search via Latent Evidence Exploration over Dynamic Raw Documents
- 2025AIR: Post-training Data Selection for Reasoning via Attention Head Influence