【深度学习】Align before Fuse: Vision and Language Representation Learning with Momentum Distillation

Posted by ShawnD on February 17, 2022

Abstract

大规模视觉语言表征学习基于 Transformer 模型取得了极大进展。但是由于 visual token 和 word token 没有对齐, 多模态编码器学习图像-文本交互仍然具有挑战性。

这篇文章引入一种对比损失在融合之前通过跨模态注意力对齐图像和文本表征。

为了从网络上的噪声数据中学习, 这篇文章提出动量蒸馏的自训练方法, 其可以学习一个动量模型产生的伪标签。

这篇文章从互信息最大角度提供了 ALBEF 的理论分析, 表明不同的训练任务可以解释为生成图像-文本对视图的不同方式。

Conclusion

这篇文章提出一种视觉语言表征学习方法, 叫做 ALBEF。

ALBEF 使用一个多模态编码器将单模态文本表征和图像表征在融合之前对齐。

在理论上和实验上验证了所提出图像-文本对比学习和动量蒸馏的有效性。

与其他方法相比, ALBEF 有更好的性能以及在下游任务上有更快的推理速度。