arXiv reaDer
よりスマートにトレーニング: 希少なデータで深部腹部 CT レジストレーションを学習
Train smarter, not harder: learning deep abdominal CT registration on scarce data
目的: この研究は、腹部イメージングのための畳み込みニューラル ネットワーク ベースの画像から画像へのレジストレーションを改善するためのトレーニング戦略を調査することを目的としています。方法: さまざまなトレーニング戦略、損失関数、および転移学習スキームが考慮されました。さらに、動的損失重み付けを可能にする損失層に加えて、オンザフライで人工学習画像ペアを生成する拡張層が提案されました。結果: トレーニング ステップでのセグメンテーションを使用したガイド登録は、深層学習ベースの画像登録に有益であることが証明されました。脳 MRI データセットから腹部 CT データセットへの事前トレーニング済みモデルを微調整すると、後者のアプリケーションのパフォーマンスがさらに向上し、満足のいくパフォーマンスを得るために大規模なデータセットの必要性がなくなりました。動的損失加重も、推論の実行時間に影響を与えることなく、パフォーマンスをわずかに改善しました。結論: シンプルな概念を使用して、一般的に使用されるディープ イメージ レジストレーション アーキテクチャである VoxelMorph のパフォーマンスを改善しました。今後の作業では、フレームワークである DDMR をさまざまなデータセットで検証して、その価値をさらに評価する必要があります。
Purpose: This study aims to explore training strategies to improve convolutional neural network-based image-to-image registration for abdominal imaging. Methods: Different training strategies, loss functions, and transfer learning schemes were considered. Furthermore, an augmentation layer which generates artificial training image pairs on-the-fly was proposed, in addition to a loss layer that enables dynamic loss weighting. Results: Guiding registration using segmentations in the training step proved beneficial for deep-learning-based image registration. Finetuning the pretrained model from the brain MRI dataset to the abdominal CT dataset further improved performance on the latter application, removing the need for a large dataset to yield satisfactory performance. Dynamic loss weighting also marginally improved performance, all without impacting inference runtime. Conclusion: Using simple concepts, we improved the performance of a commonly used deep image registration architecture, VoxelMorph. In future work, our framework, DDMR, should be validated on different datasets to further assess its value.
updated: Mon Nov 28 2022 19:03:01 GMT+0000 (UTC)
published: Mon Nov 28 2022 19:03:01 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アソシエイト