Graph Segmentation-Based Pseudo-Labeling for Semi-Supervised Pathology Image Classification

Hong Kyu Shin, Kwang Hyun Uhmn, Kyuyeon Choi, Zhixin Xu, Seung Won Jung, Sung Jea Ko

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)

Abstract

Pathology image classification is an important step in cancer diagnosis and precision treatment. Training a pathology image classification model in a fully supervised manner requires exhaustive pixel-level manual annotations from pathologists, which may not be practical in real applications. Semi-supervised learning (SSL) has been widely used to exploit large amounts of unlabeled data to facilitate model training with a small set of labeled data. However, due to the limited annotations, it still suffers from the issue of inaccurate pseudo-labels of unlabeled data. In this paper, we propose a novel framework for semi-supervised pathology image classification, which incorporates graph-based segmentation to refine initial pseudo-labels of tissue regions by considering local and global contextual relationships of patches in whole-slide images (WSIs). Moreover, we define a new energy function for graph construction that allows the graph to take into account the uncertainty of network predictions on unlabeled data. Extensive experiments on two different pathology image datasets demonstrate the effectiveness of our method compared with state-of-the-art SSL baselines. In particular, when using 5% labeled data, our approach outperforms a strong baseline by 2.81% AUC.

Original languageEnglish
Pages (from-to)93960-93970
Number of pages11
JournalIEEE Access
Volume10
DOIs
Publication statusPublished - 2022

Bibliographical note

Publisher Copyright:
© 2013 IEEE.

Keywords

  • Graph-based segmentation
  • pathology
  • pseudo-labeling
  • semi-supervised learning

ASJC Scopus subject areas

  • General Computer Science
  • General Materials Science
  • General Engineering
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Graph Segmentation-Based Pseudo-Labeling for Semi-Supervised Pathology Image Classification'. Together they form a unique fingerprint.

Cite this