Back to Search Start Over

Mitigating Hallucination in Visual-Language Models via Re-Balancing Contrastive Decoding

Authors :
Liang, Xiaoyu
Yu, Jiayuan
Mu, Lianrui
Zhuang, Jiedong
Hu, Jiaqi
Yang, Yuchen
Ye, Jiangnan
Lu, Lu
Chen, Jian
Hu, Haoji
Publication Year :
2024

Abstract

Although Visual-Language Models (VLMs) have shown impressive capabilities in tasks like visual question answering and image captioning, they still struggle with hallucinations. Analysis of attention distribution in these models shows that VLMs tend to processing textual tokens rather than visual tokens. This imbalance of attention distribution causes VLMs to favor textual knowledge in the case of multimodal knowledge conflicts, resulting in differences from the image information. In this paper, we propose Re-Balancing Contrastive Decoding (RBD) method, which employs textual and visual branches to recalibrate attention distribution in VLMs. Specifically, the textual branch injects image noise to stimulate the model's dependency on text, thereby reducing textual bias. Concurrently, the visual branch focuses on the selection of significant tokens, refining the attention mechanism to highlight the primary subject. This dual-branch strategy enables the RBD method to diminish textual bias while enhancing visual information. Experimental results demonstrate that our method, RBD, outperforms the existing methods by the CHAIR and POPE metrics, mitigate hallucinations without reducing the model's general capabilities.<br />Comment: PRCV

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2409.06485
Document Type :
Working Paper