Minimizing Labeling Cost for Nuclei Instance Segmentation and Classification with Cross-domain Images and Weak Labels

Authors

  • Siqi Yang University of Queensland
  • Jun Zhang Tencent AI Lab
  • Junzhou Huang Tencent AI Lab
  • Brian C. Lovell University of Queensland
  • Xiao Han Tencent AI Lab

DOI:

https://doi.org/10.1609/aaai.v35i1.16150

Keywords:

Healthcare, Medicine & Wellness, Transfer/Adaptation/Multi-task/Meta/Automated Learning, Segmentation, Unsupervised & Self-Supervised Learning

Abstract

Nucleus instance segmentation and classification in histopathological images is an essential prerequisite in pathology diagnosis/prognosis. However, nucleus annotations (e.g., segmentation and labeling) require domain experts, and annotating nuclei at pixel-level is time-consuming and labor-intensive. Moreover, nuclei from different cancer types vary in shapes and appearances. These inter-cancer variations require careful annotations for specific cancer types. Therefore, to minimize the labeling cost, we propose a novel application that considers each cancer type as an individual domain and apply domain adaptation techniques to improve the segmentation/classification performance among different cancer types. Unlike the previous studies that focus on unsupervised or weakly-supervised domain adaptation independently, we would like to discover what kinds of labeling can achieve the most cost-effective domain adaptation performance in nucleus instance segmentation and classification. Specifically, we propose a unified framework that is applicable to different level annotations: no annotations, image-level, and point-level annotations. Cyclic adaptation with pseudo labels and adversarial discriminator are utilized for unsupervised domain alignment. Image-level or point-level annotations are additionally adopted to supervise the nucleus classification and refine the pseudo labels. Experiments demonstrate the effectiveness and efficacy of the proposed framework (jointly using unsupervised and weakly supervised learning) on adapting the segmentation and classification model from one cancer type to 18 other cancer types.

Downloads

Published

2021-05-18

How to Cite

Yang, S., Zhang, J., Huang, J., Lovell, B. . C., & Han, X. (2021). Minimizing Labeling Cost for Nuclei Instance Segmentation and Classification with Cross-domain Images and Weak Labels. Proceedings of the AAAI Conference on Artificial Intelligence, 35(1), 697-705. https://doi.org/10.1609/aaai.v35i1.16150

Issue

Section

AAAI Technical Track on Application Domains