用熵引导边界监督提升乳腺超声分割的特异性和概率可靠性。
Specificity- and Calibration-Aware Breast Ultrasound Segmentation via Entropy-Guided Boundary Supervision

- 根据预测熵和真实边界图加权边界损失,聚焦网络不确定区域。
- 无病灶图像中假阳性从19/20降至5/20,显著降低误报。
- 结合推理阶段温度缩放,提升概率校准性,适合临床部署。
乳腺超声病变分割面临两个相关挑战:含病变图像中,斑点噪声、低组织对比度和后方声影导致边界泄漏和轮廓不完整;无病变图像中,这些伪影会引发类似实性病变区域的假阳性激活。本文通过修改训练目标,统一解决两类问题:不等权重地处理边界像素,而是根据每个像素的预测熵和真实边界图对轮廓惩罚进行加权,使梯度更集中于网络仍不确定的病变边缘位置。在BUSI数据集上评估,与无边界监督和均匀加权边界交叉熵两种基线相比,在97张含病变测试图像上,平均Dice分数分别为0.7624和0.7616(配对威尔科克斯检验p=0.27),表明分割质量保持不变。主要改进体现在特异性:在20张无病变图像上,假阳性从14/20和19/20降至5/20(麦克尼马尔检验p=0.012和0.0005),置信区间无重叠,差异显著且具实际意义。后续引入后处理空间温度缩放,将期望校准误差从0.0201降至0.0095,未改变分割掩膜。熵引导边界监督与空间校准形成互补,分别在训练和推理阶段提升特异性和概率可靠性,适用于U-Net框架。
原文摘要 · Abstract (English)
Lesion segmentation in breast ultrasound involves two related challenges. In images with lesions, speckle noise, low tissue contrast, and posterior acoustic shadowing cause boundary leakage and incomplete contour delineation. In images without lesions, those same artifacts generate false-positive activations in regions resembling solid lesion tissue. This study addresses both failure modes through a single modification to the training objective. Rather than weighting every boundary pixel equally, the proposed loss scales contour penalties by per-pixel predictive entropy and the ground-truth boundary map, concentrating gradient emphasis on lesion margin locations where the network remains uncertain. The loss was evaluated on the BUSI dataset through a controlled ablation against two baselines: a model without boundary supervision and a model with uniformly weighted boundary binary cross-entropy. Across 97 lesion-containing test images, mean Dice scores were statistically indistinguishable between the proposed method and the no-boundary baseline (0.7624 versus 0.7616, paired Wilcoxon p = 0.27), confirming that lesion segmentation quality is preserved. The primary effect appears in specificity. False-positive activations on 20 no-lesion test images fell from 14 of 20 and 19 of 20 for the two baselines to 5 of 20 with the proposed approach (McNemar p = 0.012 and 0.0005). Non-overlapping Wilson 95% confidence intervals confirm the difference is both statistically significant and practically substantial. A post-hoc spatial temperature scaling step further reduced expected calibration error from 0.0201 to 0.0095 without altering segmentation masks. Entropy-guided boundary supervision and spatial calibration thus function as complementary training-level and inference-level refinements that improve specificity and probability reliability within a U-Net framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。