Weixu Zhang
2026
Context-Fidelity Boosting: Enhancing Faithful Generation through Watermark-Inspired Decoding
Weixu Zhang | Fanghua Ye | Qiang Gao | Jian Li | Haolun Wu | Yuxing Tian | Sijing Duan | Nan Du | Xiaolong Li
Findings of the Association for Computational Linguistics: ACL 2026
Weixu Zhang | Fanghua Ye | Qiang Gao | Jian Li | Haolun Wu | Yuxing Tian | Sijing Duan | Nan Du | Xiaolong Li
Findings of the Association for Computational Linguistics: ACL 2026
Large language models (LLMs) often produce content that contradicts or overlooks information provided in the input context, a phenomenon known as faithfulness hallucination. In this paper, we propose Context-Fidelity Boosting (CFB), a lightweight and general decoding-time framework that effectively reduces such hallucinations by boosting the generation probability of context-relevant tokens. Motivated by logit-shaping principles in watermarking techniques, CFB leverages token-level logit adjustments based on their presence or salience in the input context. Specifically, we develop three boosting strategies, static, context-aware, and token-aware that progressively incorporate distributional divergence, attention scores, and semantic similarity. Notably, CFB requires no retraining or architectural changes, making it compatible with a wide range of LLMs. Experiments on summarization and question answering tasks across multiple open-source LLMs show that CFB consistently improves faithfulness metrics, with minimal generation overhead. Our implementation is fully open-sourced.
ReAttn: Improving Attention-based Re-ranking via Attention Re-weighting
Yuxing Tian | Fengran Mo | Weixu Zhang | Yiyan Qi | Jian-Yun Nie
Findings of the Association for Computational Linguistics: EACL 2026
Yuxing Tian | Fengran Mo | Weixu Zhang | Yiyan Qi | Jian-Yun Nie
Findings of the Association for Computational Linguistics: EACL 2026
The strong capabilities of recent Large Language Models (LLMs) have made them highly effective for zero-shot re-ranking task. Attention-based re-ranking methods, which derive relevance scores directly from attention weights, offer an efficient and interpretable alternative to generation-based re-ranking methods. However, they still face two major limitations. First, attention signals are highly concentrated a small subset of tokens within a few documents, making others indistinguishable. Second, attention often overemphasizes phrases lexically similar to the query, yielding biased rankings that irrelevant documents with mere lexical resemblance are regarded as relevant. In this paper, we propose ReAttn, a post-hoc re-weighting strategy for attention-based re-ranking methods. It first compute the cross-document IDF weighting to down-weight attention on query-overlapping tokens that frequently appear across the candidate documents, reducing lexical bias and emphasizing distinctive terms. It then employs entropy-based regularization to mitigate over-concentrated attention, encouraging a more balanced distribution across informative tokens. Both adjustments operate directly on existing attention weights without additional training or supervision. Extensive experiments demonstrate the effectiveness of our method.
Preference Heads in Large Language Models: A Mechanistic Framework for Interpretable Personalization
Weixu Zhang | Ye Yuan | Changjiang Han | Yuxing Tian | Zipeng Sun | Linfeng Du | Jikun Kang | Hong Kang | Xue Liu | Haolun Wu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Weixu Zhang | Ye Yuan | Changjiang Han | Yuxing Tian | Zipeng Sun | Linfeng Du | Jikun Kang | Hong Kang | Xue Liu | Haolun Wu
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Large Language Models (LLMs) exhibit strong implicit personalization ability, yet most existing approaches treat this behavior as a black box, relying on prompt engineering or fine tuning on user data. In this work, we adopt a mechanistic interpretability perspective and hypothesize the existence of a sparse set of Preference Heads, attention heads that encode user specific stylistic and topical preferences and exert a causal influence on generation. We introduce Differential Preference Steering (DPS), a training free framework that (1) identifies Preference Heads through causal masking analysis and (2) leverages them for controllable and interpretable personalization at inference time. DPS computes a Preference Contribution Score (PCS) for each attention head, directly measuring its causal impact on user aligned outputs. During decoding, we contrast model predictions with and without Preference Heads, amplifying the difference between personalized and generic logits to selectively strengthen preference aligned continuations. Experiments on widely used personalization benchmarks across multiple LLMs demonstrate consistent gains in personalization fidelity while preserving content coherence and low computational overhead. Beyond empirical improvements, DPS provides a mechanistic explanation of where and how personalization emerges within transformer architectures.
SWE-QA-Pro: A Representative Benchmark and Scalable Training Recipe for Repository-Level Code Understanding
Songcheng Cai | Zhiheng Lyu | Yuansheng Ni | Xiangchao Chen | Baichuan Zhou | Shenzhe Zhu | Yi Lu | Haozhe Wang | Chi Ruan | Benjamin Schneider | Weixu Zhang | Xiang Li | Andy Zheng | Yuyu Zhang | Ping Nie | Wenhu Chen
Findings of the Association for Computational Linguistics: ACL 2026
Songcheng Cai | Zhiheng Lyu | Yuansheng Ni | Xiangchao Chen | Baichuan Zhou | Shenzhe Zhu | Yi Lu | Haozhe Wang | Chi Ruan | Benjamin Schneider | Weixu Zhang | Xiang Li | Andy Zheng | Yuyu Zhang | Ping Nie | Wenhu Chen
Findings of the Association for Computational Linguistics: ACL 2026
Agentic repository-level code understanding is essential for automating complex software engineering tasks, yet the field lacks reliable benchmarks. Existing evaluations often overlook the long tail topics and rely on popular repositories where Large Language Models (LLMs) can cheat via memorized knowledge. To address this, we introduce SWE-QA-Pro, a benchmark constructed from diverse, long-tail repositories with executable environments. We enforce topical balance via issue-driven clustering to cover under-represented task types and apply a rigorous difficulty calibration process: questions solvable by direct-answer baselines are filtered out. This results in a dataset where agentic workflows significantly outperform direct answering (e.g., a ~13-point gap for Claude Sonnet 4.5), confirming the necessity of agentic codebase exploration. Furthermore, to tackle the scarcity of training data for such complex behaviors, we propose a scalable synthetic data pipeline that powers a two-stage training recipe: Supervised Fine-Tuning (SFT) followed by Reinforcement Learning from AI Feedback (RLAIF). This approach allows small open models to learn efficient tool usage and reasoning. Empirically, a Qwen3-8B model trained with our recipe surpasses GPT-4o by 2.3 points on SWE-QA-Pro and substantially narrows the gap to state-of-the-art proprietary models, demonstrating both the validity of our evaluation and the effectiveness of our agentic training workflow.
Search
Fix author
Co-authors
- Yuxing Tian 3
- Haolun Wu 2
- Songcheng Cai 1
- Xiangchao Chen 1
- Wenhu Chen 1
- Nan Du 1
- Linfeng Du 1
- Sijing Duan 1
- Qiang Gao 1
- Changjiang Han 1
- Jikun Kang 1
- Hong Kang 1
- Jian Li 1
- Xiaolong Li 1
- Xiang Li 1
- Xue Liu 1
- Yi Lu 1
- Zhiheng Lyu 1
- Fengran Mo 1
- Yuansheng Ni 1
- Jian-Yun Nie 1
- Ping Nie 1
- Yiyan Qi 1
- Chi Ruan 1
- Benjamin Schneider 1
- Zipeng Sun 1
- Haozhe Wang 1
- Fanghua Ye 1
- Ye Yuan 1
- Yuyu Zhang 1
- Andy Zheng 1
- Baichuan Zhou 1
- Shenzhe Zhu 1