Papers › Transfer Learning with Jukebox for Music Source Separation

Transfer Learning with Jukebox for Music Source Separation

28 Nov 2021arXiv:2111.14200archive 2025-07-28

W. Zai El Amri, O. Tautz, H. Ritter, A. Melnik

In this work, we demonstrate how a publicly available, pre-trained Jukebox model can be adapted for the problem of audio source separation from a single mixed audio channel. Our neural network architecture, which is using transfer learning, is quick to train and the results demonstrate performance comparable to other state-of-the-art approaches that require a lot more compute resources, training data, and time. We provide an open-source code implementation of our architecture (https://github.com/wzaielamri/unmix)

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

wzaielamri/unmix officialmentioned in papermentioned on GitHubpytorchNOASSERTION report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Audio Source SeparationMusic Source SeparationTransfer Learning

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Music Source Separation MUSDB18-HQ Unmix SDR (avg) 4.188 #14 of 14 Archive leaderboard report
Music Source Separation MUSDB18-HQ Unmix SDR (bass) 4.073 #14 of 14 Archive leaderboard report
Music Source Separation MUSDB18-HQ Unmix SDR (drums) 4.925 #14 of 14 Archive leaderboard report
Music Source Separation MUSDB18-HQ Unmix SDR (others) 2.695 #14 of 14 Archive leaderboard report
Music Source Separation MUSDB18-HQ Unmix SDR (vocals) 5.06 #14 of 14 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

ConvolutionDense ConnectionsDilated ConvolutionJukeboxLayer NormalizationPosition-Wise Feed-Forward LayerResidual ConnectionVQ-VAE

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections