home/ modules/ training-and-tuning-source-separation-models

Training and Tuning Source-Separation Models

  • learner can assemble a custom bag-of-models with per-source weights for a target stem set
  • learner can apply remix, BPM/pitch-aligned, and pitch/tempo augmentation to build realistic training data
  • learner can evaluate separation quality against a standard benchmark using SDR and the shift trick

Design and document a training recipe for an improved separation model targeting your genre: specify a custom model bag with per-source weights, an augmentation pipeline (remix + BPM/pitch alignment + on-the-fly pitch/tempo shift), the MUSDB-HQ benchmark protocol, and how the shift trick and SDR will be used to validate gains.

Off-the-shelf Demucs is tuned for MUSDB’s rock/pop centre of gravity. If you live-code UK garage, footwork, or dub techno, the stems you pull for resampling and live remixing inherit that bias: mushy sub-bass splits, hi-hat bleed into “other”. This module builds the whole task of fixing that — designing a complete, documented training recipe that would measurably improve separation for your genre, the kind of recipe you could hand to a GPU box (or a collaborator) and defend every choice in it.

The arc starts supported: reproduce a known configuration by writing a custom bag YAML with per-source weights (the atom on combining checkpoints into custom model bags is your JIT how-to), then run it against MUSDB-HQ tracks and score it, leaning on the shift-trick atom for cheap SDR gains at test time. From there you work up the data side: first plain remix augmentation, then the more musical BPM- and pitch-aligned cross-mixing, then consistent on-the-fly pitch/tempo shifting — each atom a concrete technique you slot into the pipeline. The unsupported capstone asks you to compose all of these into one coherent recipe targeted at your genre, with a validation protocol you specify in advance.

Every required atom gates that recipe: you cannot specify per-source weights without the bag mechanics, cannot justify the augmentation pipeline without the three augmentation concepts, and cannot define validation without MUSDB-HQ, SDR, and the shift trick. The supporting atoms enrich your judgement rather than gate it — knowing why official models are ensembles, and why hybrid waveform/spectrogram branches are complementary, helps you pick which checkpoints belong in your bag and anticipate where your genre will stress each domain.

Atoms in this module

Required — these gate the capstone

Custom separation model bags let you combine multiple checkpoints with per-source weights
Procedure L4 Performance K
Randomly remixing stems from different songs during training forces the model to learn true source priors
Concept L4 Performance K
BPM- and pitch-aligned stem cross-mixing creates more realistic training data than random remixing
Concept L4 Performance K
On-the-fly pitch and tempo shift augmentation improves separation model generalisation across musical keys and tempos
Concept L4 Performance K
MUSDB-HQ is the standard benchmark dataset for music source separation research
Fact L3 Craft K
Averaging predictions over multiple time-shifted inputs (the shift trick) improves separation quality
Concept L3 Craft K
SDR (signal-to-distortion ratio) is the standard metric for evaluating stem separation quality
Concept L3 Craft K

Supporting — enrichment, not gating

Demucs uses a weighted ensemble (BagOfModels) of individually trained checkpoints for best performance
Concept L3 Craft K
Hybrid source separation processes audio in both waveform and spectrogram domains simultaneously
Concept L3 Craft K