[论文解读] A Deep Metric for Multimodal Registration
本文提出一种基于卷积神经网络(CNN)的深度度量方法,用于多模态3D医学图像配准,通过训练网络来分类对齐与非对齐的图像块。该方法在不同数据集上的跨被试T1-T2可变形配准中,显著优于互信息,泛化能力出色,即使训练样本较少也能有效工作,并可通过基于梯度的更新实现高效的连续优化。
Multimodal registration is a challenging problem in medical imaging due the high variability of tissue appearance under different imaging modalities. The crucial component here is the choice of the right similarity measure. We make a step towards a general learning-based solution that can be adapted to specific situations and present a metric based on a convolutional neural network. Our network can be trained from scratch even from a few aligned image pairs. The metric is validated on intersubject deformable registration on a dataset different from the one used for training, demonstrating good generalization. In this task, we outperform mutual information by a significant margin.
研究动机与目标
- 为解决多模态医学图像配准中的挑战,即不同成像模态间组织外观存在显著差异。
- 开发一种通用的、可学习的相似性度量,能够适应多样的模态组合,而无需依赖手工设计的度量方法。
- 仅使用少量对齐图像对即可实现有效配准,这在医学影像中较为常见。
- 在与训练数据不同的数据集上验证该度量,展示其强大的泛化能力。
- 将学习到的度量集成到连续优化框架中,用于可变形配准。
提出的方法
- 该方法将配准问题建模为二分类任务:区分来自不同模态的对齐与非对齐图像块。
- 使用3D卷积神经网络(CNN)对固定图像和变形移动图像的对应图像块输出一个差异度分数。
- 通过在线采样和数据增强,从少量对齐图像体数据中生成正样本(对齐)和负样本(非对齐)的图像块对,从零开始训练网络。
- 将CNN输出的差异度图用作连续优化框架中的相似性度量,通过反向传播计算梯度以更新变换参数。
- 将度量集成到连续优化方案中,采用梯度下降法,利用度量相对于移动图像的导数指导变换参数的更新。
- 在B样条变换的可变形配准设置中评估该方法,结果表明其与连续和离散优化框架均兼容。
实验结果
研究问题
- RQ1基于深度学习的相似性度量能否在多模态配准中泛化到不同的成像协议和被试群体?
- RQ2与传统度量(如互信息)相比,基于CNN的度量在跨被试3D配准中的表现如何?
- RQ3该度量在仅使用少量对齐图像对的情况下,能否实现有效训练?
- RQ4所学习的度量是否能产生平滑且可靠的梯度,适用于可变形配准中的连续优化?
- RQ5该度量能否成功应用于与训练数据不同的数据集,证明其超越训练域的泛化能力?
主要发现
- 所提出的基于CNN的度量在跨被试T1-T2可变形配准中显著优于互信息,Dice和Jaccard评分提升约4个百分点,且具有统计学显著性(p < 0.01)。
- 该方法在与训练数据(IXI)不同的数据集(ALBERTs)上表现良好,证明其对扫描仪和人群特征差异具有鲁棒性。
- 即使仅使用3对对齐图像进行训练,该度量的性能也与使用掩码的互信息相当;当使用6对或11对时,其性能持续优于MI和MI+M。
- 该度量产生平滑且行为良好的梯度,具有正确的符号和局部极小值,支持在连续优化中实现稳定收敛。
- 使用CNN度量的配准速度仅约为互信息的两倍,主要开销来自图像重采样,而非度量计算本身。
- 该方法与离散优化框架兼容,例如基于MRF的公式,此时仅需网络的前向传播即可计算一元势能。
更好的研究,从现在开始
从论文设计到论文写作,大幅缩短您的研究时间。
无需绑定信用卡
本解读由 AI 生成,并经人工编辑审核。