Deep reinforcement learning-based image classification achieves perfect testing set accuracy for MRI brain tumors with a training set of only 30 images

by   Joseph Stember, et al.

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 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.



There are no comments yet.


page 6

page 7

page 9


Reinforcement learning using Deep Q Networks and Q learning accurately localizes brain tumors on MRI with very small training sets

Purpose Supervised deep learning in radiology suffers from notorious inh...

Deep reinforcement learning with automated label extraction from clinical reports accurately classifies 3D MRI brain volumes

Purpose: Image classification is perhaps the most fundamental task in im...

Unsupervised deep clustering and reinforcement learning can accurately segment MRI brain tumors with very small training sets

Purpose: Lesion segmentation in medical imaging is key to evaluating tre...

Deep Learning models for benign and malign Ocular Tumor Growth Estimation

Relatively abundant availability of medical imaging data has provided si...

Osteoarthritis Disease Detection System using Self Organizing Maps Method based on Ossa Manus X-Ray

Osteoarthritis is a disease found in the world, including in Indonesia. ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.