Search papers, labs, and topics across Lattice.
This paper introduces Learning Inference-time Modality Enhancement (LIME), a training-free framework to mitigate hallucinations in MLLMs by enhancing the utilization of perceptual inputs during decoding. LIME uses Layer-wise Relevance Propagation (LRP) to quantify token-level contributions and then enforces a relevance-based objective that promotes increased reliance on perceptual inputs via inference-time updates to the model's key-value representations. Experiments across vision and audio domains demonstrate that LIME reduces hallucinations and enhances grounding without sacrificing generation quality.
MLLMs hallucinate less when you nudge them to pay more attention to non-text inputs during inference, without any training.
Multimodal large language models (MLLMs) have revolutionized the landscape of AI, demonstrating impressive capabilities in tackling complex vision and audio-language tasks. However, a critical challenge remains: these models often suffer from hallucinations, generating outputs that diverge from the provided perceptual inputs. This tendency stems from an inherent imbalance in modality utilization during inference, where the dominance of textual tokens undermines the potential of perceptual inputs. As a result, the model frequently resorts to textual language priors at the expense of grounded evidence. To tackle this issue, we propose Learning Inference-time Modality Enhancement (LIME), a training-free framework designed to bolster multimodal grounding by explicitly enhancing modality usage during decoding. LIME leverages Layer-wise Relevance Propagation (LRP) to quantify token-level contributions and defines a relevance-based objective that promotes increased reliance on perceptual inputs. This objective is enforced through inference-time updates to the model's key-value representations, without modifying model parameters or requiring additional training data. We evaluate LIME across multiple multimodal benchmarks in both vision and audio domains, demonstrating consistent reductions in hallucinations and enhanced grounding while preserving generation quality. Further analysis shows that LIME increases modality contribution and produces more localized and semantically aligned relevance patterns.
