Stem-JEPA: A Joint-Embedding Predictive Architecture for Musical Stem Compatibility Estimation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Riou, Alain, Lattner, Stefan, Hadjeres, Gaëtan, Anslow, Michael, Peeters, Geoffroy
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909279891488768
author Riou, Alain
Lattner, Stefan
Hadjeres, Gaëtan
Anslow, Michael
Peeters, Geoffroy
author_facet Riou, Alain
Lattner, Stefan
Hadjeres, Gaëtan
Anslow, Michael
Peeters, Geoffroy
contents This paper explores the automated process of determining stem compatibility by identifying audio recordings of single instruments that blend well with a given musical context. To tackle this challenge, we present Stem-JEPA, a novel Joint-Embedding Predictive Architecture (JEPA) trained on a multi-track dataset using a self-supervised learning approach. Our model comprises two networks: an encoder and a predictor, which are jointly trained to predict the embeddings of compatible stems from the embeddings of a given context, typically a mix of several instruments. Training a model in this manner allows its use in estimating stem compatibility - retrieving, aligning, or generating a stem to match a given mix - or for downstream tasks such as genre or key estimation, as the training paradigm requires the model to learn information related to timbre, harmony, and rhythm. We evaluate our model's performance on a retrieval task on the MUSDB18 dataset, testing its ability to find the missing stem from a mix and through a subjective user study. We also show that the learned embeddings capture temporal alignment information and, finally, evaluate the representations learned by our model on several downstream tasks, highlighting that they effectively capture meaningful musical features.
format Preprint
id arxiv_https___arxiv_org_abs_2408_02514
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Stem-JEPA: A Joint-Embedding Predictive Architecture for Musical Stem Compatibility Estimation
Riou, Alain
Lattner, Stefan
Hadjeres, Gaëtan
Anslow, Michael
Peeters, Geoffroy
Sound
Machine Learning
Audio and Speech Processing
This paper explores the automated process of determining stem compatibility by identifying audio recordings of single instruments that blend well with a given musical context. To tackle this challenge, we present Stem-JEPA, a novel Joint-Embedding Predictive Architecture (JEPA) trained on a multi-track dataset using a self-supervised learning approach. Our model comprises two networks: an encoder and a predictor, which are jointly trained to predict the embeddings of compatible stems from the embeddings of a given context, typically a mix of several instruments. Training a model in this manner allows its use in estimating stem compatibility - retrieving, aligning, or generating a stem to match a given mix - or for downstream tasks such as genre or key estimation, as the training paradigm requires the model to learn information related to timbre, harmony, and rhythm. We evaluate our model's performance on a retrieval task on the MUSDB18 dataset, testing its ability to find the missing stem from a mix and through a subjective user study. We also show that the learned embeddings capture temporal alignment information and, finally, evaluate the representations learned by our model on several downstream tasks, highlighting that they effectively capture meaningful musical features.
title Stem-JEPA: A Joint-Embedding Predictive Architecture for Musical Stem Compatibility Estimation
topic Sound
Machine Learning
Audio and Speech Processing
url https://arxiv.org/abs/2408.02514