Expert-level detection of pathologies from unannotated chest X-ray images via self-supervised learning.

Journal: Nature biomedical engineering
Published Date:

Abstract

In tasks involving the interpretation of medical images, suitably trained machine-learning models often exceed the performance of medical experts. Yet such a high-level of performance typically requires that the models be trained with relevant datasets that have been painstakingly annotated by experts. Here we show that a self-supervised model trained on chest X-ray images that lack explicit annotations performs pathology-classification tasks with accuracies comparable to those of radiologists. On an external validation dataset of chest X-rays, the self-supervised model outperformed a fully supervised model in the detection of three pathologies (out of eight), and the performance generalized to pathologies that were not explicitly annotated for model training, to multiple image-interpretation tasks and to datasets from multiple institutions.

Authors

  • Ekin Tiu
    Stanford University Department of Computer Science, Stanford, CA, USA.
  • Ellie Talius
    Stanford University Department of Computer Science, Stanford, CA, USA.
  • Pujan Patel
    Stanford University Department of Computer Science, Stanford, CA, USA.
  • Curtis P Langlotz
    Stanford University, University Medical Line, Stanford, CA, 94305, US.
  • Andrew Y Ng
  • Pranav Rajpurkar
    Harvard Medical School, Department of Biomedical Informatics, Cambridge, MA, 02115, US.