arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.08751q-bio.NCcs.LG

用于神经脉冲数据的潜在扩散模型

Latent Diffusion for Neural Spiking Data

University of Tübingen(蒂宾根大学);Tübingen AI Center(蒂宾根人工智能中心);Max Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所)

Jaivardhan Kapoor, Auguste Schulz, Julius Vetter, Felix Pei, Richard Gao, Jakob H. Macke

AI总结:

本文提出LDNS潜在扩散生成模型,结合S4自编码器与条件扩散,实现神经脉冲数据的低维推断及行为条件下的逼真生成。

摘要:

现代神经科学数据集使得人们能够对复杂行为与许多同时记录的神经元活动之间的关系展开前所未有的探究。尽管潜在变量模型能够成功地从此类记录中提取低维嵌入,但利用它们生成逼真的脉冲数据,尤其是以依赖于行为的方式生成,仍然构成挑战。本文提出用于神经脉冲数据的潜在扩散模型(Latent Diffusion for Neural Spiking data,LDNS),这是一种具有低维潜在空间的基于扩散的生成模型:LDNS采用带有结构化状态空间(S4)层的自编码器,将离散的高维脉冲数据投影到连续的、时间对齐的潜在变量中。在这些推断出的潜在变量上,我们训练具有强表达能力的(条件)扩散模型,从而能够采样具有逼真单神经元和群体脉冲统计特征的神经活动。我们在合成数据上验证LDNS,其能够准确恢复潜在结构、放电率和脉冲统计特征。接下来,我们通过生成模拟人类尝试言语期间皮层活动的可变长度数据,证明了该模型的灵活性。我们展示了如何为LDNS配备一个具有强表达能力的观测模型,该模型可解释并非由潜在状态介导的单神经元动态,从而进一步提升生成样本的逼真度。最后,在多种伸手行为期间的运动皮层活动上训练的条件LDNS,能够在给定伸手方向或未见过的伸手轨迹时生成逼真的脉冲数据。总之,LDNS同时实现了低维潜在变量推断和神经脉冲数据集的逼真条件生成,为模拟可通过实验检验的假说开辟了更多可能性。

英文摘要:

Modern datasets in neuroscience enable unprecedented inquiries into the relationship between complex behaviors and the activity of many simultaneously recorded neurons. While latent variable models can successfully extract low-dimensional embeddings from such recordings, using them to generate realistic spiking data, especially in a behavior-dependent manner, still poses a challenge. Here, we present Latent Diffusion for Neural Spiking data (LDNS), a diffusion-based generative model with a low-dimensional latent space: LDNS employs an autoencoder with structured state-space (S4) layers to project discrete high-dimensional spiking data into continuous time-aligned latents. On these inferred latents, we train expressive (conditional) diffusion models, enabling us to sample neural activity with realistic single-neuron and population spiking statistics. We validate LDNS on synthetic data, accurately recovering latent structure, firing rates, and spiking statistics. Next, we demonstrate its flexibility by generating variable-length data that mimics human cortical activity during attempted speech. We show how to equip LDNS with an expressive observation model that accounts for single-neuron dynamics not mediated by the latent state, further increasing the realism of generated samples. Finally, conditional LDNS trained on motor cortical activity during diverse reaching behaviors can generate realistic spiking data given reach direction or unseen reach trajectories. In summary, LDNS simultaneously enables inference of low-dimensional latents and realistic conditional generation of neural spiking datasets, opening up further possibilities for simulating experimentally testable hypotheses.

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

↑