让脑源成像的贝叶斯推理自动学习优化更新规则。
Structure-Preserving Correction Learning for Sparse Bayesian Inference in Brain Source Imaging

- 将经典贝叶斯求解器展开为可训练神经结构,保留原始推导框架。
- 学习到的修正项使重建精度和收敛速度优于基准方法。
- 适合需要可解释性与高精度的脑电/磁源成像研究者使用。
传统的稀疏型II类贝叶斯方法在脑电/磁源成像中可联合估计源与噪声超参数,但依赖固定迭代更新规则。尽管这些规则具有理论基础且可解释,却无法根据数据自适应调整。本文提出通过将经典联合超参数学习求解器展开为可训练神经架构,学习更新机制本身,同时保持底层贝叶斯结构。该框架初始时精确复现经典求解器,随后通过逐步增强的修正学习机制(从可学习偏置到自适应MLP、注意力上下文精修)进行丰富。训练过程不替代贝叶斯推断为黑箱预测器,而是学习结构化修正项,维持原始更新动态的可解释性与模型驱动特性。实验表明,所学修正变体在重建性能与收敛行为上均优于基线展开求解器,同时保持算法透明性。
原文摘要 · Abstract (English)
Classical sparse Type-II Bayesian methods for M/EEG brain imaging support joint estimation of source and noise hyperparameters, but rely on fixed iterative update rules. Although these updates are principled and interpretable, their dynamics cannot be adapted from data. We propose to learn the update mechanism itself while preserving the underlying Bayesian structure by unfolding a classical joint hyperparameter-learning solver into a trainable neural architecture whose layers mirror the original iterations. The resulting framework is initialized to recover the classical solver exactly before training and is enriched through progressively more expressive correction-learning mechanisms, ranging from learnable biases to adaptive MLP and attention-based contextual refinements. In this way, training does not replace Bayesian inference with a black-box predictor, but instead learns structured correction terms while retaining the interpretability and model-based character of the original update dynamics. Structured correction learning therefore aims to improve empirical reconstruction performance without replacing the original model-based inference mechanism. Experimental results show that the learned correction variants improve reconstruction performance and convergence behavior over the baseline unfolded solver while preserving its algorithmic transparency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。