当前位置:首页 > 科技资讯 > 正文

NOBLE深度学习框架:加速神经元模拟,挑战传统计算瓶颈

苏黎世联邦理工学院、加州理工学院与阿尔伯塔大学等机构组成的联合团队,提出了一种名为 NOBLE 的深度学习框架。这一框架是首个通过人类大脑皮层实验数据验证性能的规模化深度学习框架,实现了直接从实验数据中学习神经元的非线性动力学行为,其模拟速度比传统数值求解器快出 4200 倍。

人脑通过数百种神经元构成的复杂回路塑造认知功能,这是一个深奥难解的谜题。随着多模态数据的积累,科学家逐步揭示了人类神经元的显著异质性。然而,这些差异如何影响大脑的信息处理过程,如特定基因表达与神经系统疾病之间的内在联系,仍是一个悬而未决的难题。

传统上,研究者常借助基于三维多隔室偏微分方程(PDE)的模型来模拟神经元活动。这类模型虽然能够较好还原生物真实性,但存在计算成本高昂的致命缺陷。单个神经元的模型优化可能需要消耗约 60 万 CPU 核心小时,且参数稍有变动就容易导致模拟结果与实验数据严重偏离。此外,这类确定性模型难以捕捉实验中所观察到的内在变异性。

面对这些挑战,NOBLE 框架应运而生。它是首个通过人类大脑皮层实验数据验证性能的规模化深度学习框架,首次实现了直接从实验数据中学习神经元的非线性动力学行为。其核心突破是构建了一个统一的「神经算子」,能够将神经元特征的连续潜在空间映射为电压响应的集合,而无需为每个模型单独训练替代系统。

在针对小白蛋白阳性(PVALB)神经元数据集的测试中,NOBLE 不仅准确复现了 50 个已知模型及 10 个未见模型的亚阈值与放电动力学行为,其模拟速度更比传统数值求解器快出 4,200 倍。

相关研究成果已入选 NeurIPS 2025。

NOBLE深度学习框架:加速神经元模拟,挑战传统计算瓶颈 NOBLE 深度学习 神经元模拟 计算效率 第1张

  • 论文地址:https://go.hyper.ai/Ramfp

数据集:涵盖 60 个 HoF 模型、250 代进化优化、16 项生理指标

为验证 NOBLE 框架的有效性,研究团队构建了包含小白蛋白阳性(PVALB)神经元的专用数据集。这些模型基于 NEURON 模拟环境构建,采用「活性(all active)」离子通道配置,通过多目标进化优化框架生成,旨在复现实验记录的电生理特征。

具体而言,数据集包含 60 个 HoF 模型,其中 50 个用于训练(分布内模型),10 个作为未见模型用于测试(分布外模型)。每个模型均经过 250 代进化优化,采集不同代次的电压响应。如下图所示,通过空间离散化将电缆方程转化为耦合常微分方程组求解,最终选择能最小化模拟与实验特征平均 z 分数误差的参数组合。

NOBLE深度学习框架:加速神经元模拟,挑战传统计算瓶颈 NOBLE 深度学习 神经元模拟 计算效率 第2张

对神经元进行进化优化过程

数据生成过程采用两阶段优化策略:先拟合被动亚阈值响应,再捕获尖峰阈值以上的活性动态及完整频率-电流曲线。时间序列数据以 0.02ms 时间步长采样 515ms 时长,经 3 倍时间子采样后保留 8583 个时间点。

如下图所示,除了电压变化曲线,数据集还标注了 16 个关键「生理指标」,包括显示形态、实验电压走线、模拟电压走线、尖峰波形和频率电流曲线等,为评估 AI 模型提供了全面标准。

NOBLE深度学习框架:加速神经元模拟,挑战传统计算瓶颈 NOBLE 深度学习 神经元模拟 计算效率 第3张

NOBLE:FNO 驱动+双输入嵌入的神经算子框架

NOBLE 框架的核心创新在于将神经算子与生物信息学中的潜在嵌入技术深度融合,构建了一个从神经元特征到电压响应的端到端映射系统。该框架以傅里叶神经算子(FNO)为底层架构,能够高效处理神经元电生理的时空序列数据。

模型的「翻译」能力源自两套关键输入嵌入设计:神经元特征(neuron features)嵌入与电流注入(current injection)嵌入。

NOBLE深度学习框架:加速神经元模拟,挑战传统计算瓶颈 NOBLE 深度学习 神经元模拟 计算效率 第4张

NOBLE 能正确捕捉多样化的神经元动力学,速度比传统求解器快 4200 倍

为系统评估 NOBLE 框架的综合性能,研究团队围绕五大核心方向设计了多维度实验。实验以小白蛋白阳性(PVALB)神经元的 50 个 HoF 模型作为主要训练数据。

在基础精度(分布内测试)方面,NOBLE 对未参与训练的电流注入信号仍表现出优异的预测能力。

在泛化能力(分布外测试)评估中,NOBLE 在面对未见过的 HoF 模型时仍保持高精度预测。

在计算效率方面,NOBLE 表现出突破性的速度优势。测试结果表明,其单条电压轨迹预测仅需 0.5 毫秒。

在创新生成能力方面,研究团队重点验证了 NOBLE 在已知神经元特征之间「插值创作」新神经元模型的潜力。

  • GitHub 链接:github.com/neuraloperator/noble

神经算子的学界突破与产业落地共振

神经算子与神经元建模的交叉融合正在学术界与产业界激起深层共振。在学术前沿,苏黎世联邦理工学院与卡内基梅隆大学联合提出的几何感知算子变换框架(GAOT),突破了复杂几何域建模的瓶颈。

  • 论文标题:Geometry Aware Operator Transformer as an Efficient and Accurate Neural Surrogate for PDEs on Arbitrary Domains
  • 论文链接:https://arxiv.org/abs/2505.18781

与此同时,麻省理工学院开发的「多细胞整合脑」(miBrain)模型在神经元实体构建方面取得重要进展。

  • 论文标题:Engineered 3D immuno-glial-neurovascular human miBrain model
  • 论文链接:https://doi.org/10.1073/pnas.2511596122

产业界则致力于将学术成果工程化、实用化。英伟达推出的 Modulus 与 PhysicsNeMo 开源框架构建了神经算子落地的核心基础。