Suppressing the EOS token during sequence training makes pretrained VLMs generate longer, more detailed image captions, with a trade-off of more hallucinations.
Learning Descriptive Image Captioning via Semipermeable Maximum Likelihood Estimation
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Image captioning aims to describe visual content in natural language. As 'a picture is worth a thousand words', there could be various correct descriptions for an image. However, with maximum likelihood estimation as the training objective, the captioning model is penalized whenever its prediction mismatches with the label. For instance, when the model predicts a word expressing richer semantics than the label, it will be penalized and optimized to prefer more concise expressions, referred to as conciseness optimization. In contrast, predictions that are more concise than labels lead to richness optimization. Such conflicting optimization directions could eventually result in the model generating general descriptions. In this work, we introduce Semipermeable MaxImum Likelihood Estimation (SMILE), which allows richness optimization while blocking conciseness optimization, thus encouraging the model to generate longer captions with more details. Extensive experiments on two mainstream image captioning datasets MSCOCO and Flickr30K demonstrate that SMILE significantly enhances the descriptiveness of generated captions. We further provide in-depth investigations to facilitate a better understanding of how SMILE works.
fields
cs.CV 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
The Devil is in the EOS: Sequence Training for Detailed Image Captioning
Suppressing the EOS token during sequence training makes pretrained VLMs generate longer, more detailed image captions, with a trade-off of more hallucinations.