arXiv reaDer
堅牢で一般化された医療画像セグメンテーションフレームワークに向けて
Towards to Robust and Generalized Medical Image Segmentation Framework
ディープラーニングベースのコンピューター支援診断は、医用画像をレビューおよび分析するために徐々に展開されています。ただし、このパラダイムは、堅牢性と一般化が不十分なため、実際の臨床アプリケーションでは制限されています。問題は、トレーニングデータが不足しているため、より不吉です。この論文では、転移学習の観点から課題に取り組んでいます。自然画像ドメインから医用画像ドメインに知識を転送するという一般的な設定とは異なり、同じドメインからの知識がモデルの堅牢性と一般化をさらに高めることがわかります。したがって、堅牢な一般化された医療画像セグメンテーションのための新しい2段階のフレームワークを提案します。まず、ローカルおよびグローバルな知識を学習するために、教師なしタイルワイズオートエンコーダ事前トレーニングアーキテクチャが提案されます。次に、補助再構築ネットワークと組み合わせたダウンストリームセグメンテーションモデルを設計します。再構築ブランチは、モデルがより一般的なセマンティック機能をキャプチャするように促します。マルチチェストX線データセットでの肺セグメンテーションの実験が行われます。包括的な結果は、特に限られたトレーニングデータのシナリオの下で、破損に対する提案されたフレームワークの優れた堅牢性と、目に見えないデータセットでの高い一般化パフォーマンスを示しています。
Deep learning-based computer-aided diagnosis is gradually deployed to review and analyze medical images. However, this paradigm is restricted in real-world clinical applications due to the poor robustness and generalization. The issue is more sinister with a lack of training data. In this paper, we address the challenge from the transfer learning point of view. Different from the common setting that transferring knowledge from the natural image domain to the medical image domain, we find the knowledge from the same domain further boosts the model robustness and generalization. Therefore, we propose a novel two-stage framework for robust generalized medical image segmentation. Firstly, an unsupervised tile-wise autoencoder pretraining architecture is proposed to learn local and global knowledge. Secondly, the downstream segmentation model coupled with an auxiliary reconstruction network is designed. The reconstruction branch encourages the model to capture more general semantic features. Experiments of lung segmentation on multi chest X-ray datasets are conducted. Comprehensive results demonstrate the superior robustness of the proposed framework to corruption and high generalization performance on unseen datasets, especially under the scenario of the limited training data.
updated: Tue Feb 22 2022 06:23:53 GMT+0000 (UTC)
published: Mon Aug 09 2021 05:58:49 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アソシエイト