arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.08397eess.AScs.SD

Ultra-Lightweight Speech Separation via Group Communication

Columbia University(哥伦比亚大学)

Yi Luo, Cong Han, Nima Mesgarani

英文摘要:

Model size and complexity remain the biggest challenges in the deployment of speech enhancement and separation systems on low-resource devices such as earphones and hearing aids. Although methods such as compression, distillation and quantization can be applied to large models, they often come with a cost on the model performance. In this paper, we provide a simple model design paradigm that explicitly designs ultra-lightweight models without sacrificing the performance. Motivated by the sub-band frequency-LSTM (F-LSTM) architectures, we introduce the group communication (GroupComm), where a feature vector is split into smaller groups and a small processing block is used to perform inter-group communication. Unlike standard F-LSTM models where the sub-band outputs are concatenated, an ultra-small module is applied on all the groups in parallel, which allows a significant decrease on the model size. Experiment results show that comparing with a strong baseline model which is already lightweight, GroupComm can achieve on par performance with 35.6 times fewer parameters and 2.3 times fewer operations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑