変形可能な画像レジストレーションは、医療画像分析のさまざまなタスクで重要な役割を果たします。従来のエネルギー最適化またはディープネットワークから派生した登録アルゴリズムを成功させるには、登録エネルギーを適切に設計するか、特定のタイプの医療データ用にネットワークアーキテクチャを注意深く調整するために、コンピューターの専門家による多大な努力が必要です。前述の問題に取り組むために、このペーパーでは、アーキテクチャとそれに対応するトレーニング目標の両方を協調的に最適化する自動学習登録アルゴリズム(AutoReg)を提案します。これにより、コンピュータ以外の専門家、たとえば医療/臨床ユーザーが既製の登録を簡単に見つけることができます。さまざまなシナリオのアルゴリズム。具体的には、自動検索メカニズムと協調的な最適化を使用して、登録ネットワークのアーキテクチャと目的を推測するためのトリプルレベルのフレームワークを確立します。マルチサイトボリュームデータセットとさまざまなレジストレーションタスクで画像レジストレーション実験を実施します。広範な結果は、AutoRegが特定のボリュームに最適なディープレジストレーションネットワークを自動的に学習し、最先端のパフォーマンスを実現し、主流のUNetアーキテクチャよりも計算効率を大幅に向上させる可能性があることを示しています(3D画像ペアの場合は0.558〜0.270秒)同じ構成)。
Deformable image registration plays a critical role in various tasks of medical image analysis. A successful registration algorithm, either derived from conventional energy optimization or deep networks requires tremendous efforts from computer experts to well design registration energy or to carefully tune network architectures for the specific type of medical data. To tackle the aforementioned problems, this paper proposes an automated learning registration algorithm (AutoReg) that cooperatively optimizes both architectures and their corresponding training objectives, enabling non-computer experts, e.g., medical/clinical users, to conveniently find off-the-shelf registration algorithms for diverse scenarios. Specifically, we establish a triple-level framework to deduce registration network architectures and objectives with an auto-searching mechanism and cooperating optimization. We conduct image registration experiments on multi-site volume datasets and various registration tasks. Extensive results demonstrate that our AutoReg may automatically learn an optimal deep registration network for given volumes and achieve state-of-the-art performance, also significantly improving computation efficiency than the mainstream UNet architectures (from 0.558 to 0.270 seconds for a 3D image pair on the same configuration).