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

Journal: BMC medical imaging
Published Date:

Abstract

BACKGROUND: Supervised deep learning in radiology suffers from notorious inherent limitations: 1) It requires large, hand-annotated data sets; (2) It is non-generalizable; and (3) It lacks explainability and intuition. It has recently been proposed that reinforcement learning addresses all three of these limitations. Notable prior work applied deep reinforcement learning to localize brain tumors with radiologist eye tracking points, which limits the state-action space. Here, we generalize Deep Q Learning to a gridworld-based environment so that only the images and image masks are required.

Authors

  • J N Stember
    Department of Radiology, Columbia University Medical Center - NYPH, New York, NY, 10032, USA. joestember@gmail.com.
  • H Shalu
    Department of Aerospace Engineering, Indian Institute of Technology Madras, Chennai, 600 036 India.