AAAI 2022technical32 citations

Label-Efficient Hybrid-Supervised Learning for Medical Image Segmentation

Junwen Pan, Qi Bi, Yanzhan Yang, Pengfei Zhu, Cheng Bian

Abstract

Due to the lack of expertise for medical image annotation, the investigation of label-efficient methodology for medical image segmentation becomes a heated topic. Recent progresses focus on the efficient utilization of weak annotations together with few strongly-annotated labels so as to achieve comparable segmentation performance in many unprofessional scenarios. However, these approaches only concentrate on the supervision inconsistency between strongly- and weakly-annotated instances but ignore the instance inconsistency inside the weakly-annotated instances, which inevitably leads to performance degradation. To address this problem, we propose a novel label-efficient hybrid-supervised framework, which considers each weakly-annotated instance individually and learns its weight guided by the gradient direction of the strongly-annotated instances, so that the high-quality prior in the strongly-annotated instances is better exploited and the weakly-annotated instances are depicted more precisely. Specially, our designed dynamic instance indicator (DII) realizes the above objectives, and is adapted to our dynamic co-regularization (DCR) framework further to alleviate the erroneous accumulation from distortions of weak annotations. Extensive experiments on two hybrid-supervised medical segmentation datasets demonstrate that with only 10% strong labels, the proposed framework can leverage the weak labels efficiently and achieve competitive performance against the 100% strong-label supervised scenario.

BibTeX
@inproceedings{aaai2022_labelefficienthy,
  title = {Label-Efficient Hybrid-Supervised Learning for Medical Image Segmentation},
  author = {Junwen Pan and Qi Bi and Yanzhan Yang and Pengfei Zhu and Cheng Bian},
  booktitle = {AAAI 2022},
  year = {2022}
}