arXiv reaDer
新しい白質領域の少数ショット セグメンテーションのための知識の伝達
Knowledge Transfer for Few-shot Segmentation of Novel White Matter Tracts
畳み込みニューラル ネットワーク (CNN) は、拡散磁気共鳴画像 (dMRI) に基づく白質 (WM) 管セグメンテーションの最先端のパフォーマンスを実現しています。これらの CNN では、トレーニングのために関心のある WM トラクトの手作業による多数の描写が必要であり、一般的には労働集約的でコストがかかります。高価な手動描写は、新規の WM トラクト、つまり既存のマニュアル描写に含まれていないトラクトを分析する場合に、特に不利になる可能性があります。新規の WM トラクトを正確にセグメント化するには、既存の WM トラクトについて学んだ知識を転送することが望ましいため、新規の WM トラクトのいくつかの描写だけでも、CNN はセグメンテーションのために適切に学習できます。この論文では、数ショット設定での新しい WM トラクトのセグメンテーションへのそのような知識の転送を探ります。この目的のために古典的な微調整戦略を使用できますが、既存の WM トラクトをセグメント化するための最後のタスク固有レイヤーの情報は完全に破棄されます。この最後のレイヤーの重みは、新しい WM トラクトをセグメント化するための貴重な情報を保持できるため、情報を完全に破棄することは最適ではないと仮定します。特に、新規の WM トラクトは既存の WM トラクトと相関し、新規の WM トラクトのセグメンテーションは既存の WM トラクトのロジットで予測できると想定しています。このようにして、微調整のためにランダムな初期化よりも最終層のより良い初期化を達成することができます。さらに、既存の WM トラクトをセグメント化するための最後のレイヤーの知識のより適応的な使用は、古典的な微調整の前にウォームアップ ステージを挿入するだけで便利に実現できることを示します。提案された方法は、公開されている dMRI データセットで評価され、新しい WM トラクトの数ショット セグメンテーションのための方法の利点を示しています。
Convolutional neural networks (CNNs) have achieved stateof-the-art performance for white matter (WM) tract segmentation based on diffusion magnetic resonance imaging (dMRI). These CNNs require a large number of manual delineations of the WM tracts of interest for training, which are generally labor-intensive and costly. The expensive manual delineation can be a particular disadvantage when novel WM tracts, i.e., tracts that have not been included in existing manual delineations, are to be analyzed. To accurately segment novel WM tracts, it is desirable to transfer the knowledge learned about existing WM tracts, so that even with only a few delineations of the novel WM tracts, CNNs can learn adequately for the segmentation. In this paper, we explore the transfer of such knowledge to the segmentation of novel WM tracts in the few-shot setting. Although a classic fine-tuning strategy can be used for the purpose, the information in the last task-specific layer for segmenting existing WM tracts is completely discarded. We hypothesize that the weights of this last layer can bear valuable information for segmenting the novel WM tracts and thus completely discarding the information is not optimal. In particular, we assume that the novel WM tracts can correlate with existing WM tracts and the segmentation of novel WM tracts can be predicted with the logits of existing WM tracts. In this way, better initialization of the last layer than random initialization can be achieved for fine-tuning. Further, we show that a more adaptive use of the knowledge in the last layer for segmenting existing WM tracts can be conveniently achieved by simply inserting a warmup stage before classic fine-tuning. The proposed method was evaluated on a publicly available dMRI dataset, where we demonstrate the benefit of our method for few-shot segmentation of novel WM tracts.
updated: Sun May 30 2021 11:57:41 GMT+0000 (UTC)
published: Sun May 30 2021 11:57:41 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アソシエイト