SWEAR: Section-Wise Efficient Audio Reasoning via Self-Rated Difficulty
Organizations: MoE Key Lab of Artificial Intelligence, X-LANCE Lab, Shanghai Jiao Tong University, China · Shanghai AI Lab, China · KlingAI Research, China
Abstract
Recent work on large audio-language models (LALMs) explores adaptive reasoning to balance accuracy and efficiency. However, existing approaches often rely on coarse-grained difficulty estimation, overlooking the diverse factors that determine reasoning effort. We address this limitation by deriving five-dimensional difficulty scores and compressed reasoning traces from complementary offline assessments. Based on this characterization, we propose SWEAR, a difficulty-aware reasoning framework that learns difficulty signals during supervised fine-tuning and uses reinforcement learning to adapt reasoning allocation across reasoning stages. Experiments on MMAU-test-mini and MMAR show that SWEAR improves accuracy while substantially reducing generated tokens, shortening reasoning for easy instances without sacrificing performance on difficult ones.
Figures & tables
| MMAU-test-mini | MMAR | |||||||||||||||
| Model | Easy | Medium | Hard | Sound | Music | Speech | Overall | Token | Easy | Medium | Hard | Sound | Music | Speech | Overall | Token |
| Qwen2.5-Omni-7B [ 17 ] | 73.87 | 70.96 | 61.26 | 71.47 | 66.47 | 68.17 | 68.7 | 14.34 | 60.18 | 60.36 | 56.16 | 58.18 | 50.00 | 62.93 | 58.9 | 13.18 |
| Public Baselines | ||||||||||||||||
| Audio-Reasoner-7B [ 16 ] | 69.37 | 69.16 | 61.26 | 66.67 | 64.67 | 68.77 | 66.7 | 504.36 | 49.40 | 43.84 | 43.84 | 45.45 | 39.81 | 48.98 | 45.6 | 537.18 |
| Ke-Omni-R-7B [ 22 ] | 79.88 | 75.75 | 64.86 | 78.08 | 70.36 | 72.07 | 73.5 | 76.64 | 58.38 | 56.46 | 49.85 | 55.15 | 42.23 | 58.84 | 55.0 | 70.34 |
| Audio-Flamingo 3 [ 2 ] | 78.68 | 76.65 | 65.17 | 77.48 | 69.46 | 73.27 | 73.4 | 199.90 | 61.68 | 55.26 | 45.95 | 49.70 | 48.06 | 58.16 | 54.5 | 189.10 |
| MMAU | MMAR | ||||
| Setting | Acc. | Token | Acc. | Token | |
| SWEAR | 4 | 73.5 | 197.1 | 62.0 | 195.9 |
| Component ablations | |||||
| w/o rating calibration | 4 | 70.3 | 188.5 | 59.4 | 193.6 |
| w/o length shaping | 4 | 74.2 | 216.6 | 60.4 | 225.5 |
| w/o adaptive KL | 4 | 71.6 | 198.4 | 61.5 | 197.7 |
Efficient Language Model Reasoning
Efficient language model reasoning studies how to improve problem-solving accuracy while reducing inference, memory, latency, and training costs.
See all 373 papers