Explore the Reinforcement Learning for the LLM based ASR and TTS system
By: Changfeng Gao , Yabin Li , Keyu An and more
Potential Business Impact:
Makes talking computers understand and speak better.
In recent years, large language models (LLMs) have played an important role in automatic speech recognition (ASR) and text-to-speech (TTS) systems. While reinforcement learning (RL) has significantly enhanced LLM performance in text-based tasks, its application to ASR and TTS remains underexplored due to the complexity of training audio-based models. In this study, we propose a lightweight RL framework tailored for audio-based LLMs that can process audio inputs and generate audio outputs. Based on this framework, we evaluate the effectiveness of reinforcement learning on both ASR and TTS tasks. For the ASR task, we experiment with different rule-based reward functions within the Group Relative Policy Optimization (GRPO) framework and investigate the impact of RL data construction. For the TTS task, we compare GRPO with Differentiable Reward Optimization (DiffRO) and further combine the two approaches to achieve improved performance. Our experiments demonstrate that RL can significantly enhance the performance of both ASR and TTS systems, even with limited training data and a small number of optimization steps.
Similar Papers
Reinforcement Learning Outperforms Supervised Fine-Tuning: A Case Study on Audio Question Answering
Sound
Teaches computers to understand and answer questions about sounds.
Customizing Speech Recognition Model with Large Language Model Feedback
Computation and Language
Helps computers understand rare words in speech.
FunAudio-ASR Technical Report
Computation and Language
Makes talking computers understand messy, noisy speech.