用预训练Transformer实现跨场景的脑神经运动解码,提升泛化能力。
RPNT: Robust Pre-trained Neural Transformer -- A Pathway for Generalized Motor Decoding
- 基于神经元放电数据设计专用Transformer架构,融合多维位置编码与上下文注意力机制。
- 在跨会话、跨任务、跨被试和跨脑区的解码任务中显著优于现有模型。
- 适合脑机接口、神经科学等领域需要强泛化能力的研究者使用。
脑运动解码旨在将神经活动解析并转化为行为。解码模型需在不同脑区、实验会话、行为类型和被试间保持泛化能力,这对实际应用至关重要。现有模型仅部分解决此问题。本文提出预训练神经变压器模型RPNT(Robust Pretrained Neural Transformer),通过预训练实现鲁棒泛化,进而支持下游任务的有效微调。我们系统研究了适用于神经尖峰活动建模的Transformer组件,因文本与图像模型的组件无法直接迁移至神经数据。最终的RPNT架构包含三个核心组件:1)多维旋转位置编码,用于整合实验元数据如脑区坐标、会话编号和行为类型;2)基于卷积核的上下文注意力机制,在全局注意力中学习局部时序结构,以应对神经群体活动的非平稳性;3)具有随机因果掩码策略和对比表示的鲁棒自监督学习目标。我们在两个具有显著泛化挑战的数据集上预训练了两版RPNT:a)多会话、多任务、多被试的微电极基准数据集;b)来自多个皮层区域的高密度Neuropixel 1.0探针多脑区记录数据。预训练后,我们在跨会话、跨类型、跨被试和跨脑区的下游行为解码任务中评估了其泛化性能,结果表明RPNT在各项任务中均持续优于现有解码模型。
原文摘要 · Abstract (English)
Brain motor decoding aims to interpret and translate neural activity into behaviors. Decoding models should generalize across variations, such as recordings from different brain sites, experimental sessions, behavior types, and subjects, will be critical for real-world applications. Current decoding models only partially address these challenges. In this work, we develop a pretrained neural transformer model, RPNT - Robust Pretrained Neural Transformer, designed to achieve robust generalization through pretraining, which in turn enables effective finetuning for downstream motor decoding tasks. We achieved the proposed RPNT architecture by systematically investigating which transformer building blocks could be suitable for neural spike activity modeling, since components from models developed for other modalities, such as text and images, do not transfer directly to neural data. The final RPNT architecture incorporates three unique enabling components: 1) Multidimensional rotary positional embedding to aggregate experimental metadata such as site coordinates, session ids and behavior types; 2) Context-based attention mechanism via convolution kernels operating on global attention to learn local temporal structures for handling non-stationarity of neural population activity; 3) Robust self-supervised learning objective with stochastic causal masking strategies and contrastive representations. We pretrained two versions of RPNT on distinct datasets that present significant generalization challenges: a) Multi-session, multi-task, and multi-subject microelectrode benchmark; b) Multi-site recordings using high-density Neuropixel 1.0 probes from many cortical locations. After pretraining, we evaluated RPNT generalization on cross-session, cross-type, cross-subject, and cross-site downstream behavior decoding tasks. Our RPNT consistently outperforms the existing decoding models on these tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。