MGE-LDM: Joint Latent Diffusion for Simultaneous Music Generation and Source Extraction
This addresses the need for versatile music processing tools for audio engineers and creators, offering a novel approach that integrates multiple tasks into a single model, though it builds on existing latent diffusion methods.
The paper tackled the problem of simultaneous music generation and source extraction by proposing MGE-LDM, a unified latent diffusion framework that enables complete mixture generation, source imputation, and text-conditioned extraction of arbitrary sources, achieving flexible, class-agnostic manipulation without predefined instrument categories.
We present MGE-LDM, a unified latent diffusion framework for simultaneous music generation, source imputation, and query-driven source separation. Unlike prior approaches constrained to fixed instrument classes, MGE-LDM learns a joint distribution over full mixtures, submixtures, and individual stems within a single compact latent diffusion model. At inference, MGE-LDM enables (1) complete mixture generation, (2) partial generation (i.e., source imputation), and (3) text-conditioned extraction of arbitrary sources. By formulating both separation and imputation as conditional inpainting tasks in the latent space, our approach supports flexible, class-agnostic manipulation of arbitrary instrument sources. Notably, MGE-LDM can be trained jointly across heterogeneous multi-track datasets (e.g., Slakh2100, MUSDB18, MoisesDB) without relying on predefined instrument categories. Audio samples are available at our project page: https://yoongi43.github.io/MGELDM_Samples/.