arXiv reaDer
MRIセグメンテーションにおけるU-Netレイヤーへのドメインシフトの影響の構造
Anatomy of Domain Shift Impact on U-Net Layers in MRI Segmentation
ドメイン適応(DA)メソッドは、医療画像セグメンテーションタスクで広く使用されており、異なる分散のトレイン(ソース)データとテスト(ターゲット)データの問題に取り組んでいます。ターゲットドメインからの限られた数の注釈付きサンプルを使用して、監視対象のDAタスクを検討します。これは、最も関連性の高い臨床設定の1つである、注釈付きデータの可能な最小量に基づいて十分に正確なモデルを構築することに対応します。既存の方法は、主に、事前にトレーニングされた畳み込みニューラルネットワーク(CNN)の特定のレイヤーを微調整します。ただし、どのレイヤーを微調整するのが適切かについてのコンセンサスはありません。たとえば、低レベルドメインシフトのある画像の最初のレイヤーや、高レベルドメインシフトのある画像のより深いレイヤーなどです。この目的のために、SpotTUnetを提案します。これは、最適に微調整する必要のあるレイヤーを自動的に選択するCNNアーキテクチャです。より具体的には、ターゲットドメインで、このメソッドは、特定のレイヤーを微調整するか、事前にトレーニングされたネットワークから再利用するかを示すポリシーを追加で学習します。注釈付きデータが極端に不足している場合でも、この方法が柔軟性のない微調整方法の最良のものと同じレベルで実行されることを示します。次に、SpotTUnetポリシーが、ネットワークに対するドメインシフトの影響をレイヤーごとに視覚化することを示します。これは、堅牢なドメイン一般化手法の開発にさらに使用できます。 SpotTUnetのパフォーマンスを広範囲に評価するために、明示的なドメインシフトを特徴とする脳MR画像の公開されているデータセット(CC359)を使用します。再現可能な実験パイプラインをリリースします。
Domain Adaptation (DA) methods are widely used in medical image segmentation tasks to tackle the problem of differently distributed train (source) and test (target) data. We consider the supervised DA task with a limited number of annotated samples from the target domain. It corresponds to one of the most relevant clinical setups: building a sufficiently accurate model on the minimum possible amount of annotated data. Existing methods mostly fine-tune specific layers of the pretrained Convolutional Neural Network (CNN). However, there is no consensus on which layers are better to fine-tune, e.g. the first layers for images with low-level domain shift or the deeper layers for images with high-level domain shift. To this end, we propose SpotTUnet - a CNN architecture that automatically chooses the layers which should be optimally fine-tuned. More specifically, on the target domain, our method additionally learns the policy that indicates whether a specific layer should be fine-tuned or reused from the pretrained network. We show that our method performs at the same level as the best of the nonflexible fine-tuning methods even under the extreme scarcity of annotated data. Secondly, we show that SpotTUnet policy provides a layer-wise visualization of the domain shift impact on the network, which could be further used to develop robust domain generalization methods. In order to extensively evaluate SpotTUnet performance, we use a publicly available dataset of brain MR images (CC359), characterized by explicit domain shift. We release a reproducible experimental pipeline.
updated: Sat Jul 10 2021 21:13:55 GMT+0000 (UTC)
published: Sat Jul 10 2021 21:13:55 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アソシエイト