We examine the use of the Conformer architecture for continuous speech separation. Conformer allows the separation model to efficiently capture both local and global context information, which is helpful for speech separation. Experimental results using the LibriCSS dataset show that the Conformer separation model achieves state of the art results for both single-channel and multi-channel settings.
For a detailed description and experimental results, please refer to our paper: Continuous Speech Separation with Conformer (Accepted by ICASSP 2021).
python 3.10.9, torch 2.0.1+cu118
-
Download the overlapped speech of LibriCSS dataset.
- from Google drive:
wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1PdloA-V8HGxkRu9MnT35_civpc3YXJsT' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\n/p')&id=1PdloA-V8HGxkRu9MnT35_civpc3YXJsT" -O overlapped_speech.zip && rm -rf /tmp/cookies.txt && unzip overlapped_speech.zip && rm overlapped_speech.zip
- from Microsoft Azure Storage:
wget "https://valle.blob.core.windows.net/share/CSS_with_Conformer/overlapped_speech.zip?sv=2020-08-04&st=2023-03-01T07%3A51%3A05Z&se=2033-03-02T07%3A51%3A00Z&sr=c&sp=rl&sig=QJXmSJG9DbMKf48UDIU1MfzIro8HQOf3sqlNXiflY1I%3D" -O overlapped_speech.zip && rm -rf /tmp/cookies.txt && unzip overlapped_speech.zip && rm overlapped_speech.zip
-
Download the Conformer separation models.
- from Microsoft Azure Storage:
wget "https://valle.blob.core.windows.net/share/CSS_with_Conformer/checkpoints.zip?sv=2020-08-04&st=2023-03-01T07%3A51%3A05Z&se=2033-03-02T07%3A51%3A00Z&sr=c&sp=rl&sig=QJXmSJG9DbMKf48UDIU1MfzIro8HQOf3sqlNXiflY1I%3D" -O checkpoints.zip && rm -rf /tmp/cookies.txt && unzip checkpoints.zip && rm checkpoints.zip
-
Run the separation.
3.1 Single-channel separation
export MODEL_NAME=1ch_conformer_base python3 separate.py \ --checkpoint checkpoints/$MODEL_NAME \ --mix-scp utils/overlapped_speech_1ch.scp \ --dump-dir separated_speech/monaural/utterances_with_$MODEL_NAME \ --device-id 0 \ --num_spks 2
The separated speech can be found in the directory 'separated_speech/monaural/utterances_with_$MODEL_NAME'
3.2 Seven-channel separation
export MODEL_NAME=conformer_base python3 separate.py \ --checkpoint checkpoints/$MODEL_NAME \ --mix-scp utils/overlapped_speech_7ch.scp \ --dump-dir separated_speech/7ch/utterances_with_$MODEL_NAME \ --device-id 0 \ --num_spks 2 \ --mvdr True
The separated speech can be found in the directory 'separated_speech/7ch/utterances_with_$MODEL_NAME'
If you find our work useful, please cite our paper:
@inproceedings{CSS_with_Conformer,
title={Continuous speech separation with conformer},
author={Chen, Sanyuan and Wu, Yu and Chen, Zhuo and Wu, Jian and Li, Jinyu and Yoshioka, Takuya and Wang, Chengyi and Liu, Shujie and Zhou, Ming},
booktitle={ICASSP 2021-2021 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)},
pages={5749--5753},
year={2021},
organization={IEEE}
}