arXiv reaDer
Double-Uncertainty Assisted Spatial and Temporal Regularization Weighting for Learning-based Registration
画像レジストレーション問題の不適切な性質に関連する問題に取り組むために、研究者は正則化を使用して解空間を制約します。ほとんどの学習ベースの登録アプローチでは、正則化は通常固定の重みを持ち、空間変換のみを制約します。このような規則には2つの制限があります:(1)特定の画像ペアの正則化強度は画像のコンテンツに関連付ける必要があるため、「1つの値がすべてに適合する」スキームは理想的ではありません。 (2)変換を空間的に正則化する(ただし、異なる推定の時間的一貫性を見落とす)だけでは、不適切な状況に対処するための最良の戦略ではない可能性があります。この研究では、平均教師ベースの登録フレームワークを提案します。このフレームワークには、教師モデルの時間的アンサンブル予測が学生モデルの予測と一致するように促すことにより、追加の時間的正則化項が組み込まれています。また、各トレーニングステップで、摂動された教師モデルから導出された変換の不確実性と外観の不確実性を考慮して、空間正則化と時間正則化の重みを自動的に調整します。マルチモーダルおよびユニモーダルの登録タスクで実験を行い、その結果は、私たちの戦略が従来の学習ベースのベンチマーク手法よりも優れていることを示しています。
In order to tackle the difficulty associated with the ill-posed nature of the image registration problem, researchers use regularization 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: (1) The regularization strength of a specific image pair should be associated with the content of the images, thus the ``one value fits all'' scheme is not ideal; (2) Only spatially regularizing the transformation (but overlooking the temporal consistency of different estimations) may not be the best strategy to cope with the ill-posedness. In this study, we propose a mean-teacher based registration framework. This framework incorporates an additional temporal regularization term by encouraging the teacher model's temporal ensemble prediction to be consistent with that of the student model. At each training step, it also automatically adjusts the weights of the spatial regularization and the temporal regularization by taking account of the transformation uncertainty and appearance uncertainty derived from the perturbed teacher model. We perform experiments on multi- and uni-modal registration tasks, and the results show that our strategy outperforms the traditional and learning-based benchmark methods.
updated: Thu Jul 15 2021 03:17:06 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)アソシエイト