改进多类别分割的损失函数,让小类和罕见类别获得更公平的训练信号。
Instance Awareness of Multi-class Semantic Segmentation Loss Functions

- 将单类敏感损失扩展到多类,通过逐类分解平衡类别间差异
- 多类CC损失提升前景Dice达0.64,罕见类Dice显著改善
- 在每个组件内局部加权,避免全局权重失衡,提升训练稳定性
针对多类别语义分割中的类别不平衡问题,本文将仅适用于单类的实例敏感损失(如CC损失、blob损失)通过一对多分类分解方式拓展至多类场景。该方法通过类别均匀平均,使每个类别无论实例多少均贡献相等梯度。进一步地,将反尺寸加权机制限定在每个组件的局部空间上下文中,避免全局加权导致的训练不稳定。在BraTS-METS 2025数据集(260个测试样本)上,多类CC损失使前景Dice提升至0.64±0.26(基线0.59±0.27),显著改善罕见类别表现;多类blob损失在阈值0.5下实现最佳全景质量(0.40±0.24,基线0.38±0.25)与识别质量(0.53±0.29,基线0.49±0.30)。将反尺寸加权引入组件内后,罕见类Dice达0.44±0.36,但检测质量略有下降。
原文摘要 · Abstract (English)
Instance-sensitive losses for semantic segmentation such as blob loss and CC loss were designed to address instance imbalance, ensuring small lesions generate the same gradient as large ones, but operate only on single-class segmentation. In multi-class settings, class imbalance poses an additional problem: rare classes with few instances receive a disproportionately small share of the training signal. We show that extending instance-sensitive losses to multi-class segmentation via a one-vs-rest class decomposition repurposes them to also address class imbalance, as uniform averaging over classes ensures each class contributes equally regardless of frequency. We further show that inverse-size weighting, which destabilizes training when applied globally due to weight imbalances across rare and common classes, becomes effective when integrated within the per-component loss, confining the reweighting to each component's spatial context. On the BraTS-METS 2025 dataset (260 test cases), multi-class CC loss improves foreground Dice (0.64 +/- 0.26 vs. 0.59 +/- 0.27 baseline) and rare-class Dice, while maintaining Panoptic Quality at DSC threshold 0.5. Multi-class blob loss achieves the best Panoptic Quality at threshold 0.5 (0.40 +/- 0.24 vs. 0.38 +/- 0.25 baseline) and recognition quality (0.53 +/- 0.29 vs. 0.49 +/- 0.30). Integrating inverse-size weighting within the per-component loss increases rare-class Dice to 0.44 +/- 0.36 at the cost of reduced detection quality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。