yipjiaqi/spgm-opt

Demo

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.

Results

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

Model	Data Augmentation	WSJ0-2Mix (SI-SNRi)
spgm (paper)	SpeedPerturb	22.1
spgm-base	DynamicMixing	22.7
spgm-opt	DynamicMixing	23.0

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.

Citation

If you find this model useful, please cite:

@INPROCEEDINGS{yip2023spgm,
  author={Yip, Jia Qi and Zhao, Shengkui and Ma, Yukun and Ni, Chongjia and Zhang, Chong and Wang, Hao and Nguyen, Trung Hieu and Zhou, Kun and Ng, Dianwen and Chng, Eng Siong and others},
  booktitle={ICASSP 2024 - 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)}, 
  title={SPGM: Prioritizing Local Features for enhanced speech separation performance},
  year={2024},
}