Addressing Class Imbalance in Semi-supervised Image Segmentation: A Study on Cardiac MRI

08/31/2022
by   Hritam Basak, et al.
0

Due to the imbalanced and limited data, semi-supervised medical image segmentation methods often fail to produce superior performance for some specific tailed classes. Inadequate training for those particular classes could introduce more noise to the generated pseudo labels, affecting overall learning. To alleviate this shortcoming and identify the under-performing classes, we propose maintaining a confidence array that records class-wise performance during training. A fuzzy fusion of these confidence scores is proposed to adaptively prioritize individual confidence metrics in every sample rather than traditional ensemble approaches, where a set of predefined fixed weights are assigned for all the test cases. Further, we introduce a robust class-wise sampling method and dynamic stabilization for a better training strategy. Our proposed method considers all the under-performing classes with dynamic weighting and tries to remove most of the noises during training. Upon evaluation on two cardiac MRI datasets, ACDC and MMWHS, our proposed method shows effectiveness and generalizability and outperforms several state-of-the-art methods found in the literature.

READ FULL TEXT
research
07/22/2023

DHC: Dual-debiased Heterogeneous Co-training Framework for Class-imbalanced Semi-supervised Medical Image Segmentation

The volume-wise labeling of 3D medical images is expertise-demanded and ...
research
07/24/2023

Semi-Supervised Medical Image Segmentation with Co-Distribution Alignment

Medical image segmentation has made significant progress when a large am...
research
10/11/2021

Semi-Supervised Semantic Segmentation via Adaptive Equalization Learning

Due to the limited and even imbalanced data, semi-supervised semantic se...
research
06/14/2023

C^3PS: Context-aware Conditional Cross Pseudo Supervision for Semi-supervised Medical Image Segmentation

Semi-supervised learning (SSL) methods, which can leverage a large amoun...
research
05/14/2022

Voxel-wise Adversarial Semi-supervised Learning for Medical Image Segmentation

Semi-supervised learning for medical image segmentation is an important ...
research
08/08/2022

Bayesian Pseudo Labels: Expectation Maximization for Robust and Efficient Semi-Supervised Segmentation

This paper concerns pseudo labelling in segmentation. Our contribution i...
research
07/22/2020

Learning Directional Feature Maps for Cardiac MRI Segmentation

Cardiac MRI segmentation plays a crucial role in clinical diagnosis for ...

Please sign up or login with your details

Forgot password? Click here to reset