Losses Can Be Blessings: Routing Self-Supervised Speech Representations Towards Efficient Multilingual and Multitask Speech Processing
Yonggan Fu, Yang Zhang, Kaizhi Qian, Zhifan Ye, Zhongzhi Yu, Cheng-I Jeff Lai, Yingyan (Celine) Lin
Abstract
Self-supervised learning (SSL) for rich speech representations has achieved empirical success in low-resource Automatic Speech Recognition (ASR) and other speech processing tasks, which can mitigate the necessity of a large amount of transcribed speech and thus has driven a growing demand for on-device ASR and other speech processing. However, advanced speech SSL models have become increasingly large, which contradicts the limited on-device resources. This gap could be more severe in multilingual/multitask scenarios requiring simultaneously recognizing multiple languages or executing multiple speech processing tasks. Additionally, strongly overparameterized speech SSL models tend to suffer from overfitting when being finetuned on low-resource speech corpus. This work aims to enhance the practical usage of speech SSL models towards a win-win in both enhanced efficiency and alleviated overfitting via our proposed S-Router framework, which for the first time discovers that simply discarding no more than 10% of model weights via only finetuning model connections of speech SSL models can achieve better accuracy over standard weight finetuning on downstream speech processing tasks. More importantly, S-Router can serve as an all-in-one technique to enable (1) a new finetuning scheme, (2) an efficient multilingual/multitask solution, (3) a state-of-the-art ASR pruning technique, and (4) a new tool to quantitatively analyze the learned speech representation. We believe S-Router has provided a new perspective for practical deployment of speech SSL models. Our codes are available at: https://github.com/GATECH-EIC/S3-Router.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c3be758a-bfd8-4470-8309-46d4bf153209Cited by top-tier papers3
- Unveiling and Harnessing Hidden Attention Sinks: Enhancing Large Language Models without Training through Attention CalibrationZhongzhi Yu, Zheng Wang, Yonggan Fu, Huihong Shi et al.ICML 2024 · 63 citations
- Master-ASR: Achieving Multilingual Scalability and Low-Resource Adaptation in ASR with Modular LearningZhongzhi Yu, Yang Zhang, Kaizhi Qian, Cheng Wan et al.ICML 2023 · 17 citations
- Robust Tickets Can Transfer Better: Drawing More Transferable Subnetworks in Transfer LearningYonggan Fu, Ye Yuan, Shang Wu, Jiayi Yuan et al.DAC 2023 · 1 citation
Builds on9
- wav2vec 2.0: A Framework for Self-Supervised Learning of Speech RepresentationsAlexei Baevski, Yuhao Zhou, Abdelrahman Mohamed, Michael AuliNeurIPS 2020 · 9,451 citations
- Gradient Surgery for Multi-Task LearningTianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine et al.NeurIPS 2020 · 2,261 citations
- data2vec: A General Framework for Self-supervised Learning in Speech, Vision and LanguageAlexei Baevski, Wei-Ning Hsu, Qiantong Xu, Arun Babu et al.ICML 2022 · 1,123 citations
- Proving the Lottery Ticket Hypothesis: Pruning is All You NeedEran Malach, Gilad Yehudai, Shai Shalev-Shwartz, Ohad ShamirICML 2020 · 327 citations
- PARP: Prune, Adjust and Re-Prune for Self-Supervised Speech RecognitionCheng-I Jeff Lai, Yang Zhang, Alexander H. Liu, Shiyu Chang et al.NeurIPS 2021 · 91 citations
Related papers
- Don't Be So Sure! Boosting ASR Decoding via Confidence RelaxationTomer Wullach, Shlomo E. ChazanAAAI 2023 · 2 citations
- CA-SSLR: Condition-Aware Self-Supervised Learning Representation for Generalized Speech ProcessingYen-Ju Lu, Jing Liu, Thomas Thebaud, Laureano Moro-Velázquez et al.NeurIPS 2024 · 5 citations
- Multi-resolution HuBERT: Multi-resolution Speech Self-Supervised Learning with Masked Unit PredictionJiatong Shi, Hirofumi Inaguma, Xutai Ma, Ilia Kulikov et al.ICLR 2024 · 39 citations
- Difference in Task Performance on Sparse Speech RepresentationsWenjie Peng, Chen Chen, Thomas HainACL 2026
- Task-Customized Self-Supervised Pre-training with Scalable Dynamic RoutingZhili Liu, Jianhua Han, Lanqing Hong, Hang Xu et al.AAAI 2022 · 30 citations
