arXiv reaDer
Adversarial Continual Learning for Multi-Domain Hippocampal Segmentation
医用画像のディープラーニングは、データの可用性に関する時間的およびプライバシー関連の制限に悩まされています。実行可能なモデルを引き続き取得するために、継続学習は、データが利用可能になったときに、順番にトレーニングすることを目的としています。継続的な学習方法が直面する主な課題は、壊滅的な忘却、つまり、以前に遭遇したデータのパフォーマンスの低下を防ぐことです。この問題により、医療アプリケーションのセグメンテーションモデルの継続的なトレーニングが非常に困難になります。ただし、多くの場合、少なくとも2つの異なるドメインからのデータが利用可能であり、ドメイン固有の情報を無視する方法でモデルをトレーニングするために利用できます。 2つ以上のデータセットの同時可用性を活用して、敵対的な方法でコンテンツとドメイン間のもつれを解くことを学習するアーキテクチャを提案します。ドメイン不変のコンテンツ表現は、継続的なセマンティックセグメンテーションの基礎を築きます。私たちのアプローチは、ドメイン適応からインスピレーションを得て、脳MRIにおける海馬セグメンテーションの継続的な学習と組み合わせています。私たちの方法が壊滅的な忘却を減らし、最先端の継続的な学習方法よりも優れていることを示します。
Deep learning for medical imaging suffers from temporal and privacy-related restrictions on data availability. To still obtain viable models, continual learning aims to train in sequential order, as and when data is available. The main challenge that continual learning methods face is to prevent catastrophic forgetting, i.e., a decrease in performance on the data encountered earlier. This issue makes continuous training of segmentation models for medical applications extremely difficult. Yet, often, data from at least two different domains is available which we can exploit to train the model in a way that it disregards domain-specific information. We propose an architecture that leverages the simultaneous availability of two or more datasets to learn a disentanglement between the content and domain in an adversarial fashion. The domain-invariant content representation then lays the base for continual semantic segmentation. Our approach takes inspiration from domain adaptation and combines it with continual learning for hippocampal segmentation in brain MRI. We showcase that our method reduces catastrophic forgetting and outperforms state-of-the-art continual learning methods.
updated: Tue Jul 20 2021 09:43:54 GMT+0000 (UTC)
published: Mon Jul 19 2021 10:55:21 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト