MARS-Sep: Multimodal-Aligned Reinforced Sound Separation
About
Universal sound separation faces a fundamental misalignment: models optimized for low-level signal metrics often produce semantically contaminated outputs, failing to suppress perceptually salient interference from acoustically similar sources. We introduce a preference alignment perspective, analogous to aligning LLMs with human intent. To address this, we introduce MARS-Sep, a reinforcement learning framework that reformulates separation as decision making. Instead of simply regressing ground-truth masks, MARS-Sep learns a factorized Beta mask policy that is steered by a preference reward model and optimized by a stable, clipped trust-region surrogate. The reward, derived from a progressively-aligned audio-text-vision encoder, directly incentivizes semantic consistency with query prompts. Extensive experiments on multiple benchmarks demonstrate consistent gains in Text-, Audio-, and Image-Queried separation, with notable improvements in signal metrics and semantic quality. Our code is available at https://github.com/mars-sep/MARS-Sep. Sound separation samples are available at https://mars-sep.github.io/.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Sound Separation | MUSIC-clean+ | CLAPt6.94 | 18 | |
| Text Query Sound Separation | VGGSOUND clean+ | Mean SDR6.91 | 6 | |
| Image Query Sound Separation | VGGSOUND clean+ | Mean SDR6.93 | 5 | |
| Audio Query Sound Separation | VGGSOUND clean+ | Mean SDR7.93 | 4 | |
| Sound Separation | VGGSOUND clean+ | CLAPt Score9.03 | 3 | |
| Text-Query based Sound Separation | Human User Study Semantic Alignment | Pairwise Preference23.2 | 2 |