深度学习在医学图像配准中的应用

根据模型,策略函数,以及流行度分为7类

深度学习在医学图像配准中的应用

  • Deep similarity-based methods

传统的基于像素的图像度量方法对于图像对具有相似分布的情况是work很好的. 但是图像质量本身不好,比如超声图像,就使得传统的相似度度量在单模态的图像中表现很差. 互信息被用于多模态之间的相似度.  correlation ratio-based MI, contextual conditioned MI, modality independent neighborhood descriptor (MIND) 被用于提高MI性能.

基于深度相似度度量的方法利用CNN代替上述传统的相似度测量方法. 

    • 通过训练CNN, 使用已经经过专家配对的数据作为Ground-truth,学习一个度量网络.输出一个相似分数. 

    • 但是通过配对的数据训练一个相似度度量网络, 是很难的, 因为很难获取大量的数据. 另外, 已经人工配准的质量会影响相似度度量网络精度. 数据增广,  dithering and symmetrizing技术被用于训练, 缓和训练对配准质量的要求, 但是仍然需要粗配准的图像对.

    • 为了彻底消除对配对的数据对的依赖,  stacked autoencoders (SAE), 作为无监督学习技术,可以将图像编码到低维空间计算相似度,这个可以减少计算量. SAE学到的特征表示被用于Demons 和HAMMER进行图像配准.

这类方法最大的问题与传统方法一样, 只不过度量方法变成了神经网络或特征由神经网络提取,配准过程仍需要反复迭代,这是非常耗时的. 随着有关直接变换预测的论文越来越多,这一领域的研究前景不容乐观.

  • Reinforcement learning in medical image registration

    • 结合强化学习与卷积神经网络, 将配准任务分解为一系列的分类问题. 这种策略就是找到一些列的action,如旋转,平移, 从而迭代地进行图像配准.

    • 但是这些搜索空间, 比如旋转,平移,缩放, 只能用于刚性配准, 因为刚性配准可以由这些参数空间表示. 而非刚性的配准更高维,且非线性, 这样会导致搜索空间非常大. 因此就要降维, 使用低维的参数空间去建立统计形变模型( statistical deformation model, SDM). PCA就是在B样条形变向量场( B-spline deformation

    • vector field, DVF)构建SDM.  以位移主元的模态作为未知向量,进行优化。

  • 这类方法的局限在于, 变换模型限于低维度空间, 这导致大多数这类方法使用刚性变换模型. 尽管可以预测SDM模型的低维参数空间, 使之适用于非刚性配准, 但是精确度和灵活性受限. 这类基于RL方法似乎也不再受欢迎.  越来越多的直接变换预测方法被提出, 因此将配准任务分解为一系列的分类问题正在逐渐消失.

  • 监督学习的变换预测

这种直接预测变换的方法有几个挑战:1)输出参数空间维度很高;2) 缺乏带有ground-truth变换的数据集;3) 预测变换的正则. 关于这三的挑战,分别有以下解决方案:1) groound-truth变换生成;2)图像重采样;3)变换正则.

监督学习是指使用ground truth变换对网络进行训练,标签变换是*变形的稠密DVF和刚性变换的参数为6的参数向量

  • 通过随机旋转或平移图像, CNN预测旋转或平移参数, 但是这种参数的方法表示的变换只能适用于刚性配准. artificial affine transformations也被用于合成训练集.这些方法在真实的数据上却不work.

    使用传统方法去配准,将其变换当作ground-truth用于训练神经网络,其基本思想就是随机生成的变换与真实的变换差异太大.

    使用来自少部分数据已知的变换, 生成大量各样的图像对,学习高表示能力的统计外观模型 statistical appearance models (SAM).

    • groound-truth变换生成

      • Random transformation generation

      • traditional registration-generated transformation

      • model-based transformations

    • 监督学习方法

    训练时, 损失计算采用相应的解剖结构标记,而不是稠密的体素级空间对应

    使用两种类型的损失训练.

    监督学习的方法最主要的限制还是在用于训练的变换的生成.因为变换合成的变换可能不能反映真实的生理结构, 或不能抓住真实配准场景中大量的变化情况,或者训练阶段的图像-变换对与真实的图像不同. 尝试不同的变换生成模型,域适应以及多种损失和DVF正则是上述限制的解决方案.

      • Weak supervision

      • Dual supervision

    • 无监督学习的变换预测

无监督学习指的是在没有Ground-truth的情况下进行图像配准, 难点在于设计一个合适的损失函数.

      • 卷积自编码将图像编码为向量, 称之为特征向量, 将其用于相似度测量. STN可以直接插入到卷积网络中, 从而与相似度测量标准一起进行无监督学习.

      • DSC loss可以与相似度损失以及变换平滑损失一起构成目标函数.

      • 配准考虑辅助分割损失, 甚至两个任务是相互补充,可以提高对方的性能.

      • 逆一致性正则项惩罚变换和逆变换的不同. 损失函数由以下损失项组成:  similarity loss, a transformation smoothness loss, an inverse consistent loss and an anti-folding loss

      • cycle consistent loss, identity loss(当浮动图像与固定图像一样, 损失为0)

      • 使用辨别器提供对抗损失,惩罚不真实的变形图像

    无监督学习克服了缺乏标注数据的困难. 由于缺乏ground-truth作为监督, 各种损失函数被提出来. 这些无监督的配准方法有些能超越无监督性能. 但是这些方法都是集中在单模态配准.

训练一个神经网络,当作损失.判别器去辨别一个变换是生成的还是ground-truth, 一张图像是真实的还是由预测的变换形变得到的,对齐是正效果还是负效果.以上是从图像或者变换的外观层面引进损失, 也有人尝试引入生物力学约束, 辨别一个变换是通过有限元分析预测的还是生成的.

多模态配准没有像单模态那样受益于深度学习, 这主要是由于各个模态之间的外观差别太大了. 因此, GAN用于将不同模态的图像映射到单模态.

GAN在医学图像配准是一个有前景的方向, 如新颖的对抗损失, 图像域翻译.

Some auxiliary loss terms include the DSC loss of the anatomical labels or TRE loss of pre-selected landmark pairs

  • GAN在医学图像配准中的应用

    • 为预测的变换提供正则

    • 进行交叉域的图像映射

  • 一些损失函数

    • intensity-based image appearance loss:  SSD, MSE, MAE, MI, MIND, SSIM, CC and its variants.

    • deep similarity-based image appearance loss:  calculates the correlation between the learnt feature-based image descriptors

    • transformation smoothness constraint:  usually involve the calculation of the first and second orders of spatial derivatives of predicted transformation

    • transformation physical fidelity loss :  includes inverse consistency loss, negative Jacobian determinant loss, identity loss, antifolding loss and so on

    • transformation error loss with respect to ground truth transformation:  the error between predicted and ground truth transformations

    • adversarial loss:  the trainable network-based loss