RemixIT: Continual Self-Training of Speech Enhancement Models via Bootstrapped Remixing

• Published in: IEEE journal of selected topics in signal processing Vol. 16; no. 6; pp. 1329 - 1341
• Main Authors: Tzinis, Efthymios, Adi, Yossi, Ithapu, Vamsi K., Xu, Buye, Smaragdis, Paris, Kumar, Anurag
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.10.2022; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Domains; Empirical analysis; Mixtures; Noise measurement; Recording; Self-supervised learning; semi-supervised self-training; Speech enhancement; Speech processing; Teachers; Training; Waveforms; zero-shot domain adaptation
• ISSN: 1932-4553; 1941-0484
• DOI: 10.1109/JSTSP.2022.3200911

We present RemixIT , a simple yet effective self-supervised method for training speech enhancement without the need of a single isolated in-domain speech nor a noise waveform. Our approach overcomes limitations of previous methods which make them dependent on clean in-domain target signals and thus, sensitive to any domain mismatch between train and test samples. RemixIT is based on a continuous self-training scheme in which a pre-trained teacher model on out-of-domain data infers estimated pseudo-target signals for in-domain mixtures. Then, by permuting the estimated clean and noise signals and remixing them together, we generate a new set of bootstrapped mixtures and corresponding pseudo-targets which are used to train the student network. Vice-versa, the teacher periodically refines its estimates using the updated parameters of the latest student models. Experimental results on multiple speech enhancement datasets and tasks not only show the superiority of our method over prior approaches but also showcase that RemixIT can be combined with any separation model as well as be applied towards any semi-supervised and unsupervised domain adaptation task. Our analysis, paired with empirical evidence, sheds light on the inside functioning of our self-training scheme wherein the student model keeps obtaining better performance while observing severely degraded pseudo-targets.