M<mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML" altimg="si6.svg" display="inline" id="d1e2549"><mml:msup><mml:mrow/><mml:mrow><mml:mn>3</mml:mn></mml:mrow></mml:msup></mml:math>ixup: A multi-modal data augmentation approach for image captioning

Abstract

Despite the great success, most models in image captioning (IC) are still stuck in the dilemma of generating simple and non-discriminative captions. In this paper, we study this problem from the perspective of data augmentation and propose a novel method called Multi-modal Mixup ( M 3 ixup). Compared with the original Mixup strategy designed for image classification , the proposed M 3 ixup has three novel designs to mix IC samples from the aspects of visual features, sentence embeddings and loss values, respectively. In practice, M 3 ixup can not only enrich the diversity of IC training data , but also enforce the model to focus more on visual information for captioning, thereby alleviating the negative effect of dataset bias and addressing the issue of simple captioning. To validate M 3 ixup, we apply it to three baseline models and conduct extensive experiments on MS COCO. The experimental results demonstrate that our proposed M 3 ixup can not only improve the discriminability and quality of generated captions, but also help the baseline models obtain obvious performance gains, i.e. , improving the CIDEr scores of the state-of-the-art model from 133.8 to 135.3 on off-line testing and 135.4 to 137.1 on online testing.

Topics & Concepts