arXiv:2504.04458eess.IVcs.CV2025-04

针对医学图像分割中的类别不平衡问题,提出自适应损失函数CALF提升模型性能。

CALF: A Conditionally Adaptive Loss Function to Mitigate Class-Imbalanced Segmentation

  • 基于偏度与峰度统计量动态评估数据不平衡程度,自适应调整损失函数。
  • 在UPENN-GBM、BraTS等数据集上显著提升分割精度,尤对小区域目标效果明显。
  • 适合处理标注稀缺、罕见病灶的医学图像分割任务,代码开源可复现。

不平衡数据集给医学诊断中深度学习模型的分割任务带来重大挑战,尤其在标注质量有限、罕见病例或兴趣区域(ROIs)尺度较小的情况下。此类情况会干扰模型训练,导致分割边界偏离真实区域。传统损失函数如二元交叉熵会复制标注偏差,限制模型泛化能力。本文提出一种新型、基于统计的条件自适应损失函数(CALF),针对深度学习训练中的不平衡数据集设计。该方法通过偏度与峰度统计量估计不平衡严重程度,进而实施适当变换以平衡数据集,同时保持数据异质性。其整合了预处理、数据筛选与动态损失选择的多阶段流程,实现最优效果。我们在多个大规模开源数据集(UPENN-GBM、UCSF、LGG、BraTS)上进行了定性与定量评估,验证了该方法的有效性,表现出显著的分割改进。代码已公开:https://anonymous.4open.science/r/MICCAI-Submission-43F9/。

原文摘要 · Abstract (English)

Imbalanced datasets pose a considerable challenge in training deep learning (DL) models for medical diagnostics, particularly for segmentation tasks. Imbalance may be associated with annotation quality limited annotated datasets, rare cases, or small-scale regions of interest (ROIs). These conditions adversely affect model training and performance, leading to segmentation boundaries which deviate from the true ROIs. Traditional loss functions, such as Binary Cross Entropy, replicate annotation biases and limit model generalization. We propose a novel, statistically driven, conditionally adaptive loss function (CALF) tailored to accommodate the conditions of imbalanced datasets in DL training. It employs a data-driven methodology by estimating imbalance severity using statistical methods of skewness and kurtosis, then applies an appropriate transformation to balance the training dataset while preserving data heterogeneity. This transformative approach integrates a multifaceted process, encompassing preprocessing, dataset filtering, and dynamic loss selection to achieve optimal outcomes. We benchmark our method against conventional loss functions using qualitative and quantitative evaluations. Experiments using large-scale open-source datasets (i.e., UPENN-GBM, UCSF, LGG, and BraTS) validate our approach, demonstrating substantial segmentation improvements. Code availability: https://anonymous.4open.science/r/MICCAI-Submission-43F9/.

医学分割不平衡学习损失函数

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。