返回
摘要
En 中文
Semi-supervised learning (SSL) has emerged as a powerful technique to mitigate the scarcity of labeled data. However, the effectiveness of most SSL methods relies on the assumption of a balanced class distribution, which often proves unrealistic, especially in medical imaging scenarios. To address this challenge, we propose Class-Specific Thresholding (CST) for imbalanced SSL. Specifically, CST dynamically integrates the model's learning status, class-specific learning effects, and data class distribution to estimate confidence thresholds for each class. These thresholds enhance the reliability of selected unlabeled data, particularly for minority classes. Additionally, we introduce a class-sensitive unsupervised loss function that further intensifies the model's focus on minority classes at minimal computational cost by leveraging predicted class distributions from previous inputs. Extensive experiments on highly imbalanced skin disease and endoscopy image datasets demonstrate that CST significantly outperforms state-of-the-art SSL methods, particularly in improving the accuracy for minority classes. These results validate the effectiveness of CST in addressing class imbalance.
Keyword:
Training
Predictive models
Data models
Computational modeling
Thresholding (Imaging)
Accuracy
Sensitivity
Deep learning
semi-supervised learning
classification
class imbalance
期刊
IF:
9.6
论文数:
1.1W
被引数:
1.7W
机构
引用论文
HybridVPS: Hybrid-Supervised Video Polyp Segmentation Under Low-Cost LabelsHybridVPS: 低成本标签下的混合监督视频息肉分割
Semi-Supervised EEG Signals Classification System for Epileptic Seizure Detection用于癫痫发作检测的半监督脑电信号分类系统
The impact of class imbalance in classification performance metrics based on the binary confusion matrix
PATTERN RECOGNITION
IF7.6
Deep semi-supervised multiple instance learning with self-correction for DME classification from OCT images
MEDICAL IMAGE ANALYSIS
IF11.8

