これまで、最適化ベースのレジストレーション モデルでは、空間的に変化する正則化を使用して、さまざまな画像領域での変形の変化を説明していました。ただし、深層学習ベースの登録モデルは、空間的に不変な正則化にほとんど依存していました。ここでは、ニューラル ネットワークを使用して空間的に変化する変形正則化をデータから直接学習するエンド ツー エンドのフレームワークを紹介します。提案された正則化のハイパーパラメーターはネットワークに調整され、正則化の強度を簡単に調整できます。提案された方法は、Transformer ベースのモデルに基づいて構築されていますが、任意のネットワーク アーキテクチャに容易に適応させることができます。公開されているデータセットを使用して提案されたアプローチを徹底的に評価し、スムーズな変形を維持しながら大幅なパフォーマンスの向上を観察しました。この作品のソースコードは、公開後に公開されます。
In the past, optimization-based registration models have used spatially-varying regularization to account for deformation variations in different image regions. However, deep learning-based registration models have mostly relied on spatially-invariant regularization. Here, we introduce an end-to-end framework that uses neural networks to learn a spatially-varying deformation regularizer directly from data. The hyperparameter of the proposed regularizer is conditioned into the network, enabling easy tuning of the regularization strength. The proposed method is built upon a Transformer-based model, but it can be readily adapted to any network architecture. We thoroughly evaluated the proposed approach using publicly available datasets and observed a significant performance improvement while maintaining smooth deformation. The source code of this work will be made available after publication.