Uncertainty Awareness Enables Efficient Labeling for Cancer Subtyping in Digital Pathology

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sivaroopan, Nirhoshan, Galappaththige, Chamuditha Jayanga, Ekanayake, Chalani, Watawana, Hasindri, Rodrigo, Ranga, Edussooriya, Chamira U. S., Wadduwage, Dushan N.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918057429958656
author Sivaroopan, Nirhoshan
Galappaththige, Chamuditha Jayanga
Ekanayake, Chalani
Watawana, Hasindri
Rodrigo, Ranga
Edussooriya, Chamira U. S.
Wadduwage, Dushan N.
author_facet Sivaroopan, Nirhoshan
Galappaththige, Chamuditha Jayanga
Ekanayake, Chalani
Watawana, Hasindri
Rodrigo, Ranga
Edussooriya, Chamira U. S.
Wadduwage, Dushan N.
contents Machine-learning-assisted cancer subtyping is a promising avenue in digital pathology. Cancer subtyping models, however, require careful training using expert annotations so that they can be inferred with a degree of known certainty (or uncertainty). To this end, we introduce the concept of uncertainty awareness into a self-supervised contrastive learning model. This is achieved by computing an evidence vector at every epoch, which assesses the model's confidence in its predictions. The derived uncertainty score is then utilized as a metric to selectively label the most crucial images that require further annotation, thus iteratively refining the training process. With just 1-10% of strategically selected annotations, we attain state-of-the-art performance in cancer subtyping on benchmark datasets. Our method not only strategically guides the annotation process to minimize the need for extensive labeled datasets, but also improves the precision and efficiency of classifications. This development is particularly beneficial in settings where the availability of labeled data is limited, offering a promising direction for future research and application in digital pathology.
format Preprint
id arxiv_https___arxiv_org_abs_2506_11439
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Uncertainty Awareness Enables Efficient Labeling for Cancer Subtyping in Digital Pathology
Sivaroopan, Nirhoshan
Galappaththige, Chamuditha Jayanga
Ekanayake, Chalani
Watawana, Hasindri
Rodrigo, Ranga
Edussooriya, Chamira U. S.
Wadduwage, Dushan N.
Computer Vision and Pattern Recognition
Machine-learning-assisted cancer subtyping is a promising avenue in digital pathology. Cancer subtyping models, however, require careful training using expert annotations so that they can be inferred with a degree of known certainty (or uncertainty). To this end, we introduce the concept of uncertainty awareness into a self-supervised contrastive learning model. This is achieved by computing an evidence vector at every epoch, which assesses the model's confidence in its predictions. The derived uncertainty score is then utilized as a metric to selectively label the most crucial images that require further annotation, thus iteratively refining the training process. With just 1-10% of strategically selected annotations, we attain state-of-the-art performance in cancer subtyping on benchmark datasets. Our method not only strategically guides the annotation process to minimize the need for extensive labeled datasets, but also improves the precision and efficiency of classifications. This development is particularly beneficial in settings where the availability of labeled data is limited, offering a promising direction for future research and application in digital pathology.
title Uncertainty Awareness Enables Efficient Labeling for Cancer Subtyping in Digital Pathology
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2506.11439