提出乘法损失函数,提升医学细胞图像分割在数据稀缺下的表现。
Multiplicative Loss for Enhancing Semantic Segmentation in Medical and Cellular Images
- 用乘法组合交叉熵与Dice损失,动态调节梯度以稳定优化。
- 在有限数据下显著提升分割性能,优于传统加法组合与现有方法。
- 无需调参,适合数据稀少的医学图像分析场景。
我们提出了两种新颖的损失函数——乘法损失(Multiplicative Loss)和置信度自适应乘法损失(Confidence-Adaptive Multiplicative Loss),用于医学与细胞图像的语义分割。尽管交叉熵(Cross Entropy)和Dice Loss被广泛使用,但其加法组合对超参数敏感且在数据有限时表现不佳。医学图像因隐私、伦理及标注成本高而存在数据稀缺问题,亟需鲁棒高效的训练目标。我们的乘法损失将交叉熵与Dice Loss相乘,根据预测置信度动态调节梯度:对高置信度正确预测降低惩罚,对错误的高置信度预测增强梯度,从而稳定优化过程。在此基础上,置信度自适应乘法损失借鉴Focal Loss的思想,引入基于置信度的指数缩放机制,结合预测概率与Dice系数,强化困难样本的学习。在细胞与医学分割基准测试中,该框架持续优于经过调优的加法组合及其他现有损失函数,提供了一种简单、有效且无需调参的鲁棒分割方案,适用于极端数据稀缺场景。
原文摘要 · Abstract (English)
We propose two novel loss functions, Multiplicative Loss and Confidence-Adaptive Multiplicative Loss, for semantic segmentation in medical and cellular images. Although Cross Entropy and Dice Loss are widely used, their additive combination is sensitive to hyperparameters and often performs suboptimally, especially with limited data. Medical images suffer from data scarcity due to privacy, ethics, and costly annotations, requiring robust and efficient training objectives. Our Multiplicative Loss combines Cross Entropy and Dice losses multiplicatively, dynamically modulating gradients based on prediction confidence. This reduces penalties for confident correct predictions and amplifies gradients for incorrect overconfident ones, stabilizing optimization. Building on this, Confidence-Adaptive Multiplicative Loss applies a confidence-driven exponential scaling inspired by Focal Loss, integrating predicted probabilities and Dice coefficients to emphasize difficult samples. This enhances learning under extreme data scarcity by strengthening gradients when confidence is low. Experiments on cellular and medical segmentation benchmarks show our framework consistently outperforms tuned additive and existing loss functions, offering a simple, effective, and hyperparameter-free mechanism for robust segmentation under challenging data limitations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。