用可调控金字塔动态加权,提升细小结构分割精度。
Steerable Pyramid Weighted Loss: Multi-Scale Adaptive Weighting for Semantic Segmentation
- 基于可调控金字塔生成多尺度自适应权重图。
- 在三个数据集上达到更高像素精确率,计算开销极低。
- 适合需要精细分割的医学图像和遥感场景。
语义分割是计算机视觉的核心任务,广泛应用于生物医学成像、遥感和自动驾驶。尽管交叉熵和Dice损失在一般情况下表现良好,但在处理细小结构或紧密排列物体时仍存在困难。已有方法通过基于距离变换的权重图提高误分类像素的权重,但此类方法计算成本高且无法随网络预测动态调整。本文提出一种基于可调控金字塔的加权(SPW)损失函数,能高效生成自适应权重图。与依赖静态或迭代更新距离图的传统边界感知损失不同,本方法利用可调控金字塔在多个频率带中动态强调不同尺度的特征,同时保持计算效率。通过将网络预测融入权重计算,实现训练过程中的自适应优化。我们在SNEMI3D、GlaS和DRIVE数据集上评估该方法,对比11种先进损失函数,结果表明所提SPW损失在像素精度和分割准确率上均更优,且计算开销极小。本工作为提升语义分割性能,特别是多尺度特征表示的应用,提供了一种高效有效的解决方案。代码已公开于https://anonymous.4open.science/r/SPW-0884。
原文摘要 · Abstract (English)
Semantic segmentation is a core task in computer vision with applications in biomedical imaging, remote sensing, and autonomous driving. While standard loss functions such as cross-entropy and Dice loss perform well in general cases, they often struggle with fine structures, particularly in tasks involving thin structures or closely packed objects. Various weight map-based loss functions have been proposed to address this issue by assigning higher loss weights to pixels prone to misclassification. However, these methods typically rely on precomputed or runtime-generated weight maps based on distance transforms, which impose significant computational costs and fail to adapt to evolving network predictions. In this paper, we propose a novel steerable pyramid-based weighted (SPW) loss function that efficiently generates adaptive weight maps. Unlike traditional boundary-aware losses that depend on static or iteratively updated distance maps, our method leverages steerable pyramids to dynamically emphasize regions across multiple frequency bands (capturing features at different scales) while maintaining computational efficiency. Additionally, by incorporating network predictions into the weight computation, our approach enables adaptive refinement during training. We evaluate our method on the SNEMI3D, GlaS, and DRIVE datasets, benchmarking it against 11 state-of-the-art loss functions. Our results demonstrate that the proposed SPW loss function achieves superior pixel precision and segmentation accuracy with minimal computational overhead. This work provides an effective and efficient solution for improving semantic segmentation, particularly for applications requiring multiscale feature representation. The code is avaiable at https://anonymous.4open.science/r/SPW-0884
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。