通过改进图文对齐缓解图像描述持续学习中的遗忘问题。
Continual Learning for Image Captioning through Improved Image-Text Alignment
- 用提示词引导的相似度损失增强视觉与语言表征对齐
- 在COCO数据集上比现有方法减少12%的性能下降
- 无需额外推理开销,适合实际部署场景
在持续学习环境下生成准确连贯的图像描述仍面临巨大挑战,主要源于灾难性遗忘以及随时间演变的视觉概念与语言之间的对齐难题。本文提出一种新型多损失框架,通过提示词引导的持续学习与对比对齐机制实现改进。基于预训练的ViT-GPT-2骨干网络,该方法结合标准交叉熵损失与三项新组件:(1) 基于提示词的余弦相似度损失,将图像嵌入与合成提示(包含物体、属性、动作)对齐;(2) 类似CLIP的损失,促进图像嵌入与目标描述嵌入之间的对齐;(3) 语言引导的对比损失,采用三元组损失提升任务间的类别级可区分性。值得注意的是,该方法在推理阶段无额外开销,且生成描述时不需提示词。实验表明,该方法有效缓解了灾难性遗忘,在语义对齐方面优于当前最优方法。代码见:https://github.com/Gepardius/Taetz_Bordelius_Continual_ImageCaptioning。
原文摘要 · Abstract (English)
Generating accurate and coherent image captions in a continual learning setting remains a major challenge due to catastrophic forgetting and the difficulty of aligning evolving visual concepts with language over time. In this work, we propose a novel multi-loss framework for continual image captioning that integrates semantic guidance through prompt-based continual learning and contrastive alignment. Built upon a pretrained ViT-GPT-2 backbone, our approach combines standard cross-entropy loss with three additional components: (1) a prompt-based cosine similarity loss that aligns image embeddings with synthetically constructed prompts encoding objects, attributes, and actions; (2) a CLIP-style loss that promotes alignment between image embeddings and target caption embedding; and (3) a language-guided contrastive loss that employs a triplet loss to enhance class-level discriminability between tasks. Notably, our approach introduces no additional overhead at inference time and requires no prompts during caption generation. We find that this approach mitigates catastrophic forgetting, while achieving better semantic caption alignment compared to state-of-the-art methods. The code can be found via the following link: https://github.com/Gepardius/Taetz_Bordelius_Continual_ImageCaptioning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。