arXiv reaDer
深層強化学習ベースの画像分類は、わずか30枚の画像のトレーニングセットでMRI脳腫瘍の完全なテストセット精度を実現します
Deep reinforcement learning-based image classification achieves perfect testing set accuracy for MRI brain tumors with a training set of only 30 images
目的:画像分類は、人工知能の画像化における基本的なタスクである可能性があります。最近、強化学習が、ごくわずかなトレーニングセットでも、病変の位置特定とセグメンテーションの精度を高めることができることを示しました。ここでは、画像分類のための強化学習を紹介します。特に、正常な対腫瘍を含む2DMRI脳画像にアプローチを適用します。材料と方法:Deep Q学習とTD(0)Q学習を組み合わせて使用できるように、マルチステップの画像分類を適用しました。 30枚の画像(正常15枚と腫瘍を含む15枚)のセットでトレーニングしました。 30枚の画像(正常15枚と腫瘍を含む15枚)の別のセットでテストしました。比較のために、同じトレーニング画像とテスト画像のセットで、教師あり深層学習分類ネットワークのトレーニングとテストも行いました。結果:教師ありアプローチはトレーニングデータにすぐに適合し、テストセットでは期待どおりにパフォーマンスが低下しましたが(57%の精度、ランダムな推測をわずかに上回りました)、強化学習アプローチは100%の精度を達成しました。結論:脳腫瘍の分類への強化学習の原理実証アプリケーションを示しました。わずか30枚の画像のトレーニングセットで完璧なテストセットの精度を達成しました。
Purpose: Image classification may be the fundamental task in imaging artificial intelligence. We have recently shown that reinforcement learning can achieve high accuracy for lesion localization and segmentation even with minuscule training sets. Here, we introduce reinforcement learning for image classification. In particular, we apply the approach to normal vs. tumor-containing 2D MRI brain images. Materials and Methods: We applied multi-step image classification to allow for combined Deep Q learning and TD(0) Q learning. We trained on a set of 30 images (15 normal and 15 tumor-containing). We tested on a separate set of 30 images (15 normal and 15 tumor-containing). For comparison, we also trained and tested a supervised deep-learning classification network on the same set of training and testing images. Results: Whereas the supervised approach quickly overfit the training data and as expected performed poorly on the testing set (57% accuracy, just over random guessing), the reinforcement learning approach achieved an accuracy of 100%. Conclusion: We have shown a proof-of-principle application of reinforcement learning to the classification of brain tumors. We achieved perfect testing set accuracy with a training set of merely 30 images.
updated: Thu Feb 04 2021 21:31:22 GMT+0000 (UTC)
published: Thu Feb 04 2021 21:31:22 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アソシエイト