arXiv:2409.11059cs.CVcs.LG2024-09被引 4

用轻量框架逐步对齐图文音视频,小数据也能超好用。

OneEncoder: A Lightweight Framework for Progressive Alignment of Modalities

  • 先训轻量通用投影模块对齐图像和文本,再逐步接入新模态。
  • 在小数据集上训练,分类、问答等任务表现超越大模型。
  • 无需重训,加新模态只需对接已有对齐结果,适合资源有限场景。

跨模态对齐学习通过整合文本、图像、音频和视频等多源信息构建统一表示,实现视觉问答与音视频内容分析等应用。现有方法依赖大型专用编码器,需在大规模对齐数据集(如图文、文本音频、图像音频)上微调或从头训练,存在三大局限:(i) 训练成本高昂;(ii) 获取大规模成对数据困难;(iii) 增加新模态需重训整个框架。为此,我们提出 OneEncoder,一种轻量级渐进式多模态对齐框架,支持图像、文本、音频、视频四模态。首先训练一个轻量级通用投影模块(UP)对齐图像与文本;随后冻结预训练的 UP,逐步将新模态对齐至已对齐的模态空间。该框架高效且低成本,即使在缺乏大规模对齐数据时仍表现优异。在小型配对数据集上训练后,其在分类、查询与视觉问答等任务中性能优于依赖大数据与专用编码器的方法。

原文摘要 · Abstract (English)

Cross-modal alignment Learning integrates information from different modalities like text, image, audio and video to create unified models. This approach develops shared representations and learns correlations between modalities, enabling applications such as visual question answering and audiovisual content analysis. Current techniques rely on large modality-specific encoders, necessitating fine-tuning or training from scratch on vast aligned datasets (e.g., text-image, text-audio, image-audio). This approach has limitations: (i) it is very expensive due to the need for training large encoders on extensive datasets, (ii) acquiring aligned large paired datasets is challenging, and (iii) adding new modalities requires retraining the entire framework to incorporate these modalities. To address these issues, we propose OneEncoder, a lightweight framework that progressively represents and aligns four modalities (image, text, audio, video). Initially, we train a lightweight Universal Projection module (UP) to align image and text modalities. Then, we freeze the pretrained UP and progressively align future modalities to those already aligned. OneEncoder operates efficiently and cost-effectively, even in scenarios where vast aligned datasets are unavailable, due to its lightweight design. Trained on small paired datasets, it shows strong performance in tasks like classification, querying, and visual question answering, surpassing methods that rely on large datasets and specialized encoders.

多模态轻量化渐进对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。