用脉冲神经网络模块化求解核反应堆方程,零遗忘可扩展。
SCNO: Spiking Compositional Neural Operator -- Towards a Neuromorphic Foundation Model for Nuclear PDE Solving
- 模块化设计:用小块脉冲网络分别学对流、扩散等基础算子
- 解决耦合方程误差最低,参数仅9.5万,比传统模型少80%
- 适合需要快速添加新物理的核能仿真,支持脑启发硬件部署
神经算子已成为求解偏微分方程的强大代理模型,但通常为针对单一方程的单体模型,依赖高能耗GPU,且遇新物理需从头训练。本文提出脉冲组合神经算子(SCNO),通过结合脉冲与常规组件,解决上述三类问题。SCNO维护一个小型脉冲神经算子块库,每个块仅训练于单一基本微分算子(对流、扩散、反应),并通过轻量级输入条件聚合器组合,求解训练时未见的耦合方程。一个小修正网络学习跨耦合残差,同时冻结所有块与聚合器,保证零遗忘的模块化扩展。在八个方程族(含五组耦合系统及一组核能相关的1组中子扩散方程)上评估显示,加入修正后的SCNO在五组耦合方程中四组取得最低相对$L^2$误差,相比单体脉冲DeepONet(平均3次种子下提升62%)、标准ANN DeepONet(最高提升65%),而参数仅需9.5万,远低于单体基线的46.2万。据我们所知,这是首个组合式脉冲神经算子,也是首个证明模块化类脑神经算子求解的可行性案例。
原文摘要 · Abstract (English)
Neural operators have emerged as powerful surrogates for partial differential equation (PDE) solvers, yet they are typically trained as monolithic models for individual PDEs, require energy-intensive GPU hardware, and must be retrained from scratch when new physics emerge. We introduce the Spiking Compositional Neural Operator (SCNO), a modular architecture combining spiking and conventional components that addresses all three limitations. SCNO maintains a library of small spiking neural operator blocks, each trained on a single elementary differential operator (convection, diffusion, reaction), and composes them through a lightweight input-conditioned aggregator to solve coupled PDEs not seen during block training. A small correction network learns cross-coupling residuals while keeping all blocks and the aggregator frozen, preserving zero-forgetting modular expansion by construction. We evaluate SCNO on eight PDE families including five coupled systems and a nuclear-relevant 1-group neutron diffusion equation. SCNO with correction achieves the lowest relative $L^2$ error on four of five coupled PDEs, outperforming both a monolithic spiking DeepONet (by up to 62%, mean over 3 seeds) and a standard ANN DeepONet (by up to 65%), while requiring only 95K trainable parameters versus 462K for the monolithic baseline. To our knowledge, this is the first compositional spiking neural operator and the first proof-of-concept for modular neuromorphic PDE solving with built-in forgetting-free expansion.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。