Dual Prototypical Self-Supervised Learning for One-shot Medical Image Segmentation.

Journal: Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
PMID:

Abstract

Medical image segmentation using deep learning typically requires a large quantity of well-annotated data. However, the acquisition of pixel-level annotations is arduous and expensive, often requiring the expertise of experienced medical professionals. Recent advancements in few-shot learning can help address label scarcity in medical image segmentation by leveraging a small amount of labeled data. Most research on prototypical learning for medical image segmentation involves prototypes averaged across the entire class, ignoring fine details. In this work, we propose a novel dual prototype network and introduce part prototypes to supplement local prototypes for extracting fine-grained features and enhancing model performance in one-shot medical image segmentation. Extensive experiments on the CHAOS dataset demonstrate the effectiveness of the proposed method on one-shot segmentation, outperforming other methods by a significant margin.Clinical relevance- The proposed method can address the problem of prototypical self-supervised learning in the one-shot medical image segmentation scenario, saving annotation costs in clinical practice.

Authors

  • Ziyuan Zhao
  • Zhi Qing Ng
  • Zhongyao Cheng
  • Jiahao Wang
    Unit for Laboratory Animal Medicine, Department of Microbiology and Immunology, Center for Computational Medicine and Bioinformatics, and Comprehensive Cancer Center, University of Michigan Medical School, 1301 MSRB III, 1150 W. Medical Dr, Ann Arbor, MI, 48109, USA.
  • Xulei Yang
  • Hanry Yu
    Department of Physiology, The Institute for Digital Medicine (WisDM), Yong Loo Lin School of Medicine, MD9-04-11, 2 Medical Drive, Singapore, 117593, Singapore; Institute of Bioengineering and Bioimaging, A*STAR, The Nanos, #06-01, 31 Biopolis Way, Singapore, 138669, Singapore; CAMP, Singapore-MIT Alliance for Research and Technology, 1 CREATE Way, Level 4 Enterprise Wing, Singapore, 138602, Singapore; Mechanobiology Institute, National University of Singapore, T-Lab, #05-01, 5A Engineering Drive 1, Singapore, 117411, Singapore; Ants Innovate Pte. Ltd., 7 Temasek Boulevard #12-07, Suntec Tower One, Singapore, 038987, Singapore; Lead Contact, Singapore. Electronic address: phsyuh@nus.edu.sg.
  • Cuntai Guan