arXiv reaDer
堅牢な医療画像セグメンテーションのためのスタイルカリキュラム学習
Style Curriculum Learning for Robust Medical Image Segmentation
ディープセグメンテーションモデルのパフォーマンスは、トレーニングデータセットとテストデータセットの間の画像強度の分布シフトのために低下することがよくあります。これは、マルチベンダースキャナーを使用して取得されたデータを含む多施設共同研究で特に顕著であり、取得プロトコルはさまざまです。シフトは事前に知られていないことが多く、したがってモデル化が難しいため、この劣化に対処することは困難です。そのような分布シフトの存在下でロバストなセグメンテーションを確実にするための新しいフレームワークを提案します。私たちの貢献は3つあります。まず、カリキュラム学習の精神に触発されて、簡単なモードを使用してセグメンテーションモデルをトレーニングするための新しいスタイルのカリキュラムを設計します。カリキュラムサンプルの生成には、スタイルフュージョンを使用したスタイル転送モデルが採用されています。複雑で敵対的なスタイルのサンプルに徐々に焦点を合わせると、モデルの堅牢性が大幅に向上します。次に、カリキュラムの複雑さを主観的に定義する代わりに、自動化された勾配操作方法を採用して、困難で敵対的なサンプル生成プロセスを制御します。第三に、勾配を局所的に集約し、勾配操作中のトレーニングを安定させるための局所勾配符号戦略を提案します。提案されたフレームワークは、ターゲットデータを使用せずに未知の分布に一般化できます。公開されているM \&Msチャレンジデータセットでの広範な実験は、提案されたフレームワークが深いモデルを未知の分布にうまく一般化し、セグメンテーションの精度を大幅に向上させることができることを示しています。
The performance of deep segmentation models often degrades due to distribution shifts in image intensities between the training and test data sets. This is particularly pronounced in multi-centre studies involving data acquired using multi-vendor scanners, with variations in acquisition protocols. It is challenging to address this degradation because the shift is often not known a priori and hence difficult to model. We propose a novel framework to ensure robust segmentation in the presence of such distribution shifts. Our contribution is three-fold. First, inspired by the spirit of curriculum learning, we design a novel style curriculum to train the segmentation models using an easy-to-hard mode. A style transfer model with style fusion is employed to generate the curriculum samples. Gradually focusing on complex and adversarial style samples can significantly boost the robustness of the models. Second, instead of subjectively defining the curriculum complexity, we adopt an automated gradient manipulation method to control the hard and adversarial sample generation process. Third, we propose the Local Gradient Sign strategy to aggregate the gradient locally and stabilise training during gradient manipulation. The proposed framework can generalise to unknown distribution without using any target data. Extensive experiments on the public M\&Ms Challenge dataset demonstrate that our proposed framework can generalise deep models well to unknown distributions and achieve significant improvements in segmentation accuracy.
updated: Sun Aug 01 2021 08:56:24 GMT+0000 (UTC)
published: Sun Aug 01 2021 08:56:24 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アソシエイト