Abstract:Active learning (AL) attempts to select informative samples in a dataset to minimize the number of required labels while maximizing the performance of the model. Current AL in segmentation tasks is limited to the expansion of popular classification-based methods including entropy, MC-dropout, etc. Meanwhile, most applications in the medical field are simply migrations that fail to consider the nature of medical images, such as high class imbalance, high domain difference, and data scarcity. In this study, we address these challenges and propose a novel AL framework for medical image segmentation task. Our approach introduces a pseudo-label-based filter addressing excessive blank patches in medical abnormalities segmentation tasks, e.g., lesions, and tumors, used before the AL selection. This filter helps reduce resource usage and allows the model to focus on selecting more informative samples. For the sample selection, we propose a novel query strategy that combines both model impact and data stability by employing adversarial attack. Furthermore, we harness the adversarial samples generated during the query process to enhance the robustness of the model. The experimental results verify our framework's effectiveness over various state-of-the-art methods. Our proposed method only needs less than 14% annotated patches in 3D brain MRI multiple sclerosis (MS) segmentation tasks and 20% for Low-Grade Glioma (LGG) tumor segmentation to achieve competitive results with full supervision. These promising outcomes not only improve performance but alleviate the time burden associated with expert annotation, thereby facilitating further advancements in the field of medical image segmentation. Our code is available at https://github.com/HelenMa9998/adversarial_active_learning.

Adaptive Local-Fitting-based Active Contour Model for Medical Image Segmentation

Adaptive Energy Weight Based Active Contour Model for Robust Medical Image Segmentation

Local Weak Form Geometric Active Contours For Medical Image Segmentation

Robust Global Minimization of Active Contour Model for Multi-Object Medical Image Segmentation

Image Segmentation Based on an Active Contour Model of Partial Image Restoration with Local Cosine Fitting Energy

A convex neighbor-constrained active contour model for image segmentation

Boundary constraint factor embedded localizing active contour model for medical image segmentation

High effective medical image segmentation with model adjustable method

Active Contour Model Based on Local Bias Field Estimation for Image Segmentation.

A Robust Patch-Statistical Active Contour Model for Image Segmentation

Model-data-driven adversarial active learning for brain tumor segmentation

Multichannel convolutional neural network-based fuzzy active contour model for medical image segmentation

A Hybride Active Contour Model Driven by Global and Local Image Information

Saliency-Driven Active Contour Model for Image Segmentation

An adaptive active contour model driven by weighted local and global image fitting constraints for image segmentation

Fast and Robust Active Contours Model for Image Segmentation

ALVLS: Adaptive Local Variances-Based Level Set Framework for Medical Images Segmentation

Local Adaptive Segmentation Algorithm for 3-D Medical Image Based on Robust Feature Statistics

An active contour model based on local pre-piecewise fitting image

Active contour model-based medical image segmentation method collaborative with fuzzy C-means

Active Contour Model Based on Bias Correction and Pre-Fitting Function for Image Segmentation