yipjiaqi/

spgm-base

A demo with instructions on how to run inference on the model is available as a colab notebook here

The standalone version of this model for inference with minimal dependencies here

Training is handled by speechbrain. This can be done through my fork of the speechbrain repository found here.

Here are the SI - SNRi results (in dB) on the test set of WSJ0-2 Mix:

In the original paper accepted to ICASSP, the only data augmentation used was speed perturbation. Subsequently we trained the model using dynamic mixing, which yielded improvements in performance.

Additionally, after further exploring some hyperparameters, we obtain an optimized version of SPGM, spgm-opt that achieved 23.0dB SI-SDRi

The weights and config of spgm-base and spgm-opt have been uploaded to huggingface and can be accessed using the code in the spgm_standalone repo.

If you find this model useful, please cite: