arXiv:2506.23271cs.CV2025-06TPAMI被引 16

用轻量元令牌压缩视觉音频特征,节省内存且保持精度。

Mettle: Meta-Token Learning for Memory-Efficient Audio-Visual Adaptation

  • 通过分层蒸馏将原始特征转为紧凑元令牌,实现并行压缩。
  • 在多个基准上降低内存占用与训练时间,准确率仍具竞争力。
  • 适合资源受限场景下的跨模态模型微调,尤其适用于视频分割任务。

我们提出一种名为 Mettle(Meta-Token Learning)的简单高效方法,用于将大规模预训练变换器模型适配到下游音视频任务中。与逐层修改输出特征分布不同,Mettle 采用轻量级分层中心蒸馏(LCD)模块,将每个变换器层中的完整音视频特征并行蒸馏为紧凑的元令牌,兼顾预训练知识保留与任务特定适应。所得元令牌可直接用于分类任务,如音视频事件定位和音视频视频解析。为进一步支持细粒度分割任务(如音视频分割),我们引入元令牌注入(MTI)模块,利用顶层蒸馏出的音视频元令牌,指导早期层的特征适应。在多个音视频基准上的大量实验表明,该方法显著降低内存使用与训练时间,同时保持参数效率与竞争性准确率。

原文摘要 · Abstract (English)

We present \textbf{Met}a-\textbf{T}oken \textbf{Le}arning (Mettle), a simple and memory-efficient method for adapting large-scale pretrained transformer models to downstream audio-visual tasks. Instead of sequentially modifying the output feature distribution of the transformer backbone, Mettle utilizes a lightweight \textit{Layer-Centric Distillation (LCD)} module to distill in parallel the intact audio or visual features embedded by each transformer layer into compact meta-tokens. This distillation process considers both pretrained knowledge preservation and task-specific adaptation. The obtained meta-tokens can be directly applied to classification tasks, such as audio-visual event localization and audio-visual video parsing. To further support fine-grained segmentation tasks, such as audio-visual segmentation, we introduce a \textit{Meta-Token Injection (MTI)} module, which utilizes the audio and visual meta-tokens distilled from the top transformer layer to guide feature adaptation in earlier layers. Extensive experiments on multiple audiovisual benchmarks demonstrate that our method significantly reduces memory usage and training time while maintaining parameter efficiency and competitive accuracy.

跨模态特征压缩元令牌音视频

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