Distribution-aware Margin Calibration for Medical Image Segmentation
- URL: http://arxiv.org/abs/2011.01462v1
- Date: Tue, 3 Nov 2020 04:07:47 GMT
- Title: Distribution-aware Margin Calibration for Medical Image Segmentation
- Authors: Zhibin Li, Litao Yu, Jian Zhang
- Abstract summary: Jaccard index, also known as Intersection-over-Union (IoU) score, is one of the most critical evaluation metrics in medical image segmentation.
We present a novel data-distribution-aware margin calibration method for a better generalization of the mIoU over the whole data-distribution.
We evaluate the effectiveness of the proposed margin calibration method on two medical image segmentation datasets.
- Score: 11.391027138349482
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The Jaccard index, also known as Intersection-over-Union (IoU score), is one
of the most critical evaluation metrics in medical image segmentation. However,
directly optimizing the mean IoU (mIoU) score over multiple objective classes
is an open problem. Although some algorithms have been proposed to optimize its
surrogates, there is no guarantee provided for their generalization ability. In
this paper, we present a novel data-distribution-aware margin calibration
method for a better generalization of the mIoU over the whole
data-distribution, underpinned by a rigid lower bound. This scheme ensures a
better segmentation performance in terms of IoU scores in practice. We evaluate
the effectiveness of the proposed margin calibration method on two medical
image segmentation datasets, showing substantial improvements of IoU scores
over other learning schemes using deep segmentation models.
Related papers
- Joint Optimization of Class-Specific Training- and Test-Time Data
Augmentation in Segmentation [35.41274775082237]
This paper presents an effective and general data augmentation framework for medical image segmentation.
We adopt a computationally efficient and data-efficient gradient-based meta-learning scheme to align the distribution of training and validation data.
We demonstrate the effectiveness of our method on four medical image segmentation tasks with two state-of-the-art segmentation models, DeepMedic and nnU-Net.
arXiv Detail & Related papers (2023-05-30T14:48:45Z) - Rethinking Semi-Supervised Medical Image Segmentation: A
Variance-Reduction Perspective [51.70661197256033]
We propose ARCO, a semi-supervised contrastive learning framework with stratified group theory for medical image segmentation.
We first propose building ARCO through the concept of variance-reduced estimation and show that certain variance-reduction techniques are particularly beneficial in pixel/voxel-level segmentation tasks.
We experimentally validate our approaches on eight benchmarks, i.e., five 2D/3D medical and three semantic segmentation datasets, with different label settings.
arXiv Detail & Related papers (2023-02-03T13:50:25Z) - Efficient Bayesian Uncertainty Estimation for nnU-Net [3.8186085899889943]
We introduce a novel method to estimate nnU-Net uncertainty for medical image segmentation.
We boost the segmentation performance over the original nnU-Net via marginalizing multi-modal posterior models.
The proposed method further strengthens nnU-Net for medical image segmentation in terms of both segmentation accuracy and quality control.
arXiv Detail & Related papers (2022-12-12T23:12:19Z) - PCA: Semi-supervised Segmentation with Patch Confidence Adversarial
Training [52.895952593202054]
We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation.
PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state.
Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
arXiv Detail & Related papers (2022-07-24T07:45:47Z) - MS Lesion Segmentation: Revisiting Weighting Mechanisms for Federated
Learning [92.91544082745196]
Federated learning (FL) has been widely employed for medical image analysis.
FL's performance is limited for multiple sclerosis (MS) lesion segmentation tasks.
We propose the first FL MS lesion segmentation framework via two effective re-weighting mechanisms.
arXiv Detail & Related papers (2022-05-03T14:06:03Z) - An Embarrassingly Simple Consistency Regularization Method for
Semi-Supervised Medical Image Segmentation [0.0]
The scarcity of pixel-level annotation is a prevalent problem in medical image segmentation tasks.
We introduce a novel regularization strategy involving computation-based mixing for semi-supervised medical image segmentation.
arXiv Detail & Related papers (2022-02-01T16:21:14Z) - Distribution-aware Margin Calibration for Semantic Segmentation in
Images [78.65312390695038]
Jaccard index, also known as Intersection-over-Union (IoU), is one of the most critical evaluation metrics in image semantic segmentation.
Direct optimization of IoU score is very difficult because the learning objective is neither differentiable nor decomposable.
We propose a margin calibration method, which can be directly used as a learning objective, for an improved generalization of IoU over the data-distribution.
arXiv Detail & Related papers (2021-12-21T22:38:25Z) - Margin Preserving Self-paced Contrastive Learning Towards Domain
Adaptation for Medical Image Segmentation [51.93711960601973]
We propose a novel margin preserving self-paced contrastive Learning model for cross-modal medical image segmentation.
With the guidance of progressively refined semantic prototypes, a novel margin preserving contrastive loss is proposed to boost the discriminability of embedded representation space.
Experiments on cross-modal cardiac segmentation tasks demonstrate that MPSCL significantly improves semantic segmentation performance.
arXiv Detail & Related papers (2021-03-15T15:23:10Z) - KiU-Net: Towards Accurate Segmentation of Biomedical Images using
Over-complete Representations [59.65174244047216]
We propose an over-complete architecture (Ki-Net) which involves projecting the data onto higher dimensions.
This network, when augmented with U-Net, results in significant improvements in the case of segmenting small anatomical landmarks.
We evaluate the proposed method on the task of brain anatomy segmentation from 2D Ultrasound of preterm neonates.
arXiv Detail & Related papers (2020-06-08T18:59:24Z)
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.