Positive-unlabeled learning for binary and multi-class cell detection in histopathology images with incomplete annotations

Zipei Zhao1, Fengqian Pang2, Yaou Liu3, Zhiwen Liu1, Chuyang Ye1
1: Beijing Institute of Technology, 2: North China University of Technology, 3: Beijing Tiantan Hospital
Publication date: 2023/02/17
https://doi.org/10.59275/j.melba.2022-8g31
PDF · Code · arXiv

Abstract

Cell detection in histopathology images is of great interest to clinical practice and research, and convolutional neural networks (CNNs) have achieved remarkable cell detection results. Typically, to train CNN-based cell detection models, every positive instance in the training images needs to be annotated, and instances that are not labeled as positive are considered negative samples. However, manual cell annotation is complicated due to the large number and diversity of cells, and it can be difficult to ensure the annotation of every positive instance. In many cases, only incomplete annotations are available, where some of the positive instances are annotated and the others are not, and the classification loss term for negative samples in typical network training becomes incorrect. In this work, to address this problem of incomplete annotations, we propose to reformulate the training of the detection network as a positive-unlabeled learning problem. Since the instances in unannotated regions can be either positive or negative, they have unknown labels. Using the samples with unknown labels and the positively labeled samples, we first derive an approximation of the classification loss term corresponding to negative samples for binary cell detection, and based on this approximation we further extend the proposed framework to multi-class cell detection. For evaluation, experiments were performed on four publicly available datasets. The experimental results show that our method improves the performance of cell detection in histopathology images given incomplete annotations for network training.

Keywords

Cell detection · histopathology image analysis · incomplete annotation · positive-unlabeled learning

Bibtex @article{melba:2022:027:zhao, title = "Positive-unlabeled learning for binary and multi-class cell detection in histopathology images with incomplete annotations", author = "Zhao, Zipei and Pang, Fengqian and Liu, Yaou and Liu, Zhiwen and Ye, Chuyang", journal = "Machine Learning for Biomedical Imaging", volume = "1", issue = "December 2022 issue", year = "2022", pages = "1--30", issn = "2766-905X", doi = "https://doi.org/10.59275/j.melba.2022-8g31", url = "https://melba-journal.org/2022:027" }
RISTY - JOUR AU - Zhao, Zipei AU - Pang, Fengqian AU - Liu, Yaou AU - Liu, Zhiwen AU - Ye, Chuyang PY - 2022 TI - Positive-unlabeled learning for binary and multi-class cell detection in histopathology images with incomplete annotations T2 - Machine Learning for Biomedical Imaging VL - 1 IS - December 2022 issue SP - 1 EP - 30 SN - 2766-905X DO - https://doi.org/10.59275/j.melba.2022-8g31 UR - https://melba-journal.org/2022:027 ER -

2022:027 cover