Skip to main navigation Skip to search Skip to main content

Weakly Supervised Cell Segmentation by Point Annotation

  • Stony Brook University

Research output: Contribution to journalArticlepeer-review

72 Scopus citations

Abstract

We propose weakly supervised training schemes to train end-To-end cell segmentation networks that only require a single point annotation per cell as the training label and generate a high-quality segmentation mask close to those fully supervised methods using mask annotation on cells. Three training schemes are investigated to train cell segmentation networks, using the point annotation. First, self-Training is performed to learn additional information near the annotated points. Next, co-Training is applied to learn more cell regions using multiple networks that supervise each other. Finally, a hybrid-Training scheme is proposed to leverage the advantages of both self-Training and co-Training. During the training process, we propose a divergence loss to avoid the overfitting and a consistency loss to enforce the consensus among multiple co-Trained networks. Furthermore, we propose weakly supervised learning with human in the loop, aiming at achieving high segmentation accuracy and annotation efficiency simultaneously. Evaluated on two benchmark datasets, our proposal achieves high-quality cell segmentation results comparable to the fully supervised methods, but with much less amount of human annotation effort.

Original languageEnglish
Pages (from-to)2736-2747
Number of pages12
JournalIEEE Transactions on Medical Imaging
Volume40
Issue number10
DOIs
StatePublished - Oct 1 2021

Keywords

  • Cell segmentation
  • human in the loop
  • neural networks
  • point annotation
  • weakly supervised learning

Fingerprint

Dive into the research topics of 'Weakly Supervised Cell Segmentation by Point Annotation'. Together they form a unique fingerprint.

Cite this