用强化学习提升音频描述的时间精度和细节丰富度
AudioMap: Cloze-and-Choice Reinforcement Learning for Time-Aware Dense Audio Captioning

- 设计闭环选择奖励机制,细粒度优化多事件音频描述
- 在多个基准上达到开源模型最优,媲美闭源模型表现
- 构建首个细粒度时间标注音频数据集,含4.4万条标注
时间感知密集音频描述(TDAC)旨在生成具有精确时间边界和丰富属性的多个细粒度音频描述。现有方法依赖监督微调,难以兼顾精度与丰富性。强化学习虽具潜力,但面临两大挑战:一是现有奖励过粗,无法有效指导多事件、多属性、多关系的细粒度描述;二是自由形式描述中事件-时间对应难,缺乏可靠时序监督。为此,本文提出AudioMap,一种基于强化学习的新型TDAC框架,采用统一的填空与选择奖励范式。引入证据充分性奖励(ESR),通过非对称层次评分机制提升不同声学维度的准确性与描述丰富性;设计事件条件时间奖励(ECTR),利用时间交并比(temporal IoU)结构化绑定时间戳与事件语义,并结合双课程学习策略促进训练。此外,构建首个时间感知细粒度音频描述数据集AudioMapCap-44K,包含44,000条精心标注的音频描述。在多个基准上的实验表明,AudioMap在开源模型中达到当前最佳性能,且在多数场景下优于或媲美闭源模型。
原文摘要 · Abstract (English)
Time-aware dense audio captioning (TDAC) aims to generate multiple fine-grained attributes (dense) of the audio with precise time boundaries (time-aware). Existing methods struggle to achieve these two goals and mainly rely on supervised fine-tuning, yielding sub-optimal performance. While reinforcement learning (RL) shows promise, applying it to TDAC faces two main challenges: (1) existing rewards are too coarse to supervise multi-event, multi-attribute, and multi-relation descriptions in a fine-grained manner; and (2) temporal supervision is difficult for free-form captions, where flexible event-time expressions make reliable event-time correspondence challenging. To address these challenges, we propose AudioMap, a novel RL-based TDAC framework, which shifts to a unified cloze-and-choice reward paradigm. Specifically, we introduce the Evidence Sufficiency Reward (ESR) with an asymmetric hierarchical scoring mechanism to promote fine-grained accuracy and descriptive richness across diverse acoustic dimensions. Furthermore, we design the Event-Conditioned Temporal Reward (ECTR) to structurally bind timestamps to event semantics via temporal IoU, accompanied by a dual-curriculum learning strategy to facilitate the training process. Finally, to support this task, we construct the first time-aware fine-grained audio captioning dataset, AudioMapCap-44K, which contains 44K carefully annotated captions. Extensive experiments across diverse benchmarks show that AudioMap achieves state-of-the-art (SOTA) performance among open-source models and delivers competitive or superior results relative to proprietary models. Project page and release updates are available at https://github.com/ryysayhi/AudioMap.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。