arXiv:2410.09472cs.SDcs.AI2024-10被引 24

用检索增强生成让模型零样本生成音频描述,不需配对数据

DRCap: Decoding CLAP Latents with Retrieval-Augmented Generation for Zero-shot Audio Captioning

  • 用CLAP文本编码器预训练,推理时换音频编码器实现零样本生成
  • 通过投影和检索增强,使音频嵌入与文本语义对齐,提升描述准确性
  • 无需微调即可适应新领域,适合快速部署到不同场景的音频理解任务

尽管自动音频描述(AAC)已取得显著进展,传统全监督模型仍面临两大挑战:训练需昂贵的音视频配对数据,跨领域迁移时性能下降。为此,我们提出DRCap,一种数据高效、灵活的零样本音频描述系统,仅需文本数据训练,且无需额外微调即可快速适配新领域。DRCap结合对比语言-音频预训练(CLAP)模型与大语言模型(LLM)作为骨干网络。训练时,模型使用固定文本编码器预测真实描述;推理时,文本编码器替换为音频编码器,实现零样本生成。为缓解CLAP的模态差距,采用编码器侧投影策略与解码器侧检索增强生成策略。首先,将音频嵌入投影至文本嵌入支持集,吸收联合多模态空间中的丰富语义信息;同时,从数据存储中检索相似描述作为提示输入给LLM,引入外部知识以充分发挥其强大生成能力。在投影后的CLAP嵌入与检索到的相似描述共同指导下,模型可生成更准确、语义更丰富的文本描述。通过针对目标领域定制文本支持集与描述数据存储,DRCap实现了无训练的强域适应能力。实验表明,DRCap在同域场景下优于所有其他零样本模型,在跨域场景下达到领先性能。

原文摘要 · Abstract (English)

While automated audio captioning (AAC) has made notable progress, traditional fully supervised AAC models still face two critical challenges: the need for expensive audio-text pair data for training and performance degradation when transferring across domains. To overcome these limitations, we present DRCap, a data-efficient and flexible zero-shot audio captioning system that requires text-only data for training and can quickly adapt to new domains without additional fine-tuning. DRCap integrates a contrastive language-audio pre-training (CLAP) model and a large-language model (LLM) as its backbone. During training, the model predicts the ground-truth caption with a fixed text encoder from CLAP, whereas, during inference, the text encoder is replaced with the audio encoder to generate captions for audio clips in a zero-shot manner. To mitigate the modality gap of the CLAP model, we use both the projection strategy from the encoder side and the retrieval-augmented generation strategy from the decoder side. Specifically, audio embeddings are first projected onto a text embedding support to absorb extensive semantic information within the joint multi-modal space of CLAP. At the same time, similar captions retrieved from a datastore are fed as prompts to instruct the LLM, incorporating external knowledge to take full advantage of its strong generative capability. Conditioned on both the projected CLAP embedding and the retrieved similar captions, the model is able to produce a more accurate and semantically rich textual description. By tailoring the text embedding support and the caption datastore to the target domain, DRCap acquires a robust ability to adapt to new domains in a training-free manner. Experimental results demonstrate that DRCap outperforms all other zero-shot models in in-domain scenarios and achieves state-of-the-art performance in cross-domain scenarios.

音频描述零样本检索增强CLAP

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