Source-linked AI summary

Plausible May Not Be Faithful: Probing Object Hallucination in Vision-Language Pre-training

Wenliang Dai, Zihan Liu, Ziwei Ji, Dan Su, Pascale Fung

arXiv:2210.07688v2cs.CLcs.CV

TL;DR

VLP models can generate fluent captions containing objects that are not present in the image, and the effects of training strategies and image encodings on this faithfulness problem were unclear. The paper systematically probes models, encodings, and objectives, and introduces ObjMLM; it reports reduced hallucination alongside findings about patch features and generation objectives.

  • Problem

    VLP models can generate fluent text with nonexistent or inaccurate objects, while the effects of VLP strategies and image encodings on faithfulness had not been studied.

    Source: §1 · p. 1, §2.2 · p. 3
  • Method

    The paper systematically examines VLP models, image encodings, and training objectives, and proposes ObjMLM to enhance alignment between text tokens and visual objects during generation.

    Source: §1 · p. 2
  • Results

    17.4%: ObjMLM reduces object hallucination; patch-based features and smaller patch resolution help, while token-level alignment and generation controllability are crucial.

    Source: §1 · p. 2, §7 · p. 9
  • Takeaways & Limitations

    The authors present their findings as useful for future work building more responsible VLP models.

    Source: §7 · p. 9

    The study focuses on object hallucination, and its analysis of recent VLP models is limited to those with publicly available pre-trained checkpoints.

    Source: Limitations · p. 9

Abstract

from arXiv · show

Large-scale vision-language pre-trained (VLP) models are prone to hallucinate non-existent visual objects when generating text based on visual information. In this paper, we systematically study the object hallucination problem from three aspects. First, we examine recent state-of-the-art VLP models, showing that they still hallucinate frequently, and models achieving better scores on standard metrics (e.g., CIDEr) could be more unfaithful. Second, we investigate how different types of image encoding in VLP influence hallucination, including region-based, grid-based, and patch-based. Surprisingly, we find that patch-based features perform the best and smaller patch resolution yields a non-trivial reduction in object hallucination. Third, we decouple various VLP objectives and demonstrate that token-level image-text alignment and controlled generation are crucial to reducing hallucination. Based on that, we propose a simple yet effective VLP loss named ObjMLM to further mitigate object hallucination. Results show that it reduces object hallucination by up to 17.4% when tested on two benchmarks (COCO Caption for in-domain and NoCaps for out-of-domain evaluation).

1 Introduction

The paper investigates object hallucination in VLP, where generated captions can sound plausible yet name objects absent from or inaccurate to the image. It examines current models, image encodings, and pre-training objectives, then proposes ObjMLM to mitigate the problem.

1 Introduction

  • Object hallucination can limit VLP performance and raise safety concerns, including reduced diagnostic accuracy and potential severe consequences in biomedical image captioning.Source: §1 · p. 1
  • Recent state-of-the-art VLP models still hallucinate frequently, with about 10% of generated sentences containing at least one hallucinated object.The problem is more severe for out-of-domain images; SCST can increase hallucination even while improving CIDEr.Source: §1 · p. 1
  • Patch-based features perform best among region-, grid-, and patch-based encodings, and smaller patch resolution non-trivially reduces object hallucination.Source: §1 · p. 2
  • ITC and ITM contribute little to alleviating hallucination, while generative ICLM is the main influential factor and pre-training datasets yield distinctive behaviors.Source: §1 · p. 2
  • The study systematically examines modern VLP object hallucination and how image encoding and pre-training objectives affect it.The paper frames these investigations around measuring hallucination and finding ways to mitigate it.Source: §1 · pp. 1–2
  • 17.4%: ObjMLM reduces object hallucination without introducing extra training data.The loss enhances alignment and restriction between text tokens and visual objects during generation.Source: §1 · p. 2

2 Related Work

Related work defines hallucination as unfaithfulness to conditional input and surveys mitigation approaches across generation tasks. VLP research has evolved across architectures and objectives, while the effects of its training strategies and image encodings on faithfulness remain unclear.

2.1 Hallucination in Deep Learning

  • Hallucination means output unfaithful to conditional input, and prior mitigation approaches include data refinement, visual grounding, reconstruction, decoding, knowledge augmentation, and post-processing.Source: §2.1 · p. 2

2.2 Vision-Language Pre-training

  • VLP developed from BERT-style multimodal understanding objectives to encoder-decoder generation with causal language modeling and dual-stream models aligned by image-text contrastive loss.Source: §2.2 · pp. 2–3
  • Prior work showed object hallucination in modern VLP, but how VLP strategies and region-, grid-, or patch-based image encodings affect generated-text faithfulness remained unstudied or unclear.Source: §2.2 · p. 3

3 Evaluation Setup

The evaluation setup defines CHAIR to measure object hallucination and tests captioning models on COCO Caption and NoCaps, using COCO for in-domain and NoCaps for out-of-domain evaluation.

3.1 Evaluation Metric

  • CHAIRi measures the share of predicted objects that are hallucinated, while CHAIRs measures the share of generated sentences containing at least one hallucinated object.Both variants rely on predefined golden object categories.Source: §3.1 · p. 3

3.2 Evaluation Datasets

  • Models are finetuned on COCO Caption training data and evaluated on its test set and the NoCaps validation set for in-domain and out-of-domain performance, respectively.Source: §3.2 · p. 3

3.2.1 COCO Caption

  • The COCO Caption Karpathy split contains 82K training, 5K validation, and 5K test images, each annotated with at least five captions.CHAIR calculation maps fine-grained object words to 80 coarse-grained categories using synonyms to avoid false hallucination detections.Source: §3.2.1 · p. 3

3.2.2 NoCaps

  • NoCaps evaluates generalization to a wider range of visual concepts, including unseen object categories, using its 4,500-image validation set.The evaluation maps 600 classes to 139 coarse-grained object categories; the test set is not used because ground-truth captions are unavailable.Source: §3.2.2 · pp. 3–4

4 Object Hallucination in VLP Models

The evaluated VLP captioners still produce object hallucinations, and standard caption metrics do not reliably reflect faithfulness. SCST can improve conventional scores while increasing hallucination, particularly for out-of-domain images.

  • The models are evaluated after COCO Caption training on both the COCO test set and the NoCaps validation set.Source: §4 · p. 4
  • Higher standard scores do not guarantee greater faithfulness: VinVLLarge scores higher than VinVLBase on CIDEr and BLEU-4 but also has higher CHAIR scores.The authors recommend supplementing conventional metrics with CHAIR to assess faithfulness.Source: §4 · pp. 4–5
  • SCST raises VinVLBase CIDEr by 11.1 and BLEU-4 by 2.7, while increasing its COCO Caption CHAIRs score by 0.9.SCST optimizes CIDEr using sampled captions and is used after cross-entropy finetuning.Source: §4 · p. 5
  • 10.9% more VinVLBase captions contain at least one hallucinated object after SCST, and the problem is more serious on out-of-domain images.The side effect is alleviated for a very well-pretrained model such as OFA large.Source: §4 · p. 5
  • After SCST, captions can add detailed but incorrect objects; the paper's example includes a generated “mirror” absent from the image.The authors note this may further amplify hallucination on unfamiliar images, where model confidence may be lower.Source: §4 · p. 5

5 Probing Image Encoding Methods and VLP Objectives

The section probes visual encoders and pretraining objectives for their relationship to object hallucination, using a flexible architecture that varies visual encoding while holding the text decoder constant.

5 Probing Image Encoding Methods and VLP Objectives

  • The experiments probe region-, grid-, and patch-based image features alongside ITC, ITM, and ICLM objectives and their variants.Source: §5 · p. 5

5.1 Model Architecture

  • The flexible architecture pairs a visual encoder with a text decoder, allowing the visual encoder to change while the same BertLM decoder is retained.The explored visual encoders include region-based Faster R-CNN, grid-based CLIP ResNet, and patch-based CLIP ViT variants.Source: §5.1 · pp. 5–6

5.2 Effects of Different Image Features

  • Patch-based encoders perform best at avoiding hallucination, and smaller patch resolution further reduces it without increasing model size.Grid encoders hallucinate more than comparable-CIDEr alternatives; the authors conjecture CNN inductive biases may weaken connections among an object's characteristics.Source: §5.2 · p. 6

5.3.2 Image-Text Contrastive (ITC) Loss

  • Vanilla ITC has negligible influence on caption faithfulness, whereas token-level ITCLate slightly reduces hallucination.The authors attribute ITCLate's benefit to word-patch alignment.Source: §5.3.2 · pp. 6–7

5.3.3 Image-Text Matching (ITM) Loss

  • ITM and ITMHard are not highly correlated with hallucination and only slightly reduce it on out-of-domain images.ITMHard has negligible effect on hallucination in downstream generative tasks.Source: §5.3.3 · p. 7

5.3.4 Image-Conditioned Language Modeling

  • VG causal-LM pretraining reduces hallucination compared with the baseline even though it does not improve CIDEr.Source: §5.3.4 · pp. 7–8
  • VG captions are shorter and focus on specific image aspects, while COCO captions give broader descriptions; VG pretraining therefore creates a fine-to-coarse training process.After adding VG pretraining, captions contain on average 0.08 fewer objects on COCO and 0.24 fewer on NoCaps, and may omit details when the model is uncertain.Source: Figure 2 · p. 7, Figure 3 · p. 7, §5.3.4 · p. 8

5.3.1 Pre-training Datasets

  • CC3M improves all reported metrics and general image-translation ability, but the authors say data augmentation alone may not drastically solve hallucination.They identify controllability of vision-conditioned text generation as a promising direction.Source: §5.3.4 · p. 8

6 Object Masked Language Modeling

ObjMLM masks image-present object words and phrases to strengthen object-level image-text alignment. In an existing VLP setting, it reduced CHAIRs by 17.4% without extra pre-training data.

6 Object Masked Language Modeling

  • ObjMLM masks object words and phrases that appear in the image to improve object-level image-text alignment.It is described as a variant of MLM that masks objects in the text.Source: §6 · p. 8
  • The ObjMLM architecture pairs an image encoder with a text decoder and uses ViT-L/14 for the image encoder.The overview allows region-, grid-, or patch-based encoders.Source: Figure 4 · p. 8
  • Whole-object masking uses exact matching against COCO and NoCaps object category lists, replacing each object with one [MASK] token.Source: §6 · p. 8
  • 17.4%: plugging ObjMLM into an existing VLP setting reduced CHAIRs, compared with a 1.7% reduction from standard MLM.The authors conjecture ObjMLM constrains generation to visible objects and improves recognition of spatial relationships between objects.Source: §6 · p. 8

7 Conclusion

The conclusion reports that recent large VLP models still hallucinate frequently and that encoding and pre-training objectives matter. It presents ObjMLM as a mitigation, while noting that hallucination remains an important problem for future responsible VLP work.

7 Conclusion

  • Recent large VLP models still hallucinate frequently, and SCST harms faithfulness even when it improves standard metrics.Source: §7 · pp. 8–9
  • Patch-based input with smaller resolution helps mitigate object hallucination, while token-level alignment and generation controllability are described as crucial.Source: §7 · p. 9
  • ObjMLM reduces object hallucination by 17.4% in an existing VLP setting.Source: §7 · p. 9

Limitations

The paper focuses on object hallucination, leaving other VLP hallucination types for future work. Its survey of recent VLP models is limited to those with publicly available pre-trained checkpoints.

Limitations

  • The investigation and mitigation focus on object hallucination, leaving other VLP hallucination types for future work.Source: Limitations · p. 9
  • The recent-model study includes only models with publicly available pre-trained checkpoints.The authors cite limited GPU resources and private pre-training datasets as reasons they could not pre-train unreleased models.Source: Limitations · p. 9

A Implementation Details

Experiments used PyTorch and eight Nvidia V100 GPUs, with CLIP and BERT-base-uncased checkpoints for the image and text components.

A Implementation Details

  • Experiments used PyTorch on 8 Nvidia V100 GPUs, with official CLIP checkpoints and a 110M-parameter bert-base-uncased initialization for BertLM.Source: A Implementation Details · p. 13
  • COCO Caption fine-tuning used batch size 512, AdamW for 10 epochs, learning rate 5 × 10−5, weight decay 1 × 10−2, and 0.85 epochwise decay.Source: A Implementation Details · p. 13
  • Text-generation pre-training used a first-epoch learning-rate warmup; ITC and ITM used batch size 1024 for more negative samples.Source: A Implementation Details · p. 13

B Additional Case Studies

These cases juxtapose ground-truth captions with outputs from several VLP models, showing concrete object mismatches across kitchen, backpack, and cocktail images.

B Additional Case Studies

  • For the kitchen image, the ground truth mentions wooden and blue cabinets, while VinVLbase with SCST adds bottles in the cabinetry.Source: B Additional Case Studies · p. 13
  • For the backpack image, the ground truth specifies a green camouflage print, while VinVLbase describes a helmet on top of a bag.Source: B Additional Case Studies · p. 13
  • For the cocktail image, the ground truth mentions strawberries, while RN50x64 describes red wine and ViT-L/14 with LM & ObjMLM describes glasses of wine.Source: B Additional Case Studies · p. 13
  • Figure 5 marks hallucinated objects in red across additional generated-caption examples from different models.Source: Figure 5 · p. 13
Loading 2210.07688v2…