変形可能なペアワイズ 3D 医療画像レジストレーションのタスクで畳み込みニューラル ネットワークをトレーニングするためのさまざまなカリキュラム学習方法を探ります。私たちの知る限り、カリキュラム学習を使用して医療画像登録モデルをトレーニングすることにより、最初のトレーニング段階での簡単なトレーニング セットアップから始めて、セットアップを徐々に複雑にすることで、パフォーマンスの改善を試みたのは初めてです。一方で、カリキュラムのドロップアウトとスムージングによるカリキュラムという、2 つの既存のカリキュラム学習アプローチを検討します。一方、カリキュラムを達成するための斬新でシンプルな戦略、つまり最初は意図的にぼかした画像を使用し、その後のトレーニング段階で徐々に鮮明な画像に移行することを提案します。最先端の深層学習モデルを基にした私たちの実験は、カリキュラム学習が従来のトレーニングと比較して優れた結果につながることを示しています。さらに、比較したカリキュラム学習アプローチの中で、入力ぼかしによるカリキュラムは、精度と速度のトレードオフが最も優れていることを示しています。
We explore different curriculum learning methods for training convolutional neural networks on the task of deformable pairwise 3D medical image registration. To the best of our knowledge, we are the first to attempt to improve performance by training medical image registration models using curriculum learning, starting from an easy training setup in the first training stages, and gradually increasing the complexity of the setup. On the one hand, we consider two existing curriculum learning approaches, namely curriculum dropout and curriculum by smoothing. On the other hand, we propose a novel and simple strategy to achieve curriculum, namely to use purposely blurred images at the beginning, then gradually transit to sharper images in the later training stages. Our experiments with an underlying state-of-the-art deep learning model show that curriculum learning can lead to superior results compared to conventional training. Additionally, we show that curriculum by input blur has the best accuracy versus speed trade-off among the compared curriculum learning approaches.