Self-supervised Assisted Active Learning for Skin Lesion Segmentation
- URL: http://arxiv.org/abs/2205.07021v1
- Date: Sat, 14 May 2022 09:40:18 GMT
- Title: Self-supervised Assisted Active Learning for Skin Lesion Segmentation
- Authors: Ziyuan Zhao, Wenjing Lu, Zeng Zeng, Kaixin Xu, Bharadwaj Veeravalli,
Cuntai Guan
- Abstract summary: Label scarcity has been a long-standing issue for biomedical image segmentation, due to high annotation costs and professional requirements.
We propose a novel self-supervised assisted active learning framework in the cold-start setting, in which the segmentation model is first warmed up with self-supervised learning.
Our approach is capable of achieving promising performance with substantial improvements over existing baselines.
- Score: 18.78959113954792
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Label scarcity has been a long-standing issue for biomedical image
segmentation, due to high annotation costs and professional requirements.
Recently, active learning (AL) strategies strive to reduce annotation costs by
querying a small portion of data for annotation, receiving much traction in the
field of medical imaging. However, most of the existing AL methods have to
initialize models with some randomly selected samples followed by active
selection based on various criteria, such as uncertainty and diversity. Such
random-start initialization methods inevitably introduce under-value redundant
samples and unnecessary annotation costs. For the purpose of addressing the
issue, we propose a novel self-supervised assisted active learning framework in
the cold-start setting, in which the segmentation model is first warmed up with
self-supervised learning (SSL), and then SSL features are used for sample
selection via latent feature clustering without accessing labels. We assess our
proposed methodology on skin lesions segmentation task. Extensive experiments
demonstrate that our approach is capable of achieving promising performance
with substantial improvements over existing baselines.
Related papers
- Active Learning for WBAN-based Health Monitoring [20.872611484417433]
We consider a novel active learning problem motivated by the need of learning machine learning models for health monitoring in wireless body area network (WBAN)
Due to the limited resources at body sensors, collecting each unlabeled sample in WBAN incurs a nontrivial cost.
Our solution can drastically save the data curation cost without sacrificing the quality of the target model.
arXiv Detail & Related papers (2024-08-05T22:19:01Z) - SAM-Driven Weakly Supervised Nodule Segmentation with Uncertainty-Aware Cross Teaching [13.5553526185399]
Automated nodule segmentation is essential for computer-assisted diagnosis in ultrasound images.
Recently, segmentation foundation models like SAM have shown impressive generalizability on natural images.
In this work, we devise a novel weakly supervised framework that effectively utilizes the segmentation foundation model to generate pseudo-labels.
arXiv Detail & Related papers (2024-07-18T14:27:54Z) - Learnable Weight Initialization for Volumetric Medical Image Segmentation [66.3030435676252]
We propose a learnable weight-based hybrid medical image segmentation approach.
Our approach is easy to integrate into any hybrid model and requires no external training data.
Experiments on multi-organ and lung cancer segmentation tasks demonstrate the effectiveness of our approach.
arXiv Detail & Related papers (2023-06-15T17:55:05Z) - TAAL: Test-time Augmentation for Active Learning in Medical Image
Segmentation [7.856339385917824]
This paper proposes Test-time Augmentation for Active Learning (TAAL), a novel semi-supervised active learning approach for segmentation.
Our results on a publicly-available dataset of cardiac images show that TAAL outperforms existing baseline methods in both fully-supervised and semi-supervised settings.
arXiv Detail & Related papers (2023-01-16T22:19:41Z) - Active Pointly-Supervised Instance Segmentation [106.38955769817747]
We present an economic active learning setting, named active pointly-supervised instance segmentation (APIS)
APIS starts with box-level annotations and iteratively samples a point within the box and asks if it falls on the object.
The model developed with these strategies yields consistent performance gain on the challenging MS-COCO dataset.
arXiv Detail & Related papers (2022-07-23T11:25:24Z) - LifeLonger: A Benchmark for Continual Disease Classification [59.13735398630546]
We introduce LifeLonger, a benchmark for continual disease classification on the MedMNIST collection.
Task and class incremental learning of diseases address the issue of classifying new samples without re-training the models from scratch.
Cross-domain incremental learning addresses the issue of dealing with datasets originating from different institutions while retaining the previously obtained knowledge.
arXiv Detail & Related papers (2022-04-12T12:25:05Z) - Active Surrogate Estimators: An Active Learning Approach to
Label-Efficient Model Evaluation [59.7305309038676]
We propose Active Surrogate Estimators (ASEs) for model evaluation.
We find that ASEs offer greater label-efficiency than the current state-of-the-art.
arXiv Detail & Related papers (2022-02-14T17:15:18Z) - Novel Class Discovery in Semantic Segmentation [104.30729847367104]
We introduce a new setting of Novel Class Discovery in Semantic (NCDSS)
It aims at segmenting unlabeled images containing new classes given prior knowledge from a labeled set of disjoint classes.
In NCDSS, we need to distinguish the objects and background, and to handle the existence of multiple classes within an image.
We propose the Entropy-based Uncertainty Modeling and Self-training (EUMS) framework to overcome noisy pseudo-labels.
arXiv Detail & Related papers (2021-12-03T13:31:59Z) - Few-shot segmentation of medical images based on meta-learning with
implicit gradients [0.48861336570452174]
We propose to exploit an optimization-based implicit model agnostic meta-learning iMAML algorithm in a few-shot setting for medical image segmentation.
Our approach can leverage the learned weights from a diverse set of training samples and can be deployed on a new unseen dataset.
arXiv Detail & Related papers (2021-06-06T19:52:06Z) - Interpretability-Driven Sample Selection Using Self Supervised Learning
For Disease Classification And Segmentation [4.898744396854313]
We propose a novel sample selection methodology based on deep features leveraging information contained in interpretability saliency maps.
We show our proposed self supervised approach outperforms other approaches in selecting informative samples leading to state of the art performance with fewer samples.
arXiv Detail & Related papers (2021-04-13T10:46:33Z) - Minimax Active Learning [61.729667575374606]
Active learning aims to develop label-efficient algorithms by querying the most representative samples to be labeled by a human annotator.
Current active learning techniques either rely on model uncertainty to select the most uncertain samples or use clustering or reconstruction to choose the most diverse set of unlabeled examples.
We develop a semi-supervised minimax entropy-based active learning algorithm that leverages both uncertainty and diversity in an adversarial manner.
arXiv Detail & Related papers (2020-12-18T19:03:40Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.