用大模型自动构建190万对音视频描述数据集,提升生成质量。
AudioSetCaps: An Enriched Audio-Caption Dataset using Automated Generation Pipeline with Large Audio and Language Models
- 融合音频语言模型与大语言模型,分步提取细节并生成描述。
- 生成190万组配对数据,音文检索准确率达59.7%(音频转文本)。
- 开源全流程工具与600万数据,适合音频-语言学习研究者使用。
随着音频-语言模型的兴起,构建大规模音视频配对数据集成为模型发展的关键挑战,主要因人工标注耗时费力。尽管大语言模型提升了合成描述的效率,现有方法仍难以有效提取和融合详细音频信息。本文提出一种自动化流程:利用音频-语言模型进行细粒度内容提取,大语言模型生成合成描述,并通过基于对比语言-音频预训练(CLAP)的优化过程提升描述质量。具体地,在内容提取阶段采用提示链技术以获取精准细粒度信息,优化阶段则缓解生成描述中的幻觉问题。基于AudioSet数据集和该方法,我们构建了AudioSetCaps,包含190万组音频-描述对,是当时规模最大的音频-描述数据集。使用该数据集训练的模型在音文检索任务上达到最佳表现:文本到音频检索的R@1为46.3%,音频到文本检索为59.7%;在自动音频描述任务中CIDEr得分为84.8。此外,基于Youtube-8M和VGGSound数据集,我们进一步生成410万组合成数据。为促进音频-语言学习研究,我们已将整个流程、600万对数据及预训练模型公开于https://github.com/JishengBai/AudioSetCaps。
原文摘要 · Abstract (English)
With the emergence of audio-language models, constructing large-scale paired audio-language datasets has become essential yet challenging for model development, primarily due to the time-intensive and labour-heavy demands involved. While large language models (LLMs) have improved the efficiency of synthetic audio caption generation, current approaches struggle to effectively extract and incorporate detailed audio information. In this paper, we propose an automated pipeline that integrates audio-language models for fine-grained content extraction, LLMs for synthetic caption generation, and a contrastive language-audio pretraining (CLAP) model-based refinement process to improve the quality of captions. Specifically, we employ prompt chaining techniques in the content extraction stage to obtain accurate and fine-grained audio information, while we use the refinement process to mitigate potential hallucinations in the generated captions. Leveraging the AudioSet dataset and the proposed approach, we create AudioSetCaps, a dataset comprising 1.9 million audio-caption pairs, the largest audio-caption dataset at the time of writing. The models trained with AudioSetCaps achieve state-of-the-art performance on audio-text retrieval with R@1 scores of 46.3% for text-to-audio and 59.7% for audio-to-text retrieval and automated audio captioning with the CIDEr score of 84.8. As our approach has shown promising results with AudioSetCaps, we create another dataset containing 4.1 million synthetic audio-language pairs based on the Youtube-8M and VGGSound datasets. To facilitate research in audio-language learning, we have made our pipeline, datasets with 6 million audio-language pairs, and pre-trained models publicly available at https://github.com/JishengBai/AudioSetCaps.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。