用分层令牌融合提升音视频学习效率,兼顾性能与资源消耗。
MoLT: Mixture of Layer-Wise Tokens for Efficient Audio-Visual Learning
- 仅在深层提取并融合分层令牌,替代逐层计算,降低开销。
- 在多个音视频任务上超越现有方法,提升显著且高效。
- 适合资源受限场景下的多模态模型轻量化部署。
本文提出一种名为分层令牌混合(MoLT)的参数与内存高效的音视频学习适配框架。核心思想是用并行轻量级方案取代传统在每个Transformer层中进行的高计算量顺序适配,仅从深层提取并融合分层令牌。采用两种适配器,以分层方式将模态特异性信息与跨模态交互提炼为紧凑的潜在令牌。一个令牌融合模块动态融合这些分层令牌,考虑其相对重要性。训练中引入潜在令牌间的正交性正则化,防止冗余。通过系统分析预训练Transformer中适配位置的影响,仅在深层提取潜在令牌,避免早期层不稳定的特征传播,从而在保持参数与内存效率的同时最大化适配性能。大量实验表明,MoLT在音频-视觉问答、音频-视觉分割和音频-视觉事件定位等多个基准上均优于现有方法。
原文摘要 · Abstract (English)
In this paper, we propose Mixture of Layer-Wise Tokens (MoLT), a parameter- and memory-efficient adaptation framework for audio-visual learning. The key idea of MoLT is to replace conventional, computationally heavy sequential adaptation at every transformer layer with a parallel, lightweight scheme that extracts and fuses layer-wise tokens only from the late layers. We adopt two types of adapters to distill modality-specific information and cross-modal interaction into compact latent tokens in a layer-wise manner. A token fusion module then dynamically fuses these layer-wise tokens by taking into account their relative significance. To prevent the redundancy of latent tokens, we apply an orthogonality regularization between latent tokens during training. Through the systematic analysis of the position of adaptation in the pre-trained transformers, we extract latent tokens only from the late layers of the transformers. This strategic adaptation approach avoids error propagation from the volatile early-layer features, thereby maximizing the adaptation performance while maintaining parameter and memory efficiency. Through extensive experiments, we demonstrate that MoLT outperforms existing methods on diverse audio-visual benchmarks, including Audio-Visual Question Answering, Audio-Visual Segmentation, and Audio-Visual Event Localization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。