A Deep Active Learning Framework with Information Guided Label Generation for Medical Image Segmentation

Kuan Huang, Jianhua Huang, Weichen Wang, Meng Xu, Feifei Liu

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Scopus citations

Abstract

Acquiring medical images and their segmentation labels is often time-consuming and labor-intensive. Compared to natural images, medical images need to be screened and annotated by professional doctors in segmentation tasks, especially those containing multiple organ tissues. To reduce the workload of doctors, we propose a deep active learning-based framework for medical image segmentation. Instead of all, the proposed framework can select an optimal number of medical images for label generation by doctors. The selected images are enough to train a good segmentation model because they are diversified, informational, and unique to represent the whole dataset. The proposed method consists of three steps: (1) Using an auto-encoder to extract features from unlabeled images and applying feature clustering and image entropy to select an initial subset of images for segmentation label generation; (2) Employing the mean-teacher method to train a segmentation model in a semi-supervised manner; (3) Updating the labeled subset and reducing redundancy by a deduplication uncertainty query strategy. Extensive experiments are conducted on a breast ultrasound dataset and a knee cartilage ultrasound dataset to evaluate the performance of the proposed method. Experimental results prove that our deep active learning framework can significantly reduce the number of labeled samples while achieving comparable segmentation results to fully-labeled supervision.

Original languageEnglish
Title of host publicationProceedings - 2022 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2022
EditorsDonald Adjeroh, Qi Long, Xinghua Shi, Fei Guo, Xiaohua Hu, Srinivas Aluru, Giri Narasimhan, Jianxin Wang, Mingon Kang, Ananda M. Mondal, Jin Liu
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1562-1567
Number of pages6
ISBN (Electronic)9781665468190
DOIs
StatePublished - 2022
Event2022 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2022 - Las Vegas, United States
Duration: 6 Dec 20228 Dec 2022

Publication series

NameProceedings - 2022 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2022

Conference

Conference2022 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2022
Country/TerritoryUnited States
CityLas Vegas
Period6/12/228/12/22

Keywords

  • deep active learning
  • label generation workflow
  • medical image segmentation

Fingerprint

Dive into the research topics of 'A Deep Active Learning Framework with Information Guided Label Generation for Medical Image Segmentation'. Together they form a unique fingerprint.

Cite this