arXiv:2608.02053eess.IVcs.CV2026-08

让脑组织微结构估计模型跨不同扫描协议使用,减少重训练

Protocol generalisation for brain tissue microstructure estimation via hypernetwork-controlled geometric deep learning

论文配图:Protocol generalisation for brain tissue microstructure estimation via hypernetwork-controlled geometric deep learning
图 1 · 摘自论文原文
  • 用超网络显式建模b值影响,改进球面卷积网络
  • 在合成与真实数据上均降低误差和偏差,提升对未知b值的适应性
  • 适合临床扩散MRI参数估计算法,减少因扫描参数变化导致的重训练

基于机器学习的脑组织微结构估计比传统拟合方法计算效率更高,但当前模型普遍缺乏跨扩散MRI采集协议的泛化能力,当b向量或b值改变时需重新训练。现有解决协议泛化的方法又缺少旋转等变性。球面卷积神经网络(SCNN)具备旋转等变性和b向量泛化能力,但未考虑b值影响。本文通过超网络将b值显式引入SCNN架构,以NODDI为前向模型进行脑组织微结构估计。在合成数据上训练,并在不同b值组合的合成与真实数据上测试。结果表明,新方法在合成数据上显著降低均方根误差(RMSE)和偏差,在真实数据上与传统NODDI拟合结果一致性更高,表明其对未见b值具有更强鲁棒性,减少了重训练需求。该框架结合了b值、b向量泛化及旋转等变性,提升了深度学习在临床扩散MRI参数估计中的实用性。代码开源:https://github.com/aerdnairo/arXiv_generalisedSCNN。

原文摘要 · Abstract (English)

Brain tissue microstructure estimation with machine learning provides higher computational efficiency than conventional fitting. However, machine learning still presents important limitations that hamper its clinical utility. Specifically, current models typically lack generalisation across diffusion MRI acquisition protocols and require retraining whenever b-vectors or b-values change. Moreover, the recent machine learning methods that were developed to address protocol generalisation lack rotational equivariance. Particularly suitable for dMRI parameter estimation is a geometric deep learning model known as spherical convolutional neural network (SCNN), which guarantees rotational equivariance and b-vector generalisation. However, this architecture currently does not account for b-values. Therefore, obtaining a model that combines protocol generalisation and rotational equivariance remains an open challenge. In this paper, we directly address this issue by incorporating explicit b-value dependence into an SCNN architecture via a hypernetwork. This new approach is illustrated using NODDI as an example forward model for estimating brain tissue microstructure. To evaluate b-value generalisation, the original and newly proposed SCNN architectures are trained on synthetic data and tested on both synthetic and real data across different b-value pairs. Results demonstrate that the proposed method achieves reduced RMSE and bias on synthetic data, as well as higher agreement with conventional NODDI fitting on real data, indicating improved robustness to unseen b-values and a reduced need for retraining. By combining generalisation across b-values with generalisation across b-vectors and rotational equivariance, the proposed framework enhances the applicability of deep learning to clinical diffusion MRI parameter estimation. Code available at https://github.com/aerdnairo/arXiv\_generalisedSCNN.

扩散MRI几何深度学习模型泛化NODDI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。