Skip to content

Releases: ZFTurbo/Music-Source-Separation-Training

DTTNet (MUSDB18 Weights)

Choose a tag to compare

@ZFTurbo ZFTurbo released this 27 Aug 16:23
7671fae

Paper: https://arxiv.org/abs/2309.08684
Original repository: https://github.com/junyuchen-cjy/DTTNet-Pytorch

Metrics on Multisong Dataset (Num overlap: 4)

Instr vocals sdr: 8.7892 (Std: 3.0910)
Instr vocals bleedless: 22.4991 (Std: 5.7524)
Instr vocals fullness: 14.2901 (Std: 4.2082)
Instr vocals l1_freq: 32.4076 (Std: 10.5796)
Instr bass sdr: 8.4670 (Std: 3.6563)
Instr bass bleedless: 20.1901 (Std: 12.6365)
Instr bass fullness: 18.5676 (Std: 11.7615)
Instr bass l1_freq: 45.7439 (Std: 15.0347)
Instr drums sdr: 7.5790 (Std: 3.2534)
Instr drums bleedless: 19.8468 (Std: 5.8063)
Instr drums fullness: 11.9787 (Std: 4.7591)
Instr drums l1_freq: 28.5485 (Std: 10.9713)
Instr other sdr: 5.1253 (Std: 2.9940)
Instr other bleedless: 16.9402 (Std: 6.3507)
Instr other fullness: 15.0643 (Std: 4.2500)
Instr other l1_freq: 29.9692 (Std: 9.9267)

Metrics on MUSDB18 Dataset (Num overlap: 4)

Instr vocals sdr: 10.3124 (Std: 3.3140)
Instr vocals bleedless: 25.6473 (Std: 13.9282)
Instr vocals fullness: 19.8120 (Std: 6.2593)
Instr vocals l1_freq: 49.1044 (Std: 11.5595)
Instr bass sdr: 7.1779 (Std: 4.3300)
Instr bass bleedless: 23.0092 (Std: 17.2864)
Instr bass fullness: 18.2872 (Std: 12.8031)
Instr bass l1_freq: 55.2532 (Std: 15.6209)
Instr drums sdr: 7.6618 (Std: 2.9982)
Instr drums bleedless: 22.9381 (Std: 7.6466)
Instr drums fullness: 14.4343 (Std: 4.5705)
Instr drums l1_freq: 32.8488 (Std: 10.6579)
Instr other sdr: 6.2039 (Std: 2.5146)
Instr other bleedless: 21.2702 (Std: 7.7841)
Instr other fullness: 18.3491 (Std: 7.1196)
Instr other l1_freq: 35.3127 (Std: 10.8529)

MVSep Mega 53 Stems model

Choose a tag to compare

@ZFTurbo ZFTurbo released this 20 Apr 21:29

First version of model with 53 stems. It's based on BS Roformer architecture.

List of stems: ['accordion', 'acoustic-guitar', 'back-vocal', 'banjo', 'bass', 'bassoon', 'bells', 'bowed_strings', 'brass', 'cello', 'clarinet', 'congas', 'digital-piano', 'dobro', 'double-bass', 'drums', 'electric-guitar', 'flute', 'french-horn', 'glockenspiel', 'guitar', 'harmonica', 'harp', 'harpsichord', 'hh', 'keys', 'kick', 'lead-vocal', 'mandolin', 'marimba', 'oboe', 'organ', 'percussion', 'piano', 'saxophone', 'sitar', 'snare', 'strings', 'synth', 'tambourine', 'timpani', 'toms', 'triangle', 'trombone', 'trumpet', 'tuba', 'ukulele', 'viola', 'violin', 'vocal', 'wind', 'wind-chimes', 'woodwind']

There are some model notes:

  • We are continuing to refine and improve the model.
  • The model is memory-intensive, even with a batch size of 1. I recommend at least 16 GB of VRAM.
  • Currently, the model’s performance on individual stems may be lower than the specialized models available on the MVSep site.
  • Stems do not sum up to the original mixture, as some may contain overlapping information. For example, the "vocals" stem includes both lead and backing vocals.
  • The model returns all stems by default, but filtering out empty stems during post-processing is a logical next step.

BS PolarFormer (Weights)

Choose a tag to compare

@ZFTurbo ZFTurbo released this 22 Mar 13:55
e6279a7

Version of BS Roformer based on PoPE (Polar Coordinate Positional Embeddings).

Arxiv paper: https://arxiv.org/pdf/2509.10534

BS Mamba2 (Weights)

Choose a tag to compare

@ZFTurbo ZFTurbo released this 20 Jan 12:13

Model trained only on MUSDB18 dataset (100 songs)

Multisong dataset metrics:

Num overlap: 2
Instr vocals sdr: 8.8272 (Std: 3.5375)
Instr vocals bleedless: 31.4882 (Std: 10.2950)
Instr vocals fullness: 13.1359 (Std: 4.1619)
Instr vocals l1_freq: 33.3577 (Std: 11.3776)

MUSDB18 Test Dataset metrics:

Num overlap: 2 
Instr vocals sdr: 10.8644 (Std: 3.7046)
Instr vocals bleedless: 39.1770 (Std: 20.2013)
Instr vocals fullness: 17.2136 (Std: 5.9952)
Instr vocals l1_freq: 51.1858 (Std: 12.4056)

BS Conformer (Medium)

Choose a tag to compare

@ZFTurbo ZFTurbo released this 17 Oct 14:45

MSUDB18 Test:

Num overlap: 2
Instr drums sdr: 10.9636 (Std: 3.5740)
Instr drums bleedless: 40.1846 (Std: 12.2989)
Instr drums fullness: 15.3477 (Std: 5.6870)
Instr drums l1_freq: 43.1116 (Std: 11.7196)
Instr bass sdr: 8.1117 (Std: 4.4536)
Instr bass bleedless: 40.1839 (Std: 24.3516)
Instr bass fullness: 13.1798 (Std: 11.4510)
Instr bass l1_freq: 57.7922 (Std: 15.7611)
Instr other sdr: 7.0391 (Std: 2.9666)
Instr other bleedless: 29.7821 (Std: 11.1954)
Instr other fullness: 17.8183 (Std: 6.0078)
Instr other l1_freq: 37.9554 (Std: 12.0237)
Instr vocals sdr: 10.6349 (Std: 3.6795)
Instr vocals bleedless: 38.0876 (Std: 17.5646)
Instr vocals fullness: 16.5019 (Std: 5.7524)
Instr vocals l1_freq: 50.4559 (Std: 11.5570)
Metric avg sdr        : 9.1873
Metric avg bleedless  : 37.0596
Metric avg fullness   : 15.7119
Metric avg l1_freq    : 47.3288

MusltiSong Dataset:

Num overlap: 2
Instr drums sdr: 10.7364 (Std: 4.4866)
Instr drums bleedless: 37.6237 (Std: 11.2495)
Instr drums fullness: 11.4404 (Std: 3.3226)
Instr drums l1_freq: 36.2296 (Std: 11.8242)
Instr bass sdr: 10.3603 (Std: 4.2584)
Instr bass bleedless: 42.4855 (Std: 26.5982)
Instr bass fullness: 13.3310 (Std: 9.9044)
Instr bass l1_freq: 50.5069 (Std: 15.4443)
Instr other sdr: 5.5397 (Std: 3.3859)
Instr other bleedless: 20.9410 (Std: 9.0741)
Instr other fullness: 14.0600 (Std: 4.1029)
Instr other l1_freq: 31.2465 (Std: 10.0779)
Instr vocals sdr: 8.7594 (Std: 3.3987)
Instr vocals bleedless: 33.0732 (Std: 11.9230)
Instr vocals fullness: 13.1176 (Std: 4.2827)
Instr vocals l1_freq: 33.4560 (Std: 10.8730)
Metric avg sdr        : 8.8489
Metric avg bleedless  : 33.5308
Metric avg fullness   : 12.9872
Metric avg l1_freq    : 37.8597

SCNet Masked XL IHF Weights

Choose a tag to compare

@ZFTurbo ZFTurbo released this 08 Jul 19:12

New version of SCNet (scnet_masked) with mask - allows to reduce noise. There are weights for large XL IHF version trained on MUSDB18.

MUSDB18 Test Metrics:

Num overlap: 4
Instr drums sdr: 11.6207 (Std: 3.3777)
Instr drums bleedless: 26.9811 (Std: 9.6603)
Instr drums fullness: 20.2270 (Std: 8.8575)
Instr drums l1_freq: 43.1023 (Std: 11.7691)
Instr bass sdr: 8.9183 (Std: 4.8270)
Instr bass bleedless: 27.1082 (Std: 17.1666)
Instr bass fullness: 16.3376 (Std: 10.3509)
Instr bass l1_freq: 58.6868 (Std: 15.5151)
Instr other sdr: 7.7079 (Std: 3.0485)
Instr other bleedless: 24.1793 (Std: 9.8236)
Instr other fullness: 20.2322 (Std: 6.8875)
Instr other l1_freq: 38.1973 (Std: 12.5904)
Instr vocals sdr: 11.0675 (Std: 3.7956)
Instr vocals bleedless: 26.4048 (Std: 11.8723)
Instr vocals fullness: 20.4435 (Std: 7.2946)
Instr vocals l1_freq: 50.6262 (Std: 11.3240)
Metric avg sdr        : 9.8286
Metric avg bleedless  : 26.1683
Metric avg fullness   : 19.3101
Metric avg l1_freq    : 47.6531

Multisong dataset Metrics:

Num overlap: 4
Instr drums sdr: 11.3847 (Std: 5.0454)
Instr drums bleedless: 23.6516 (Std: 8.3218)
Instr drums fullness: 15.8085 (Std: 5.4690)
Instr drums l1_freq: 36.6667 (Std: 12.2488)
Instr bass sdr: 11.5754 (Std: 4.9476)
Instr bass bleedless: 23.8556 (Std: 15.0766)
Instr bass fullness: 21.7412 (Std: 14.2511)
Instr bass l1_freq: 52.6289 (Std: 16.7418)
Instr other sdr: 6.3082 (Std: 3.4198)
Instr other bleedless: 18.6385 (Std: 7.7289)
Instr other fullness: 16.5832 (Std: 4.5989)
Instr other l1_freq: 32.4004 (Std: 10.3759)
Instr vocals sdr: 9.4303 (Std: 3.2363)
Instr vocals bleedless: 24.0554 (Std: 7.9791)
Instr vocals fullness: 16.1403 (Std: 4.4971)
Instr vocals l1_freq: 34.2907 (Std: 10.5988)
Metric avg sdr        : 9.6747
Metric avg bleedless  : 22.5503
Metric avg fullness   : 17.5683
Metric avg l1_freq    : 38.9967

SCNet Masked Small Weights

Choose a tag to compare

@ZFTurbo ZFTurbo released this 20 Jun 07:16

New version of SCNet (scnet_masked) with mask - allows to reduce noise. There are weights for Small version trained on MUSDB18.

MUSDB18 Test Metrics:

Num overlap: 4
Instr drums sdr: 10.4807 (Std: 3.1034)
Instr drums bleedless: 23.5738 (Std: 9.6661)
Instr drums fullness: 19.3672 (Std: 8.8171)
Instr drums l1_freq: 39.7583 (Std: 11.5116)
Instr bass sdr: 8.3271 (Std: 4.5202)
Instr bass bleedless: 28.8302 (Std: 17.6783)
Instr bass fullness: 15.2740 (Std: 10.2804)
Instr bass l1_freq: 57.3428 (Std: 14.7919)
Instr other sdr: 6.4003 (Std: 3.5504)
Instr other bleedless: 21.1916 (Std: 8.4212)
Instr other fullness: 18.6111 (Std: 6.4006)
Instr other l1_freq: 34.6862 (Std: 10.9688)
Instr vocals sdr: 10.0517 (Std: 3.3388)
Instr vocals bleedless: 26.1827 (Std: 13.2979)
Instr vocals fullness: 17.8171 (Std: 6.8853)
Instr vocals l1_freq: 47.6337 (Std: 11.5299)
Metric avg sdr        : 8.8149
Metric avg bleedless  : 24.9446
Metric avg fullness   : 17.7673
Metric avg l1_freq    : 44.8552

Multisong dataset Metrics:

Num overlap: 4
Instr drums sdr: 10.5715 (Std: 4.7403)
Instr drums bleedless: 21.5980 (Std: 7.4977)
Instr drums fullness: 15.0217 (Std: 4.9323)
Instr drums l1_freq: 34.3450 (Std: 11.8224)
Instr bass sdr: 11.0098 (Std: 4.6274)
Instr bass bleedless: 24.4461 (Std: 16.1432)
Instr bass fullness: 21.4828 (Std: 15.0182)
Instr bass l1_freq: 51.1773 (Std: 16.3137)
Instr other sdr: 5.5176 (Std: 3.0846)
Instr other bleedless: 16.5156 (Std: 5.8950)
Instr other fullness: 16.1805 (Std: 4.3133)
Instr other l1_freq: 30.2492 (Std: 9.6913)
Instr vocals sdr: 8.3510 (Std: 3.0380)
Instr vocals bleedless: 21.5931 (Std: 7.6635)
Instr vocals fullness: 14.0202 (Std: 4.2627)
Instr vocals l1_freq: 31.0896 (Std: 10.2234)
Metric avg sdr        : 8.8624
Metric avg bleedless  : 21.0382
Metric avg fullness   : 16.6763
Metric avg l1_freq    : 36.7153

SCNet XL IHF Weights

Choose a tag to compare

@ZFTurbo ZFTurbo released this 18 Jun 09:25

Weights were obrained with training only on MUSDB18 training data. Main changes in config is related to high frequencies, so model should give better results in this part of spectrogram.

Results on MUSDB18 test data:

Instr drums sdr: 11.8155 (Std: 3.2582)
Instr drums bleedless: 28.3940 (Std: 11.3584)
Instr drums fullness: 20.3435 (Std: 7.6190)
Instr drums l1_freq: 43.2681 (Std: 11.5372)
Instr bass sdr: 9.2360 (Std: 5.0752)
Instr bass bleedless: 24.9027 (Std: 13.8338)
Instr bass fullness: 17.0699 (Std: 11.1160)
Instr bass l1_freq: 58.6811 (Std: 15.1790)
Instr other sdr: 7.8828 (Std: 3.1603)
Instr other bleedless: 24.7929 (Std: 9.5862)
Instr other fullness: 19.6569 (Std: 6.9291)
Instr other l1_freq: 38.2785 (Std: 12.4884)
Instr vocals sdr: 11.4219 (Std: 3.9988)
Instr vocals bleedless: 23.4196 (Std: 9.7727)
Instr vocals fullness: 20.3662 (Std: 7.8029)
Instr vocals l1_freq: 50.3840 (Std: 11.0768)
Metric avg sdr        : 10.0891
Metric avg bleedless  : 25.3773
Metric avg fullness   : 19.3591
Metric avg l1_freq    : 47.6529

Results on Multisong dataset:

Instr drums sdr: 11.5838 (Std: 5.3371)
Instr drums bleedless: 24.4482 (Std: 8.1998)
Instr drums fullness: 16.0051 (Std: 4.1486)
Instr drums l1_freq: 37.1487 (Std: 12.4679)
Instr bass sdr: 11.9436 (Std: 5.2532)
Instr bass bleedless: 23.0196 (Std: 11.7992)
Instr bass fullness: 20.1038 (Std: 13.4688)
Instr bass l1_freq: 52.8341 (Std: 16.9122)
Instr other sdr: 6.4873 (Std: 3.6878)
Instr other bleedless: 19.0051 (Std: 8.0195)
Instr other fullness: 16.3094 (Std: 5.0124)
Instr other l1_freq: 32.6761 (Std: 10.9044)
Instr vocals sdr: 9.6896 (Std: 3.5071)
Instr vocals bleedless: 22.7375 (Std: 7.0052)
Instr vocals fullness: 15.9332 (Std: 4.2005)
Instr vocals l1_freq: 34.3502 (Std: 10.8328)
Metric avg sdr        : 9.9261
Metric avg bleedless  : 22.3026
Metric avg fullness   : 17.0878
Metric avg l1_freq    : 39.2523

SCNet Tran Weights

Choose a tag to compare

@ZFTurbo ZFTurbo released this 02 Apr 07:15

Pretrained weights for new SCNet Tran model. It's model where LSTM blocks replaced with transformer blocks. Weights were obtained on MUSDB18 dataset only.

Multisong dataset metrics

Num overlap: 2
Instr drums sdr: 10.8707 (Std: 4.9156)
Instr drums bleedless: 21.1680 (Std: 6.7914)
Instr drums fullness: 14.7370 (Std: 4.3909)
Instr drums l1_freq: 34.3470 (Std: 11.9350)
Instr bass sdr: 10.9921 (Std: 4.7437)
Instr bass bleedless: 19.7168 (Std: 9.4653)
Instr bass fullness: 18.9573 (Std: 13.3901)
Instr bass l1_freq: 50.0550 (Std: 15.6142)
Instr other sdr: 5.6386 (Std: 3.1793)
Instr other bleedless: 15.4912 (Std: 5.5533)
Instr other fullness: 16.9250 (Std: 4.4698)
Instr other l1_freq: 30.2603 (Std: 9.8843)
Instr vocals sdr: 8.4272 (Std: 3.2268)
Instr vocals bleedless: 18.7054 (Std: 5.8792)
Instr vocals fullness: 15.2973 (Std: 4.1997)
Instr vocals l1_freq: 31.0975 (Std: 10.2741)
Metric avg sdr        : 8.9821
Metric avg bleedless  : 18.7704
Metric avg fullness   : 16.4791
Metric avg l1_freq    : 36.4400

MUSDB18 dataset metrics

Num overlap: 2
Instr drums sdr: 10.8139 (Std: 2.9125)
Instr drums l1_freq: 39.1727 (Std: 10.2864)
Instr drums fullness: 17.6692 (Std: 6.2694)
Instr drums bleedless: 25.0680 (Std: 10.1685)
Instr bass sdr: 8.0731 (Std: 4.6414)
Instr bass l1_freq: 55.2888 (Std: 13.7685)
Instr bass fullness: 14.9235 (Std: 9.6630)
Instr bass bleedless: 20.8816 (Std: 10.5097)
Instr other sdr: 6.8419 (Std: 2.8909)
Instr other l1_freq: 35.2425 (Std: 11.2693)
Instr other fullness: 19.2561 (Std: 6.6855)
Instr other bleedless: 20.1550 (Std: 7.3377)
Instr vocals sdr: 9.9799 (Std: 3.5204)
Instr vocals l1_freq: 45.7778 (Std: 11.0381)
Instr vocals fullness: 20.0345 (Std: 7.0781)
Instr vocals bleedless: 17.8214 (Std: 7.0943)
Metric avg sdr        : 8.9272
Metric avg l1_freq    : 43.8704
Metric avg fullness   : 17.9708
Metric avg bleedless  : 20.9815

SCNet XL

Choose a tag to compare

@ZFTurbo ZFTurbo released this 12 Dec 22:56
7d4373e

Pre-trained weights on MUSDB18HQ. Metrics table