arXiv reaDer
最適化から変形可能な画像レジストレーションを学習する:展望、モジュール、バイレベルトレーニングおよびそれ以降
Learning Deformable Image Registration from Optimization: Perspective, Modules, Bilevel Training and Beyond
従来の変形可能なレジストレーション方法は、画像ペアに対して注意深く設計された最適化モデルを解くことを目的としており、それらの計算コストは非常に高くなります。対照的に、最近の深層学習ベースのアプローチは、高速な変形推定を提供できます。これらのヒューリスティックネットワークアーキテクチャは完全にデータ駆動型であるため、トポロジの保存など、もっともらしい変形を生成するために不可欠な明示的な幾何学的制約がありません。利点を統合し、これら2つのカテゴリのアプローチの制限を回避するために、マルチスケール伝搬を介して微分同相モデルを最適化する新しい深層学習ベースのフレームワークを設計します。具体的には、微分同相写像の登録を定式化するための一般的な最適化モデルを導入し、一連の学習可能なアーキテクチャを開発して、粗い特徴空間から細かい特徴空間への伝播更新を取得します。さらに、タスク固有のハイパーパラメータの効率的な検索を可能にする、新しい2レベルの自己調整トレーニング戦略を提案します。このトレーニング戦略により、さまざまなタイプのデータに対する柔軟性が向上すると同時に、計算と人的負担が軽減されます。脳MRIデータの画像からアトラスへのレジストレーションと肝臓CTデータの画像から画像へのレジストレーションを含む3Dボリュームデータセットの画像レジストレーション実験の2つのグループを実施します。広範な結果は、微分同相写像の保証と極端な効率を備えた提案された方法の最先端の性能を示しています。また、私たちのフレームワークを挑戦的なマルチモーダル画像レジストレーションに適用し、マルチモーダル融合や画像セグメンテーションを含む医用画像分析のダウンストリームタスクをサポートするための登録方法を調査します。
Conventional deformable registration methods aim at solving an optimization model carefully designed on image pairs and their computational costs are exceptionally high. In contrast, recent deep learning based approaches can provide fast deformation estimation. These heuristic network architectures are fully data-driven and thus lack explicit geometric constraints, e.g., topology-preserving, which are indispensable to generate plausible deformations. We design a new deep learning based framework to optimize a diffeomorphic model via multi-scale propagation in order to integrate advantages and avoid limitations of these two categories of approaches. Specifically, we introduce a generic optimization model to formulate diffeomorphic registration and develop a series of learnable architectures to obtain propagative updating in the coarse-to-fine feature space. Moreover, we propose a novel bilevel self-tuned training strategy, allowing efficient search of task-specific hyper-parameters. This training strategy increases the flexibility to various types of data while reduces computational and human burdens. We conduct two groups of image registration experiments on 3D volume datasets including image-to-atlas registration on brain MRI data and image-to-image registration on liver CT data. Extensive results demonstrate the state-of-the-art performance of the proposed method with diffeomorphic guarantee and extreme efficiency. We also apply our framework to challenging multi-modal image registration, and investigate how our registration to support the down-streaming tasks for medical image analysis including multi-modal fusion and image segmentation.
updated: Thu Mar 25 2021 12:41:40 GMT+0000 (UTC)
published: Thu Apr 30 2020 03:23:45 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト