arXiv:2409.00101eess.SPcs.HC2024-09ICLR被引 116

用大模型理解脑电波,一模多用解决神经信号处理难题

NeuroLM: A Universal Multi-task Foundation Model for Bridging the Gap between Language and EEG Signals

  • 把脑电信号当外语,通过量化编码成离散符号输入大模型
  • 1.7B参数模型在2.5万小时数据上预训练,六项任务表现优异
  • 支持多任务指令微调,适合脑机接口与医疗研究者使用

近期基于脑电图(EEG)等神经信号的大规模预训练取得了显著进展,推动了脑机接口与医疗技术的发展。然而,现有模型通常需对每个下游任务进行完整微调,限制了通用性并造成资源浪费。为此,我们提出NeuroLM,首个将语言模型能力融入神经信号处理的多任务基础模型。该方法通过向量量化时频预测学习文本对齐的神经分词器,将EEG信号编码为离散神经标记;这些由冻结的向量量化编码器生成的标记被输入语言模型,通过多通道自回归学习因果脑电信息,使模型同时理解语言与脑电信号。最终,多任务指令微调使NeuroLM适应各类下游任务。我们首次证明,通过特定结合语言模型,可实现单一模型统一处理多种EEG任务。最大版本NeuroLM-XL拥有17亿参数,是目前最大的脑电处理模型,基于约2.5万小时的大型数据集预训练。在六个不同下游数据集上的评估显示其巨大潜力。

原文摘要 · Abstract (English)

Recent advancements for large-scale pre-training with neural signals such as electroencephalogram (EEG) have shown promising results, significantly boosting the development of brain-computer interfaces (BCIs) and healthcare. However, these pre-trained models often require full fine-tuning on each downstream task to achieve substantial improvements, limiting their versatility and usability, and leading to considerable resource wastage. To tackle these challenges, we propose NeuroLM, the first multi-task foundation model that leverages the capabilities of Large Language Models (LLMs) by regarding EEG signals as a foreign language, endowing the model with multi-task learning and inference capabilities. Our approach begins with learning a text-aligned neural tokenizer through vector-quantized temporal-frequency prediction, which encodes EEG signals into discrete neural tokens. These EEG tokens, generated by the frozen vector-quantized (VQ) encoder, are then fed into an LLM that learns causal EEG information via multi-channel autoregression. Consequently, NeuroLM can understand both EEG and language modalities. Finally, multi-task instruction tuning adapts NeuroLM to various downstream tasks. We are the first to demonstrate that, by specific incorporation with LLMs, NeuroLM unifies diverse EEG tasks within a single model through instruction tuning. The largest variant NeuroLM-XL has record-breaking 1.7B parameters for EEG signal processing, and is pre-trained on a large-scale corpus comprising approximately 25,000-hour EEG data. When evaluated on six diverse downstream datasets, NeuroLM showcases the huge potential of this multi-task learning paradigm.

脑机接口多模态大模型神经信号

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。