Large scale deep learning for computer aided detection of mammographic lesions.

Journal: Medical image analysis
Published Date:

Abstract

Recent advances in machine learning yielded new techniques to train deep neural networks, which resulted in highly successful applications in many pattern recognition tasks such as object detection and speech recognition. In this paper we provide a head-to-head comparison between a state-of-the art in mammography CAD system, relying on a manually designed feature set and a Convolutional Neural Network (CNN), aiming for a system that can ultimately read mammograms independently. Both systems are trained on a large data set of around 45,000 images and results show the CNN outperforms the traditional CAD system at low sensitivity and performs comparable at high sensitivity. We subsequently investigate to what extent features such as location and patient information and commonly used manual features can still complement the network and see improvements at high specificity over the CNN especially with location and context features, which contain information not available to the CNN. Additionally, a reader study was performed, where the network was compared to certified screening radiologists on a patch level and we found no significant difference between the network and the readers.

Authors

  • Thijs Kooi
    Diagnostic Image Analysis Group, Department of Radiology, Radboud University Medical Center, Nijmegen, The Netherlands. Electronic address: thijs.kooi@radboudumc.nl.
  • Geert Litjens
    Department of Pathology, Radboud University Medical Center, Nijmegen, The Netherlands.
  • Bram van Ginneken
    Diagnostic Image Analysis Group, Department of Radiology and Nuclear Medicine, Radboud University Medical Center, Nijmegen, The Netherlands; Fraunhofer Mevis, Bremen, Germany.
  • Albert Gubern-Mérida
    Diagnostic Image Analysis Group, Department of Radiology, Radboud University Medical Center, Nijmegen, The Netherlands.
  • Clara I Sanchez
  • Ritse Mann
    Diagnostic Image Analysis Group, Department of Radiology, Radboud University Medical Center, Nijmegen, The Netherlands.
  • Ard den Heeten
    Department of Radiology, University Medical Centre Amsterdam, Amsterdam, The Netherlands.
  • Nico Karssemeijer