小模型也能强推理:Mellow用少数据少参数实现顶尖音频推理能力
Mellow: a small audio language model for reasoning
- 设计轻量级音频语言模型Mellow,专注提升小模型推理能力
- 在MMAU上达52.11分,仅用50倍更少参数和60倍更少训练数据超越大模型
- 自建ReasonAQA数据集,用LLM生成合成问题强化音频理解与推理
多模态音频-语言模型(ALMs)可同时理解与推理音频与文本。通常推理性能随模型规模增长,最佳表现来自超过80亿参数的模型。然而,此前尚无研究探索小模型进行推理的可能性,尽管其在边缘设备上有重要应用前景。为此,我们提出Mellow,一款专为推理设计的小型音频语言模型。Mellow在现有小型音频语言模型中达到最先进水平,并在推理能力上超越多个更大模型。例如,在MMAU评测中得分52.11,接近当前最优的Qwen2 Audio(52.5),但仅使用其50倍更少的参数和60倍更少的训练数据(小时数)。为训练Mellow,我们构建了ReasonAQA数据集,由30%真实数据与70%合成数据组成;合成数据源自音频描述数据集,通过大型语言模型生成聚焦于音频事件、物体、声学场景、信号特性、语义及听者情绪的详细问题与多项选择题。我们在多种任务上评估Mellow的推理能力,涵盖分布内与分布外数据,包括音频理解、演绎推理与比较推理。此外,我们进行了大量消融实验,分析投影层选择、合成数据生成方法及语言模型预训练对推理性能的影响。我们的训练数据、发现与基线为开发具备推理能力的小型ALMs铺平道路。
原文摘要 · Abstract (English)
Multimodal Audio-Language Models (ALMs) can understand and reason over both audio and text. Typically, reasoning performance correlates with model size, with the best results achieved by models exceeding 8 billion parameters. However, no prior work has explored enabling small audio-language models to perform reasoning tasks, despite the potential applications for edge devices. To address this gap, we introduce Mellow, a small Audio-Language Model specifically designed for reasoning. Mellow achieves state-of-the-art performance among existing small audio-language models and surpasses several larger models in reasoning capabilities. For instance, Mellow scores 52.11 on MMAU, comparable to SoTA Qwen2 Audio (which scores 52.5) while using 50 times fewer parameters and being trained on 60 times less data (audio hrs). To train Mellow, we introduce ReasonAQA, a dataset designed to enhance audio-grounded reasoning in models. It consists of a mixture of existing datasets (30% of the data) and synthetically generated data (70%). The synthetic dataset is derived from audio captioning datasets, where Large Language Models (LLMs) generate detailed and multiple-choice questions focusing on audio events, objects, acoustic scenes, signal properties, semantics, and listener emotions. To evaluate Mellow's reasoning ability, we benchmark it on a diverse set of tasks, assessing on both in-distribution and out-of-distribution data, including audio understanding, deductive reasoning, and comparative reasoning. Finally, we conduct extensive ablation studies to explore the impact of projection layer choices, synthetic data generation methods, and language model pretraining on reasoning performance. Our training dataset, findings, and baseline pave the way for developing small ALMs capable of reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。