Search papers, labs, and topics across Lattice.
This paper introduces UniProbe, a novel learnable detector that identifies token-level hallucinations in large vision-language models (LVLMs) by analyzing their internal computational traces. By constructing a directed graph that captures the relationships between image patches, query tokens, and generated tokens, UniProbe employs a combination of graph neural networks, vision transformers, and gated recurrent units to effectively localize hallucinations without the need for full model fine-tuning. The method demonstrates significant improvements in object hallucination detection, reducing such occurrences by up to 55% during decoding while maintaining competitive latency.
UniProbe slashes object hallucinations in LVLMs by 55% during generation, all while operating with minimal latency.
Large Vision-Language Models (LVLMs) achieve impressive visual reasoning and dialogue capabilities, yet frequently hallucinate content unsupported by the visual input. Effective mitigation requires token-level localization, enabling targeted intervention without discarding the entire response. Existing detectors require expensive full-model fine-tuning, rely on external verifiers that ignore the model's generation process, or reduce internal signals to isolated features and hand-crafted statistics, discarding spatial, sequential, and relational structure. We introduce UniProbe, a lightweight, unified, learnable detector that models a frozen LVLM's heterogeneous computational trace from a single forward pass. UniProbe constructs a directed graph over image patches, query tokens, and generated tokens, with attention weights encoding their relations. It processes this trace with alternating structure-aware modules: a GNN for relational evidence, a ViT for 2-D visual geometry, and a GRU for response order. Interleaving them allows spatial, relational, and sequential evidence to interact throughout the detector. We further develop a streaming variant for hallucination-aware decoding, which detects and resamples hallucinated tokens during generation, and a self-adaptation strategy aligning the detector with the LVLM's own generations. Across diverse LVLM backbones, UniProbe achieves state-of-the-art token-level and object-hallucination detection. During decoding, it reduces object hallucinations by up to 55\% at 1.06times the latency of standard generation.
NVIDIA