arXiv reaDer
HyperMorph:画像レジストレーションのための償却されたハイパーパラメータ学習
HyperMorph: Amortized Hyperparameter Learning for Image Registration
HyperMorphを紹介します。これは、トレーニング中に重要なレジストレーションハイパーパラメータを調整する必要がない、変形可能な画像レジストレーションの学習ベースの戦略です。従来の登録方法は、最適化問題を解決して2つの画像間の空間的対応のセットを見つけますが、学習ベースの方法は、トレーニングデータセットを利用して、これらの対応を生成する関数を学習します。両方のタイプの手法の結果の品質は、ハイパーパラメータの選択に大きく依存します。残念ながら、ハイパーパラメータの調整には時間がかかり、通常、さまざまなハイパーパラメータ値を使用して多くの個別のモデルをトレーニングする必要があり、最適な結果が得られない可能性があります。この非効率性に対処するために、画像レジストレーションのための償却ハイパーパラメータ学習を導入します。これは、変形フィールドに対するハイパーパラメータの影響を学習するための新しい戦略です。提案されたフレームワークは、入力ハイパーパラメータを取り込んで登録ネットワークを変調し、そのハイパーパラメータ値に最適な変形フィールドを生成するハイパーネットワークを学習します。事実上、この戦略は、テスト時の連続間隔からハイパーパラメータ値の迅速できめ細かい発見を可能にする単一の豊富なモデルをトレーニングします。このアプローチを使用して、既存の検索戦略よりもかなり高速に複数のハイパーパラメータを最適化できることを示します。これにより、計算と人的負担が軽減され、柔軟性が向上します。また、これには、初期化に対する堅牢性の向上や、登録タスク、データセット、さらには単一の解剖学的領域に固有の最適なハイパーパラメータ値を迅速に特定する機能など、いくつかの重要な利点があることも示しています。これらはすべてHyperMorphモデルを再トレーニングする必要はありません。私たちのコードはhttp://voxelmorph.mit.eduで公開されています。
We present HyperMorph, a learning-based strategy for deformable image registration that removes the need to tune important registration hyperparameters during training. Classical registration methods solve an optimization problem to find a set of spatial correspondences between two images, while learning-based methods leverage a training dataset to learn a function that generates these correspondences. The quality of the results for both types of techniques depends greatly on the choice of hyperparameters. Unfortunately, hyperparameter tuning is time-consuming and typically involves training many separate models with various hyperparameter values, potentially leading to suboptimal results. To address this inefficiency, we introduce amortized hyperparameter learning for image registration, a novel strategy to learn the effects of hyperparameters on deformation fields. The proposed framework learns a hypernetwork that takes in an input hyperparameter and modulates a registration network to produce the optimal deformation field for that hyperparameter value. In effect, this strategy trains a single, rich model that enables rapid, fine-grained discovery of hyperparameter values from a continuous interval at test-time. We demonstrate that this approach can be used to optimize multiple hyperparameters considerably faster than existing search strategies, leading to a reduced computational and human burden and increased flexibility. We also show that this has several important benefits, including increased robustness to initialization and the ability to rapidly identify optimal hyperparameter values specific to a registration task, dataset, or even a single anatomical region - all without retraining the HyperMorph model. Our code is publicly available at http://voxelmorph.mit.edu.
updated: Mon Jan 04 2021 15:39:16 GMT+0000 (UTC)
published: Mon Jan 04 2021 15:39:16 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アソシエイト