arXiv reaDer
学習ベースの腹部登録のための二重不確実性ガイド付き空間的および時間的一貫性正則化重み付け
Double-Uncertainty Guided Spatial and Temporal Consistency Regularization Weighting for Learning-based Abdominal Registration
画像レジストレーション問題の不適切な性質に関連する問題に取り組むために、正則化はしばしば解空間を制約するために使用されます。ほとんどの学習ベースの登録アプローチでは、正則化は通常固定の重みを持ち、空間変換のみを制約します。このような規則には2つの制限があります。(i)最適な固定重みの面倒なグリッド検索に加えて、特定の画像ペアの正則化強度を画像のコンテンツに関連付ける必要があるため、「1つの値がすべてに適合する」トレーニングスキームは理想; (ii)変換を空間的に正則化することだけが、不適切な姿勢に関連するいくつかの有益な手がかりを無視する可能性があります。この研究では、教師モデルの予測が学生モデルの予測と一致するように促すことにより、追加の時間的一貫性正則化項を組み込んだ、平均教師ベースの登録フレームワークを提案します。さらに重要なことに、教師は固定の重みを検索する代わりに、変換の不確実性と外観の不確実性を利用して、空間的正則化と時間的一貫性正則化の重みを自動的に調整できます。挑戦的な腹部CT-MRI登録に関する広範な実験は、私たちのトレーニング戦略が、効率的なハイパーパラメータ調整と精度と滑らかさの間のより良いトレードオフの観点から、元の学習ベースの方法を有望に前進させることができることを示しています。
In order to tackle the difficulty associated with the ill-posed nature of the image registration problem, regularization is often used to constrain the solution space. For most learning-based registration approaches, the regularization usually has a fixed weight and only constrains the spatial transformation. Such convention has two limitations: (i) Besides the laborious grid search for the optimal fixed weight, the regularization strength of a specific image pair should be associated with the content of the images, thus the "one value fits all" training scheme is not ideal; (ii) Only spatially regularizing the transformation may neglect some informative clues related to the ill-posedness. In this study, we propose a mean-teacher based registration framework, which incorporates an additional temporal consistency regularization term by encouraging the teacher model's prediction to be consistent with that of the student model. More importantly, instead of searching for a fixed weight, the teacher enables automatically adjusting the weights of the spatial regularization and the temporal consistency regularization by taking advantage of the transformation uncertainty and appearance uncertainty. Extensive experiments on the challenging abdominal CT-MRI registration show that our training strategy can promisingly advance the original learning-based method in terms of efficient hyperparameter tuning and a better tradeoff between accuracy and smoothness.
updated: Wed Mar 02 2022 18:13:23 GMT+0000 (UTC)
published: Tue Jul 06 2021 07:19:49 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アソシエイト