arXiv reaDer
脳腫瘍分類のための転移学習ベースの能動学習フレームワーク
A Transfer Learning Based Active Learning Framework for Brain Tumor Classification
脳腫瘍は、子供と大人の間で世界的に癌関連死の主要な原因の1つです。初期段階での脳腫瘍グレード(低悪性度および高悪性度神経膠腫)の正確な分類は、予後と治療計画を成功させる上で重要な役割を果たします。ディープラーニングの最近の進歩により、人工知能対応の脳腫瘍グレーディングシステムは、放射線科医が数秒以内に医用画像を解釈するのを支援できます。ただし、深層学習手法のパフォーマンスは、注釈付きデータセットのサイズに大きく依存します。医療データの複雑さと量を考えると、大量の医療画像にラベルを付けることは非常に困難です。この作業では、脳腫瘍分類のモデルパフォーマンスの安定性と堅牢性を維持しながら、注釈コストを削減するための新しい転移学習ベースのアクティブラーニングフレームワークを提案します。 2Dスライスベースのアプローチを使用して、203人の患者の磁気共鳴画像(MRI)トレーニングデータセットと、ベースラインとして使用された66人の患者の検証データセットでモデルをトレーニングおよび微調整しました。提案された方法では、モデルは66人の患者の別のテストデータセットで82.89%の受信者動作特性(ROC)曲線(AUC)を達成しました。これは、ベースラインAUCより2.92%高く、ラベリングコストを少なくとも40%節約しました。 。メソッドの堅牢性をさらに調べるために、同じ手順を実行したバランスの取れたデータセットを作成しました。モデルは、ベースラインのAUCが78.48%であるのに対し、82%のAUCを達成しました。これにより、アクティブラーニングフレームワークで強化された提案された転移学習の堅牢性と安定性が保証され、トレーニングデータのサイズが大幅に削減されます。
Brain tumor is one of the leading causes of cancer-related death globally among children and adults. Precise classification of brain tumor grade (low-grade and high-grade glioma) at early stage plays a key role in successful prognosis and treatment planning. With recent advances in deep learning, Artificial Intelligence-enabled brain tumor grading systems can assist radiologists in the interpretation of medical images within seconds. The performance of deep learning techniques is, however, highly depended on the size of the annotated dataset. It is extremely challenging to label a large quantity of medical images given the complexity and volume of medical data. In this work, we propose a novel transfer learning based active learning framework to reduce the annotation cost while maintaining stability and robustness of the model performance for brain tumor classification. We employed a 2D slice-based approach to train and finetune our model on the Magnetic Resonance Imaging (MRI) training dataset of 203 patients and a validation dataset of 66 patients which was used as the baseline. With our proposed method, the model achieved Area Under Receiver Operating Characteristic (ROC) Curve (AUC) of 82.89% on a separate test dataset of 66 patients, which was 2.92% higher than the baseline AUC while saving at least 40% of labeling cost. In order to further examine the robustness of our method, we created a balanced dataset, which underwent the same procedure. The model achieved AUC of 82% compared with AUC of 78.48% for the baseline, which reassures the robustness and stability of our proposed transfer learning augmented with active learning framework while significantly reducing the size of training data.
updated: Mon Nov 16 2020 21:11:40 GMT+0000 (UTC)
published: Mon Nov 16 2020 21:11:40 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アソシエイト