Yixin Wu
Other people with similar names: Yixin Wu
Unverified author pages with similar names: Yixin Wu
2026
Mitigating Position Bias in Transformers via Layer-Specific Positional Embedding Scaling
Zhenghua Wang | Yiran Ding | Changze Lv | Yixin Wu | Tianlong Li | Zhibo Xu | Muling Wu | Tianyuan Shi | Shizheng Li | Qi Qian | Xuanjing Huang | Xiaoqing Zheng
Findings of the Association for Computational Linguistics: ACL 2026
Zhenghua Wang | Yiran Ding | Changze Lv | Yixin Wu | Tianlong Li | Zhibo Xu | Muling Wu | Tianyuan Shi | Shizheng Li | Qi Qian | Xuanjing Huang | Xiaoqing Zheng
Findings of the Association for Computational Linguistics: ACL 2026
Large Language Models (LLMs) still struggle with the "lost-in-the-middle" problem, where critical information located in the middle of long-context inputs is often underrepresented or lost. While existing methods attempt to address this by combining multi-scale rotary position embeddings (RoPE), they typically suffer from high latency or rely on suboptimal hand-crafted scaling strategies. To overcome these limitations, we introduce a layer-specific positional embedding scaling (LPES) method that assigns distinct scaling factors to each layer. LPES achieves a more balanced attention distribution without fine-tuning model parameters or increasing inference delay. A specially designed genetic algorithm is employed to efficiently select the optimal scaling factors for each layer by incorporating B’ezier curves to significantly reduce the search space. Extensive experiments demonstrate that LPES effectively mitigates positional attention bias and delivers consistent improvements across multiple long-context benchmarks, yielding up to an 11.2% accuracy gain on the key-value retrieval dataset.
Mitigating Hallucinations in VLMs: Enhancing Visual Attention via Head-Wise Perturbation
Zhenghua Wang | Yixin Wu | Feiran Zhang | Qi Qian | Changze Lv | Xuanjing Huang | Xiaoqing Zheng
Findings of the Association for Computational Linguistics: ACL 2026
Zhenghua Wang | Yixin Wu | Feiran Zhang | Qi Qian | Changze Lv | Xuanjing Huang | Xiaoqing Zheng
Findings of the Association for Computational Linguistics: ACL 2026
Vision–Language Models (VLMs) have demonstrated strong capabilities in tasks that require joint understanding of text and images. However, as many VLMs are built upon pre-trained large language models, they often over-rely on linguistic priors at the expense of visual features, causing persistent hallucinations. We observe that these hallucinations stem not only from insufficient visual attention but also from imbalanced activation profiles across attention heads, while hallucinated samples tend to disproportionately activate heads that fail to capture visual cues. To promote a more balanced attention distribution, we propose **HWP**, a strategy that incorporates head-wise attention perturbation via continuous multiplicative noise, coupled with a visual-guided loss focused on vision-sensitive text tokens. Beyond simply strengthening visual grounding, this design encourages a broader set of attention heads to engage with visual signals, thereby alleviating information loss caused by activation concentration on a few dominant heads. Consistent gains across different architectures and scales on multiple benchmarks demonstrate the effectiveness and robustness of our approach in mitigating VLM hallucinations.
VIB-Probe: Detecting and Mitigating Hallucinations in Vision-Language Models via Variational Information Bottleneck
Feiran Zhang | Yixin Wu | Zhenghua Wang | Xiaohua Wang | Changze Lv | Xuanjing Huang | Xiaoqing Zheng
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Feiran Zhang | Yixin Wu | Zhenghua Wang | Xiaohua Wang | Changze Lv | Xuanjing Huang | Xiaoqing Zheng
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Vision-Language Models (VLMs) have demonstrated remarkable progress in multimodal tasks, but remain susceptible to hallucinations, where generated text deviates from the underlying visual content. Existing hallucination detection methods primarily rely on output logits or external verification tools, often overlooking their internal mechanisms. In this work, we investigate the outputs of internal attention heads, postulating that specific heads carry the primary signals for truthful generation. However, directly probing these high-dimensional states is challenging due to the entanglement of visual-linguistic syntax and noise. To address this, we propose VIB-Probe, a novel hallucination detection and mitigation framework leveraging the Variational Information Bottleneck (VIB) theory. Our method extracts discriminative patterns across layers and heads while filtering out semantic nuisances through the information bottleneck principle. Furthermore, by leveraging the gradients of our VIB probe, we identify attention heads with strong causal influence on hallucinations and introduce an inference-time intervention strategy for hallucination mitigation. Extensive experiments across diverse benchmarks demonstrate that VIB-Probe significantly outperforms existing baselines in both settings. Our code will be made publicly available.