用语音模型提升脑电解码准确率,让瘫痪者用思维说话
Teaching Wav2Vec2 the Language of the Brain
- 用脑电数据替代语音模型的输入,让预训练语音模型迁移到脑信号解码
- 最佳方案达到18.54%字符错误率,比从零训练提升20.46个百分点
- 为脑机接口提供高效新方法,适合神经工程与康复医疗研究者
从神经活动解码连续口语有潜力成为瘫痪患者的临床解决方案。深度学习脑机接口(BCI)已成功将尝试发声者的神经活动映射为文本内容,但可用的BCI数据集规模很小。相比之下,语音识别任务拥有大量标注数据和预训练模型,如自监督训练的Wav2Vec2,能生成有意义的语音表示。本研究证明,Wav2Vec2学到的模式可迁移至脑数据。具体地,将音频特征提取器替换为未训练的脑特征提取器(BFE),并对预训练的Wav2Vec2进行全量微调,或从零训练,或冻结预训练权重仅训练BFE,共测试45种BFE架构。最佳结果来自全量微调,字符错误率(CER)达18.54%,比从零训练高出20.46个百分点,比冻结模型训练高15.92个百分点。结果表明,语音识别知识可有效迁移至脑解码,显著提升性能。相关代码见https://github.com/tfiedlerdev/Wav2Vec2ForBrain。
原文摘要 · Abstract (English)
The decoding of continuously spoken speech from neuronal activity has the potential to become an important clinical solution for paralyzed patients. Deep Learning Brain Computer Interfaces (BCIs) have recently successfully mapped neuronal activity to text contents in subjects who attempted to formulate speech. However, only small BCI datasets are available. In contrast, labeled data and pre-trained models for the closely related task of speech recognition from audio are widely available. One such model is Wav2Vec2 which has been trained in a self-supervised fashion to create meaningful representations of speech audio data. In this study, we show that patterns learned by Wav2Vec2 are transferable to brain data. Specifically, we replace its audio feature extractor with an untrained Brain Feature Extractor (BFE) model. We then execute full fine-tuning with pre-trained weights for Wav2Vec2, training ''from scratch'' without pre-trained weights as well as freezing a pre-trained Wav2Vec2 and training only the BFE each for 45 different BFE architectures. Across these experiments, the best run is from full fine-tuning with pre-trained weights, achieving a Character Error Rate (CER) of 18.54\%, outperforming the best training from scratch run by 20.46\% and that of frozen Wav2Vec2 training by 15.92\% percentage points. These results indicate that knowledge transfer from audio speech recognition to brain decoding is possible and significantly improves brain decoding performance for the same architectures. Related source code is available at https://github.com/tfiedlerdev/Wav2Vec2ForBrain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。