轻量级视觉语言模型训练新方法,小数据也能高效学习。
ESsEN: Training Compact Discriminative Vision-Language Transformers in a Low-Resource Setting

- 采用双塔结构+卷积网络,提升参数效率
- 仅用少量参数实现与大模型相当的性能
- 适合边缘设备或资源受限场景研究者使用
视觉语言建模日益流行,但多数模型参数达数十亿,难以在边缘设备或独立机器人平台上部署。本文受儿童语言发展和数据稀疏性的启发,系统性地探索轻量级模型与小数据训练。实验表明,在低资源环境下,双塔编码器优于单塔结构;将传统卷积网络融入双塔Transformer架构可显著提升参数效率;此外,双塔模型的跨模态融合模块在形状和规模上存在较大灵活性,仍能保持一致性能。我们提出ESsEN,一个可端到端训练、资源需求少的紧凑型视觉语言模型,在多个任务上仅用极少参数即达到与现有模型相当的效果。实验结果与工具使视觉语言建模更易被广泛研究人员使用。
原文摘要 · Abstract (English)
Vision-language modeling is rapidly increasing in popularity with an ever expanding list of available models. In most cases, these vision-language models have parameters in the tens of billions, which is necessary for some needs, but in many cases smaller models are necessary (e.g., on edge devices or independent robotic platforms). Unfortunately, there is little research in producing light-weight models or in training them with small datasets. Inspired by the language learning progression and data sparsity in child development, in this paper, we address both of these goals in a systematic fashion. We show that two-tower encoder models are superior to one-tower encoders in low-resource settings for discriminative English tasks. We show also that incorporating traditional convolutional networks into the two-tower transformer architecture can help produce parameter efficient vision-language models. Finally, we show that the cross-modal fusion module of two-tower encoders can vary significantly in shape and size while producing the same results. In addition, we present ESsEN, a compact vision-language model that can be trained end-to-end with relatively few resources that performs as well on several tasks with only a fraction of the parameters compared to other models. The experimental results and the tools we present here make vision-language modeling more accessible to a wider variety of researchers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。