arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.03044cs.LGq-bio.NC
收藏

群体变压器:学习神经活动的群体级表征

Population Transformer: Learning Population-level Representations of Neural Activity

California Institute of Technology(加州理工学院);MIT CSAIL, CBMM(麻省理工学院计算机科学与人工智能实验室、认知与脑科学中心)

Geeling Chau, Christopher Wang, Sabera Talukder, Vighnesh Subramaniam, Saraswati Soedarmadji, Yisong Yue, Boris Katz, Andrei Barbu

AI总结:

提出自监督框架PopT,通过聚合稀疏电极通道学习群体级神经表征,降低解码数据需求并提升精度,支持多模态,可解释性强,已开源。

摘要:

我们提出了一种自监督框架,能够以任意规模学习神经记录集合的群体级编码。我们解决了在神经时间序列数据中扩展模型的关键挑战,即跨受试者和数据集的稀疏且可变的电极分布。群体变压器(PopT)堆叠在预训练的时间嵌入之上,通过实现多个空间稀疏数据通道的学习聚合来增强下游解码。预训练的PopT降低了下游解码实验所需的数据量,同时提高了准确性,即使在保留的受试者和任务上也是如此。与端到端方法相比,该方法计算量轻,同时实现相似或更好的解码性能。我们进一步展示了我们的框架如何推广到多种时间序列嵌入和神经数据模态。除了解码之外,我们解释预训练和微调的PopT模型,以展示它们如何用于从大量数据中提取神经科学见解。我们发布了我们的代码以及预训练的PopT,以促进多通道颅内数据解码和可解释性的即用改进。代码可在 https://github.com/czlwang/PopulationTransformer 获取。

英文摘要:

We present a self-supervised framework that learns population-level codes for arbitrary ensembles of neural recordings at scale. We address key challenges in scaling models with neural time-series data, namely, sparse and variable electrode distribution across subjects and datasets. The Population Transformer (PopT) stacks on top of pretrained temporal embeddings and enhances downstream decoding by enabling learned aggregation of multiple spatially-sparse data channels. The pretrained PopT lowers the amount of data required for downstream decoding experiments, while increasing accuracy, even on held-out subjects and tasks. Compared to end-to-end methods, this approach is computationally lightweight, while achieving similar or better decoding performance. We further show how our framework is generalizable to multiple time-series embeddings and neural data modalities. Beyond decoding, we interpret the pretrained and fine-tuned PopT models to show how they can be used to extract neuroscience insights from large amounts of data. We release our code as well as a pretrained PopT to enable off-the-shelf improvements in multi-channel intracranial data decoding and interpretability. Code is available at https://github.com/czlwang/PopulationTransformer.

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

↑