Unsupervised Composable Representations for Audio
Giovanni Bindi, Philippe Esling
TL;DR
The paper tackles the challenge of learning compositional representations for audio in an unsupervised setting. It proposes a diffusion-based encoder–decoder framework that decomposes mixtures into latent components in a pre-trained EnCodec latent space, using a shared diffusion model and a fixed composition operator, and introduces a latent-space prior trained via masking to enable recomposition. Empirically, the approach achieves competitive unsupervised latent source separation on Slakh2100 and, in some cases, surpasses supervised baselines such as Demucs on SI-SIR, while also enabling unconditional generation and controllable variation with strong quality and diversity metrics. Overall, the work provides a scalable, lower-cost pathway to unsupervised compositional audio modeling with practical implications for source separation and generative audio tasks.
Abstract
Current generative models are able to generate high-quality artefacts but have been shown to struggle with compositional reasoning, which can be defined as the ability to generate complex structures from simpler elements. In this paper, we focus on the problem of compositional representation learning for music data, specifically targeting the fully-unsupervised setting. We propose a simple and extensible framework that leverages an explicit compositional inductive bias, defined by a flexible auto-encoding objective that can leverage any of the current state-of-art generative models. We demonstrate that our framework, used with diffusion models, naturally addresses the task of unsupervised audio source separation, showing that our model is able to perform high-quality separation. Our findings reveal that our proposal achieves comparable or superior performance with respect to other blind source separation methods and, furthermore, it even surpasses current state-of-art supervised baselines on signal-to-interference ratio metrics. Additionally, by learning an a-posteriori masking diffusion model in the space of composable representations, we achieve a system capable of seamlessly performing unsupervised source separation, unconditional generation, and variation generation. Finally, as our proposal works in the latent space of pre-trained neural audio codecs, it also provides a lower computational cost with respect to other neural baselines.
