中文

IFCap: Image-like Retrieval and Frequency-based Entity Filtering for Zero-shot Captioning

计算机视觉与模式识别 2024-09-27 v1 人工智能 计算与语言 机器学习

摘要

Recent advancements in image captioning have explored text-only training methods to overcome the limitations of paired image-text data. However, existing text-only training methods often overlook the modality gap between using text data during training and employing images during inference. To address this issue, we propose a novel approach called Image-like Retrieval, which aligns text features with visually relevant features to mitigate the modality gap. Our method further enhances the accuracy of generated captions by designing a Fusion Module that integrates retrieved captions with input features. Additionally, we introduce a Frequency-based Entity Filtering technique that significantly improves caption quality. We integrate these methods into a unified framework, which we refer to as IFCap (I\textbf{I}mage-like Retrieval and F\textbf{F}requency-based Entity Filtering for Zero-shot Cap\textbf{Cap}tioning). Through extensive experimentation, our straightforward yet powerful approach has demonstrated its efficacy, outperforming the state-of-the-art methods by a significant margin in both image captioning and video captioning compared to zero-shot captioning based on text-only training.

引用

@article{arxiv.2409.18046,
  title  = {IFCap: Image-like Retrieval and Frequency-based Entity Filtering for Zero-shot Captioning},
  author = {Soeun Lee and Si-Woo Kim and Taewhan Kim and Dong-Jin Kim},
  journal= {arXiv preprint arXiv:2409.18046},
  year   = {2024}
}

备注

Accepted to EMNLP 2024