医用画像登録のコンテキストで、画像ペア間の通常の空間変換を学習するアプローチを提示します。最適化ベースの登録手法や多くの最新の学習ベースの方法とは対照的に、変換の不規則性に直接ペナルティを課すのではなく、逆整合性ペナルティを介して変換の規則性を促進します。ニューラルネットワークを使用して、ソース画像とターゲット画像の間のマップ、およびソース画像とターゲット画像を交換するときのマップを予測します。既存のアプローチとは異なり、これら 2 つの結果のマップを構成し、この構成のヤコビアンの恒等行列からの偏差を正則化します。この正則化プログラム (GradICON) は、マップの構成の逆の一貫性を直接促進する場合と比較して、登録モデルをトレーニングするときに、後者の望ましい暗黙の正則化効果を保持しながら、はるかに優れた収束をもたらします。単一のハイパーパラメーター セットと単一の非データセット固有のトレーニング プロトコルを使用して、さまざまな実世界の医用画像データセットで最先端のレジストレーション パフォーマンスを実現します。
We present an approach to learning regular spatial transformations between image pairs in the context of medical image registration. Contrary to optimization-based registration techniques and many modern learning-based methods, we do not directly penalize transformation irregularities but instead promote transformation regularity via an inverse consistency penalty. We use a neural network to predict a map between a source and a target image as well as the map when swapping the source and target images. Different from existing approaches, we compose these two resulting maps and regularize deviations of the Jacobian of this composition from the identity matrix. This regularizer -- GradICON -- results in much better convergence when training registration models compared to promoting inverse consistency of the composition of maps directly while retaining the desirable implicit regularization effects of the latter. We achieve state-of-the-art registration performance on a variety of real-world medical image datasets using a single set of hyperparameters and a single non-dataset-specific training protocol.