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