arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.01984cs.CVcs.CL
收藏

通过紧凑双向架构实现图像描述

Image Captioning via Compact Bidirectional Architecture

School of Big Data and Statistics, Anhui University(安徽大学大数据与统计学院);Institute of Artificial Intelligence, Hefei Comprehensive National Science Center(合肥综合性国家科学中心人工智能研究院);Intelligent Interconnected Systems Laboratory of Anhui Province (Hefei University of Technology)(安徽省智能互联系统实验室(合肥工业大学));School of Computer Science and Information Engineering, Hefei University of Technology(合肥工业大学计算机与信息学院);JD Explore Academy, JD.com Inc(京东探索研究院);State Key Laboratory of Digital Intelligent Technology for Unmanned Coal Mining, Anhui University of Science and Technology(安徽理工大学深部煤矿采动响应与灾害防控国家重点实验室)

Zijie Song, Yuanen Zhou, Zhenzhen Hu, Daqing Liu, Huixia Ben, Richang Hong, Meng Wang

AI总结:

本文提出紧凑双向Transformer模型,通过并行解码器隐式和显式利用双向上下文,经消融研究发现其比显式交互机制更重要,结合词级融合进一步提升效果,并扩展至LSTM基座实现新SOTA结果。

摘要:

大多数当前图像描述模型通常从左到右生成描述。这种单向性质使它们只能利用过去的上下文,而不能利用未来的上下文。尽管基于细化的模型可以通过在第二阶段基于预检索或预生成的描述生成新的描述来利用过去和未来的上下文,但这些模型的解码器通常由两个网络(即第一阶段的检索器或描述生成器和第二阶段的描述生成器)组成,只能顺序执行。在本文中,我们介绍了一种紧凑的双向Transformer模型用于图像描述,该模型可以隐式和显式地利用双向上下文,同时解码器可以并行执行。具体而言,它通过紧密耦合从左到右(L2R)和从右到左(R2L)的流到一个紧凑的模型中,作为隐式利用双向上下文的正则化,并可选地允许双向流的显式交互,而最终的描述是从L2R或R2L流在句子层面进行融合后选择的。我们在MSCOCO基准上进行了广泛的消融研究,发现紧凑的双向架构和句子层面的融合比显式交互机制更重要。通过与词级融合无缝结合,句子层面的融合效果进一步扩大。我们进一步将传统的单流自批评训练扩展到双流版本,从而在与非视觉-语言预训练模型的比较中取得了新的SOTA结果。最后,我们通过将此紧凑的双向架构扩展到LSTM基座来验证其通用性。源代码可在https://github.com/YuanEZhou/cbtic上获得。

英文摘要:

Most current image captioning models typically generate captions from left-to-right. This unidirectional property makes them can only leverage past context but not future context. Though refinement-based models can exploit both past and future context by generating a new caption in the second stage based on pre-retrieved or pre-generated captions in the first stage, the decoder of these models generally consists of two networks~(i.e. a retriever or captioner in the first stage and a captioner in the second stage), which can only be executed sequentially. In this paper, we introduce a Compact Bidirectional Transformer model for image captioning that can leverage bidirectional context implicitly and explicitly while the decoder can be executed parallelly. Specifically, it is implemented by tightly coupling left-to-right(L2R) and right-to-left(R2L) flows into a single compact model to serve as a regularization for implicitly exploiting bidirectional context and optionally allowing explicit interaction of the bidirectional flows, while the final caption is chosen from either L2R or R2L flow in a sentence-level ensemble manner. We conduct extensive ablation studies on MSCOCO benchmark and find that the compact bidirectional architecture and the sentence-level ensemble play more important roles than the explicit interaction mechanism. By combining with word-level ensemble seamlessly, the effect of sentence-level ensemble is further enlarged. We further extend the conventional one-flow self-critical training to the two-flows version under this architecture and achieve new state-of-the-art results in comparison with non-vision-language-pretraining models. Finally, we verify the generality of this compact bidirectional architecture by extending it to LSTM backbone. Source code is available at https://github.com/YuanEZhou/cbtic.

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

↑