Reinforcement Learning Outperforms Supervised Fine-Tuning: A Case Study on Audio Question Answering
By: Gang Li , Jizhong Liu , Heinrich Dinkel and more
Potential Business Impact:
Teaches computers to understand and answer questions about sounds.
Recently, reinforcement learning (RL) has been shown to greatly enhance the reasoning capabilities of large language models (LLMs), and RL-based approaches have been progressively applied to visual multimodal tasks. However, the audio modality has largely been overlooked in these developments. Thus, we conduct a series of RL explorations in audio understanding and reasoning, specifically focusing on the audio question answering (AQA) task. We leverage the group relative policy optimization (GRPO) algorithm to Qwen2-Audio-7B-Instruct, and our experiments demonstrated state-of-the-art performance on the MMAU Test-mini benchmark, achieving an accuracy rate of 64.5%. The main findings in this technical report are as follows: 1) The GRPO algorithm can be effectively applied to large audio language models (LALMs), even when the model has only 8.2B parameters; 2) With only 38k post-training samples, RL significantly outperforms supervised fine-tuning (SFT), indicating that RL-based approaches can be effective without large datasets; 3) The explicit reasoning process has not shown significant benefits for AQA tasks, and how to efficiently utilize deep thinking remains an open question for further research; 4) LALMs still lag far behind humans auditory-language reasoning, suggesting that the RL-based approaches warrant further exploration. Our project is available at https://github.com/xiaomi-research/r1-aqa and https://huggingface.co/mispeech/r1-aqa.
Similar Papers
Explore the Reinforcement Learning for the LLM based ASR and TTS system
Sound
Makes talking computers understand and speak better.
Reinforcement Learning for Reasoning in Small LLMs: What Works and What Doesn't
Machine Learning (CS)
Makes small AI smarter with less money.
SARI: Structured Audio Reasoning via Curriculum-Guided Reinforcement Learning
Computation and Language
Helps computers understand spoken words better.