Accurate recognition of colorectal cancer with semi-supervised deep learning on pathological images

Nat Commun. 2021 Nov 2;12(1):6311. doi: 10.1038/s41467-021-26643-8.

Abstract

Machine-assisted pathological recognition has been focused on supervised learning (SL) that suffers from a significant annotation bottleneck. We propose a semi-supervised learning (SSL) method based on the mean teacher architecture using 13,111 whole slide images of colorectal cancer from 8803 subjects from 13 independent centers. SSL (~3150 labeled, ~40,950 unlabeled; ~6300 labeled, ~37,800 unlabeled patches) performs significantly better than the SL. No significant difference is found between SSL (~6300 labeled, ~37,800 unlabeled) and SL (~44,100 labeled) at patch-level diagnoses (area under the curve (AUC): 0.980 ± 0.014 vs. 0.987 ± 0.008, P value = 0.134) and patient-level diagnoses (AUC: 0.974 ± 0.013 vs. 0.980 ± 0.010, P value = 0.117), which is close to human pathologists (average AUC: 0.969). The evaluation on 15,000 lung and 294,912 lymph node images also confirm SSL can achieve similar performance as that of SL with massive annotations. SSL dramatically reduces the annotations, which has great potential to effectively build expert-level pathological artificial intelligence platforms in practice.

Publication types

  • Multicenter Study
  • Research Support, N.I.H., Extramural
  • Research Support, Non-U.S. Gov't

MeSH terms

  • Artificial Intelligence / standards*
  • Colorectal Neoplasms / classification
  • Colorectal Neoplasms / diagnostic imaging
  • Colorectal Neoplasms / pathology*
  • Deep Learning / standards*
  • Humans
  • Lung Neoplasms / classification
  • Lung Neoplasms / diagnostic imaging
  • Lung Neoplasms / pathology*
  • Lymphatic Metastasis
  • Neural Networks, Computer
  • ROC Curve
  • Supervised Machine Learning / standards*

Associated data

  • figshare/10.6084/m9.figshare.15072546.v1