pretrained_models.md

August 27, 2026 ยท View on GitHub

Pre-trained models

If you trained some good models, please, share them. You can post config and model weights in this issue.

Vocal models

Model TypeInstrumentsMetrics (SDR)ConfigCheckpoint
MDX23Cvocals / otherSDR vocals: 10.17ConfigWeights
HTDemucs4 (MVSep finetuned)vocals / otherSDR vocals: 8.78ConfigWeights
Segm Models (VitLarge23)vocals / otherSDR vocals: 9.77ConfigWeights
Swin Upernetvocals / otherSDR vocals: 7.57ConfigWeights
BS Mamba2 (Trained on MUSDB18 only)vocals / otherSDR vocals: 8.82ConfigWeights
BS Roformer (viperx edition)vocals / otherSDR vocals: 10.87ConfigWeights
MelBand Roformer (viperx edition)vocals / otherSDR vocals: 9.67ConfigWeights
MelBand Roformer (KimberleyJensen edition)vocals / otherSDR vocals: 10.98ConfigWeights
BS PolarFormervocals / otherSDR vocals: 11.00ConfigWeights

Note: Metrics measured on Multisong Dataset.

Single stem models

Model TypeInstrumentsMetrics (SDR)ConfigCheckpoint
HTDemucs4 FT DrumsdrumsSDR drums: 11.13ConfigWeights
HTDemucs4 FT BassbassSDR bass: 11.96ConfigWeights
HTDemucs4 FT OtherotherSDR other: 5.85ConfigWeights
HTDemucs4 FT Vocals (Official repository)vocalsSDR vocals: 8.38ConfigWeights
BS Roformer (viperx edition)otherSDR other: 6.85ConfigWeights
MelBand Roformer (aufr33 and viperx edition)crowdSDR crowd: 5.99ConfigWeights
MelBand Roformer (anvuew edition)dereverb---ConfigWeights
MelBand Roformer Denoise (by aufr33)denoise---ConfigWeights
MelBand Roformer Denoise Aggressive (by aufr33)denoise---ConfigWeights
Apollo LQ MP3 restoration (by JusperLee)restored---ConfigWeights
MelBand Roformer Aspiration (by SUC-DriverOld)aspirationSDR: 9.85ConfigWeights
MDX23C Phantom Centre extraction (by wesleyr36)similarityL1Freq: 72.23ConfigWeights
MelBand Roformer Vocals DeReverb/DeEcho (by SUC-DriverOld)drySDR: 10.01ConfigWeights

Note: All HTDemucs4 FT models output 4 stems, but quality is best only on target stem (all other stems are dummy).

Multi-stem models

Model TypeInstrumentsMetrics (SDR)ConfigCheckpoint
BandIt Plusspeech / music / effectsDnR test avg: 11.50 (speech: 15.64, music: 9.18 effects: 9.69)ConfigWeights
HTDemucs4bass / drums / vocals / otherMultisong avg: 9.16 (bass: 11.76, drums: 10.88 vocals: 8.24 other: 5.74)ConfigWeights
HTDemucs4 (6 stems)bass / drums / vocals / other / piano / guitarMultisong (bass: 11.22, drums: 10.22 vocals: 8.05 other: --- piano: --- guitar: ---)ConfigWeights
Demucs3 mmibass / drums / vocals / otherMultisong avg: 8.88 (bass: 11.17, drums: 10.70 vocals: 8.22 other: 5.42)ConfigWeights
DrumSep htdemucs (by inagoy)kick / snare / cymbals / tomsDrumSep test (kick: 10.52, snare: 6.05, toms: 4.68, hh-cymbals: 5.03)ConfigWeights
DrumSep mdx23c (by aufr33 and jarredou)kick / snare / toms / hh / ride / crashDrumSep test (kick: 14.54, snare: 9.79, toms: 10.63, hh: 3.19, cymbals (ride + crash): 6.08)ConfigWeights
DrumSep mdx23c (by jarredou)kick / snare / toms / hh / cymbalsDrumSep test (kick: 16.66, snare: 11.53, toms: 12.33, hh: 4.04, cymbals (ride + crash): 6.36)ConfigWeights

Multi-stem models (MUSDB18HQ)

  • Models in this list were trained only on MUSDB18HQ dataset (100 songs train data). These weights are useful for fine-tuning.
  • Instruments: bass / drums / vocals / other
Model TypeMetrics (SDR)ConfigCheckpoint
MDX23CMUSDB test avg: 7.15 (bass: 5.77, drums: 7.93 vocals: 9.23 other: 5.68)
Multisong avg: 7.02 (bass: 8.40, drums: 7.73 vocals: 7.36 other: 4.57)
ConfigWeights
DTTNetMUSDB test avg: 7.83 (bass: 7.17, drums: 7.66 vocals: 10.31 other: 6.20)
Multisong avg: 7.48 (bass: 8.46, drums: 7.57 vocals: 8.78 other: 5.12)
Configs (vocals bass drums other)Weights (vocals bass drums other)
SCNet Small (by starrytong)MUSDB test avg: 9.03 (bass: 8.89, drums: 10.44 vocals: 9.90 other: 6.89)
Multisong avg: 8.87 (bass: 11.07, drums: 10.79 vocals: 8.27 other: 5.34)
ConfigWeights
SCNet Tran SmallMUSDB test avg: 8.92 (bass: 8.07, drums: 10.81 vocals: 9.97 other: 6.84)
Multisong avg: 8.97 (bass: 10.99, drums: 10.87 vocals: 8.42 other: 5.63)
ConfigWeights
SCNet Masked SmallMUSDB test avg: 8.81 (bass: 8.32, drums: 10.48 vocals: 10.05 other: 6.40)
Multisong avg: 8.86 (bass: 11.00, drums: 10.57 vocals: 8.35 other: 5.51)
ConfigWeights
SCNet LargeMUSDB test avg: 9.32 (bass: 8.63, drums: 10.89 vocals: 10.69 other: 7.06)
Multisong avg: 9.19 (bass: 11.15, drums: 11.04 vocals: 8.94 other: 5.62)
ConfigWeights
SCNet Large (by starrytong)MUSDB test avg: 9.70 (bass: 9.38, drums: 11.15 vocals: 10.94 other: 7.31)
Multisong avg: 9.28 (bass: 11.27, drums: 11.23 vocals: 9.05 other: 5.57)
ConfigWeights
SCNet XLMUSDB test avg: 9.80 (bass: 9.23, drums: 11.51 vocals: 11.05 other: 7.41)
Multisong avg: 9.72 (bass: 11.87, drums: 11.49 vocals: 9.32 other: 6.19)
ConfigWeights
SCNet XL IHFMUSDB test avg: 10.08 (bass: 9.23, drums: 11.81 vocals: 11.42 other: 7.88)
Multisong avg: 9.92 (bass: 11.94, drums: 11.58 vocals: 9.68 other: 6.48)
ConfigWeights
SCNet Masked XL IHFMUSDB test avg: 9.82 (bass: 8.91, drums: 11.62 vocals: 11.06 other: 7.70)
Multisong avg: 9.67 (bass: 11.57, drums: 11.38 vocals: 9.43 other: 6.30)
ConfigWeights
BS RoformerMUSDB test avg: 9.65 (bass: 8.48, drums: 11.61 vocals: 11.08 other: 7.44)
Multisong avg: 9.38 (bass: 11.08, drums: 11.29 vocals: 9.19 other: 5.96)
ConfigWeights
BS Conformer (Medium)MUSDB test avg: 9.18 (bass: 8.11, drums: 10.96 vocals: 10.63 other: 7.03)
Multisong avg: 8.84 (bass: 10.36, drums: 10.73 vocals: 8.75 other: 5.53)
ConfigWeights

MelRoformer models

Table of Mel Band Roformers with different paramers