针对白内障手术镜头分割,提出自适应置信度损失提升边界精度。
Adaptive Confidence-Wise Loss for Improved Lens Structure Segmentation in AS-OCT
- 按专家标注置信度分组,动态调整损失权重。
- 在U-Net上实现6.13%的IoU提升和4.79%的边界校准误差降低。
- 适合医学图像分割中需要高精度边界的场景。
精准的晶状体结构分割对白内障手术人工晶体设计至关重要。现有深度分割网络通常对所有像素使用相等权重的交叉熵损失,忽略了晶状体子区域的不均匀性(如某些区域表现更优)以及边界区域常出现的像素级分割校准不足问题。临床上,专家在标注时会根据子区域比例、模糊边界和形状等因素赋予不同置信度。受此启发,本文提出自适应置信度损失(ACW),通过唯一区域特性设定置信阈值,将每个晶状体子区域划分为低置信与高置信组,并应用区域加权损失重新分配权重。此外,设计自适应置信阈值优化算法以动态调节阈值。为更准确量化边界区域的校准误差,提出新指标——边界期望校准误差(BECE)。在临床晶状体结构AS-OCT数据集及其他多结构数据集上的实验表明,该方法在多种主流分割网络(如MedSAM)上显著优于对比方法。尤其在U-Net上,相较传统交叉熵损失,本方法实现6.13%的IoU提升、4.33%的DSC增加以及4.79%的BECE下降。代码已开源:https://github.com/XiaoLing12138/Adaptive-Confidence-Wise-Loss。
原文摘要 · Abstract (English)
Precise lens structure segmentation is essential for the design of intraocular lenses (IOLs) in cataract surgery. Existing deep segmentation networks typically weight all pixels equally under cross-entropy (CE) loss, overlooking the fact that sub-regions of lens structures are inhomogeneous (e.g., some regions perform better than others) and that boundary regions often suffer from poor segmentation calibration at the pixel level. Clinically, experts annotate different sub-regions of lens structures with varying confidence levels, considering factors such as sub-region proportions, ambiguous boundaries, and lens structure shapes. Motivated by this observation, we propose an Adaptive Confidence-Wise (ACW) loss to group each lens structure sub-region into different confidence sub-regions via a confidence threshold from the unique region aspect, aiming to exploit the potential of expert annotation confidence prior. Specifically, ACW clusters each target region into low-confidence and high-confidence groups and then applies a region-weighted loss to reweigh each confidence group. Moreover, we design an adaptive confidence threshold optimization algorithm to adjust the confidence threshold of ACW dynamically. Additionally, to better quantify the miscalibration errors in boundary region segmentation, we propose a new metric, termed Boundary Expected Calibration Error (BECE). Extensive experiments on a clinical lens structure AS-OCT dataset and other multi-structure datasets demonstrate that our ACW significantly outperforms competitive segmentation loss methods across different deep segmentation networks (e.g., MedSAM). Notably, our method surpasses CE with 6.13% IoU gain, 4.33% DSC increase, and 4.79% BECE reduction in lens structure segmentation under U-Net. The code of this paper is available at https://github.com/XiaoLing12138/Adaptive-Confidence-Wise-Loss.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。