CLASS-M: Adaptive stain separation-based contrastive learning with pseudo-labeling for histopathological image classification

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Bodong, Manoochehri, Hamid, Ho, Man Minh, Fooladgar, Fahimeh, Chong, Yosep, Knudsen, Beatrice S., Sirohi, Deepika, Tasdizen, Tolga
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910696147517440
author Zhang, Bodong
Manoochehri, Hamid
Ho, Man Minh
Fooladgar, Fahimeh
Chong, Yosep
Knudsen, Beatrice S.
Sirohi, Deepika
Tasdizen, Tolga
author_facet Zhang, Bodong
Manoochehri, Hamid
Ho, Man Minh
Fooladgar, Fahimeh
Chong, Yosep
Knudsen, Beatrice S.
Sirohi, Deepika
Tasdizen, Tolga
contents Histopathological image classification is an important task in medical image analysis. Recent approaches generally rely on weakly supervised learning due to the ease of acquiring case-level labels from pathology reports. However, patch-level classification is preferable in applications where only a limited number of cases are available or when local prediction accuracy is critical. On the other hand, acquiring extensive datasets with localized labels for training is not feasible. In this paper, we propose a semi-supervised patch-level histopathological image classification model, named CLASS-M, that does not require extensively labeled datasets. CLASS-M is formed by two main parts: a contrastive learning module that uses separated Hematoxylin and Eosin images generated through an adaptive stain separation process, and a module with pseudo-labels using MixUp. We compare our model with other state-of-the-art models on two clear cell renal cell carcinoma datasets. We demonstrate that our CLASS-M model has the best performance on both datasets. Our code is available at github.com/BzhangURU/Paper_CLASS-M/tree/main
format Preprint
id arxiv_https___arxiv_org_abs_2312_06978
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle CLASS-M: Adaptive stain separation-based contrastive learning with pseudo-labeling for histopathological image classification
Zhang, Bodong
Manoochehri, Hamid
Ho, Man Minh
Fooladgar, Fahimeh
Chong, Yosep
Knudsen, Beatrice S.
Sirohi, Deepika
Tasdizen, Tolga
Computer Vision and Pattern Recognition
Histopathological image classification is an important task in medical image analysis. Recent approaches generally rely on weakly supervised learning due to the ease of acquiring case-level labels from pathology reports. However, patch-level classification is preferable in applications where only a limited number of cases are available or when local prediction accuracy is critical. On the other hand, acquiring extensive datasets with localized labels for training is not feasible. In this paper, we propose a semi-supervised patch-level histopathological image classification model, named CLASS-M, that does not require extensively labeled datasets. CLASS-M is formed by two main parts: a contrastive learning module that uses separated Hematoxylin and Eosin images generated through an adaptive stain separation process, and a module with pseudo-labels using MixUp. We compare our model with other state-of-the-art models on two clear cell renal cell carcinoma datasets. We demonstrate that our CLASS-M model has the best performance on both datasets. Our code is available at github.com/BzhangURU/Paper_CLASS-M/tree/main
title CLASS-M: Adaptive stain separation-based contrastive learning with pseudo-labeling for histopathological image classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.06978