用视觉变压器提升伤口组织多分类分割精度。
WoundFormer: Multi-Scale Spatial Feature Fusion for Multi-Class Wound Tissue Segmentation

- 设计空间保持的多尺度融合头,增强跨尺度特征整合。
- 在147张图像上达81.9%整体Dice分数,领先基线4.3点。
- 适合需精准识别多种伤口组织的临床辅助诊断场景。
慢性伤口如糖尿病足溃疡和压疮需要精确的组织级评估以指导治疗并监测愈合进程。尽管深度学习已推动自动化伤口分析发展,但现有方法多聚焦于二分类,难以建模组织成分的高度异质性,受限于类内差异大和标注数据少。多类别伤口组织分割仍具挑战且临床意义重大。本文提出WoundFormer,一种基于变压器的框架,通过改进层次化空间特征融合来提升多类别伤口组织分割性能。具体地,将标准SegFormer解码器替换为保持空间拓扑的多尺度聚合头,在跨尺度融合中保留特征结构,并通过卷积融合强化上下文交互。该设计提升了边界定位精度,增强了视觉相似组织间的区分能力,同时维持了变压器效率。在包含147张图像、六类组织的WoundTissueSeg数据集及另一基准DFUTissue数据集上评估,所提方法整体Dice得分为81.9%,较强的CNN与变压器基线最高提升4.3个Dice点,且在少数类别上也保持稳定提升。结果表明,显式建模层次化空间交互可增强变压器对异质性伤口组织的表征能力,支持更可靠的定量伤口评估。
原文摘要 · Abstract (English)
Chronic wounds such as diabetic foot ulcers and pressure injuries require accurate tissue-level assessment to guide treatment planning and monitor healing progression. While deep learning methods have advanced automated wound analysis, most existing approaches focus on binary segmentation and inadequately model heterogeneous tissue composition due to high intra-class variability and limited annotated data. Multi-class wound tissue segmentation, therefore, remains a challenging and clinically relevant problem. We propose WoundFormer, a transformer-based framework that enhances hierarchical spatial feature fusion for multi-class wound tissue segmentation. Specifically, we replace the standard SegFormer decoder with a spatially-preserving multi-scale aggregation head that maintains feature topology during cross-scale integration and strengthens contextual interactions through convolutional fusion. This design improves boundary localization and discrimination between visually similar tissue categories while preserving transformer efficiency. We evaluate WoundFormer on the WoundTissueSeg dataset (147 images, six tissue classes) and a second benchmark (DFUTissue dataset). The proposed method achieves an overall Dice score of 81.9%, outperforming strong CNN- and transformer-based baselines by up to 4.3 Dice points on the WoundTissueSeg benchmark, with consistent improvements across minority tissue classes. These results indicate that explicit modeling of hierarchical spatial interactions enhances transformer representations for heterogeneous wound tissue segmentation and supports more reliable quantitative wound assessment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。