arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2209.08231cs.CV
收藏

学习独特且具代表性的图像描述风格

Learning Distinct and Representative Styles for Image Captioning

Australian Institute for Machine Learning(澳大利亚机器学习研究所);University of Adelaide(阿德莱德大学)

Qi Chen, Chaorui Deng, Qi Wu

AI总结:

针对图像描述的模式坍塌问题,提出离散模式学习 DML,通过 CdVAE 学习可区分、具代表性的模式嵌入,并注入 MIC 分支控制生成风格,在 MSCOCO 上提升 Transformer 与 AoANet 的多样性和质量。

摘要:

多年来,最先进(SoTA)图像描述方法已在某些评估指标(例如 CIDEr)上取得了令人鼓舞的结果。然而,近期发现表明,这些方法生成的描述往往偏向于“平均”描述,即只捕捉训练语料中最普遍的模式(也称为语言模式),也就是所谓的模式坍塌问题。受此影响,生成描述的多样性受限,并且通常不如人类给出的自然图像描述信息丰富。本文中,我们试图通过提出一种用于图像描述的离散模式学习(DML)范式来避免这一问题。我们的创新思路是挖掘训练描述语料中丰富的模式,以学习一组“模式嵌入”,并进一步利用它们控制现有图像描述模型生成描述的模式。具体而言,所提出的 DML 优化一个双分支架构,包括图像条件离散变分自编码器(CdVAE)分支和模式条件图像描述(MIC)分支。CdVAE 分支将每条图像描述映射到学习码本中存储的某个模式嵌入,并使用纯非自回归生成目标进行训练,以使各模式既独特又具代表性。MIC 分支可由现有图像描述模型简单修改得到,其中模式嵌入被添加到原始词嵌入中作为控制信号。实验中,我们将所提出的 DML 应用于两个广泛使用的图像描述模型 Transformer 和 AoANet。结果表明,学习到的模式嵌入成功帮助这些模型生成具有不同模式的高质量图像描述,进而在 MSCOCO 数据集上同时带来更好的多样性和质量表现。

英文摘要:

Over the years, state-of-the-art (SoTA) image captioning methods have achieved promising results on some evaluation metrics (e.g., CIDEr). However, recent findings show that the captions generated by these methods tend to be biased toward the "average" caption that only captures the most general mode (a.k.a, language pattern) in the training corpus, i.e., the so-called mode collapse problem. Affected by it, the generated captions are limited in diversity and usually less informative than natural image descriptions made by humans. In this paper, we seek to avoid this problem by proposing a Discrete Mode Learning (DML) paradigm for image captioning. Our innovative idea is to explore the rich modes in the training caption corpus to learn a set of "mode embeddings", and further use them to control the mode of the generated captions for existing image captioning models. Specifically, the proposed DML optimizes a dual architecture that consists of an image-conditioned discrete variational autoencoder (CdVAE) branch and a mode-conditioned image captioning (MIC) branch. The CdVAE branch maps each image caption to one of the mode embeddings stored in a learned codebook, and is trained with a pure non-autoregressive generation objective to make the modes distinct and representative. The MIC branch can be simply modified from an existing image captioning model, where the mode embedding is added to the original word embeddings as the control signal. In the experiments, we apply the proposed DML to two widely used image captioning models, Transformer and AoANet. The results show that the learned mode embedding successfully facilitates these models to generate high-quality image captions with different modes, further leading to better performance for both diversity and quality on the MSCOCO dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑