arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.16483eess.AS

FlowSE-GRPO:通过在线强化学习训练流匹配语音增强

FlowSE-GRPO: Training Flow Matching Speech Enhancement via Online Reinforcement Learning

Haoxu Wang, Biao Tian, Yiheng Jiang, Zexu Pan, Shengkui Zhao, Bin Ma, Daren Chen, Xiangang Li

AI总结:

本文提出FlowSE-GRPO,通过在线强化学习优化语音增强,平衡多指标以提升音频质量与任务性能。

摘要:

生成式语音增强提供了一种有前途的替代传统判别方法,通过建模干净语音的分布,条件于噪声输入。通过强化学习(RL)进行后期训练对齐,能够有效将生成模型与人类偏好和下游指标对齐,在自然语言处理等领域中得到了应用,但在语音增强中仍有限,尤其是在线RL。先前工作探索了离线方法,如直接偏好优化(DPO);在线方法如组相对策略优化(GRPO)则鲜有研究。在本文中,我们首次成功地将在线GRPO整合到流匹配语音增强框架中,使在少量更新步骤下能够高效地对齐感知和任务导向的指标。与以往在大语言模型上的GRPO工作不同,我们将其算法适应于语音的连续时间序列性质以及流匹配生成模型的动力学。我们展示了优化单一奖励能够快速提高指标,但往往会导致奖励黑客,尽管评分更高,但会降音频保真度。为缓解这一问题,我们提出了一种多指标奖励优化策略,平衡竞争性目标,显著减少过拟合并提高整体性能。我们的实验验证了在线GRPO在语音增强中的有效性,并为基于RL的生成音频模型后期训练提供了实用指导。

英文摘要:

Generative speech enhancement offers a promising alternative to traditional discriminative methods by modeling the distribution of clean speech conditioned on noisy inputs. Post-training alignment via reinforcement learning (RL) effectively aligns generative models with human preferences and downstream metrics in domains such as natural language processing, but its use in speech enhancement remains limited, especially for online RL. Prior work explores offline methods like Direct Preference Optimization (DPO); online methods such as Group Relative Policy Optimization (GRPO) remain largely uninvestigated. In this paper, we present the first successful integration of online GRPO into a flow-matching speech enhancement framework, enabling efficient post-training alignment to perceptual and task-oriented metrics with few update steps. Unlike prior GRPO work on Large Language Models, we adapt the algorithm to the continuous, time-series nature of speech and to the dynamics of flow-matching generative models. We show that optimizing a single reward yields rapid metric gains but often induces reward hacking that degrades audio fidelity despite higher scores. To mitigate this, we propose a multi-metric reward optimization strategy that balances competing objectives, substantially reducing overfitting and improving overall performance. Our experiments validate online GRPO for speech enhancement and provide practical guidance for RL-based post-training of generative audio models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑