arXiv reaDer
教師なしディープクラスタリングと強化学習は、非常に小さなトレーニングセットでMRI脳腫瘍を正確にセグメント化できます
Unsupervised deep clustering and reinforcement learning can accurately segment MRI brain tumors with very small training sets
目的:医用画像における病変のセグメンテーションは、治療反応を評価するための鍵です。最近、強化学習を病変の位置特定のために放射線画像に適用できることを示しました。さらに、強化学習が教師あり深層学習の重要な制限に対処することを示しました。つまり、大量の注釈付きトレーニングデータの必要性を排除し、教師ありアプローチに欠けている貴重な直感を提供できます。ただし、病変/関心の構造のセグメンテーションの基本的なタスクには対処しませんでした。ここでは、教師なし深層学習クラスタリングと強化学習を組み合わせて、MRIで脳病変をセグメント化する方法を紹介します。材料と方法:最初に、教師なし深層学習クラスタリングを使用して画像をクラスタリングし、各MRI画像の候補病変マスクを生成しました。次に、ユーザーは10個のトレーニング画像のそれぞれに最適なマスクを選択しました。次に、マスクを選択するための強化学習アルゴリズムをトレーニングしました。対応するトレーニング済みのディープQネットワークを、10枚の画像の個別のテストセットでテストしました。比較のために、同じトレーニング/テスト画像のセットでU-net教師あり深層学習ネットワークのトレーニングとテストも行いました。結果:教師ありアプローチはトレーニングデータにすぐに適合し、テストセットでは予想どおりにパフォーマンスが低下しましたが(16%の平均ダイススコア)、教師なしディープクラスタリングと強化学習は平均83%のダイススコアを達成しました。結論:脳腫瘍をセグメント化するための教師なしディープクラスタリングと強化学習の原理実証アプリケーションを示しました。このアプローチは、手でトレースした注釈を必要とせずに、放射線科医からの最小限の入力を必要とする人間同盟のAIを表しています。
Purpose: Lesion segmentation in medical imaging is key to evaluating treatment response. We have recently shown that reinforcement learning can be applied to radiological images for lesion localization. Furthermore, we demonstrated that reinforcement learning addresses important limitations of supervised deep learning; namely, it can eliminate the requirement for large amounts of annotated training data and can provide valuable intuition lacking in supervised approaches. However, we did not address the fundamental task of lesion/structure-of-interest segmentation. Here we introduce a method combining unsupervised deep learning clustering with reinforcement learning to segment brain lesions on MRI. Materials and Methods: We initially clustered images using unsupervised deep learning clustering to generate candidate lesion masks for each MRI image. The user then selected the best mask for each of 10 training images. We then trained a reinforcement learning algorithm to select the masks. We tested the corresponding trained deep Q network on a separate testing set of 10 images. For comparison, we also trained and tested a U-net supervised deep learning network on the same set of training/testing images. Results: Whereas the supervised approach quickly overfit the training data and predictably performed poorly on the testing set (16% average Dice score), the unsupervised deep clustering and reinforcement learning achieved an average Dice score of 83%. Conclusion: We have demonstrated a proof-of-principle application of unsupervised deep clustering and reinforcement learning to segment brain tumors. The approach represents human-allied AI that requires minimal input from the radiologist without the need for hand-traced annotation.
updated: Thu Dec 24 2020 16:46:45 GMT+0000 (UTC)
published: Thu Dec 24 2020 16:46:45 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アソシエイト