arXiv:2508.03758eess.IVcs.CV2025-08被引 2

融合视觉变压器与U-Net,实现糖尿病足溃疡精准分割与可解释可视化

TransUNet-GradCAM: A Hybrid Transformer-U-Net with Self-Attention and Explainable Visualizations for Foot Ulcer Segmentation

  • 用混合架构结合局部细节与全局上下文特征
  • 内部验证Dice达0.8886,跨数据集零样本迁移仍保持0.78以上
  • 支持临床实用,预测面积与真实值相关性高达0.975

自动化分割糖尿病足溃疡在临床诊断、治疗规划和伤口长期监测中至关重要。由于溃疡在临床图像中呈现异质性外观、不规则形态及复杂背景,该任务仍具挑战性。传统卷积神经网络(如U-Net)虽具备强定位能力,但受限于感受野,难以建模长程空间依赖。为此,本文采用TransUNet架构,将视觉变压器(ViT)的全局注意力机制融入U-Net结构,兼顾全局上下文特征提取与细粒度空间分辨率。模型在公开的足溃疡分割挑战(FUSeg)数据集上训练,采用鲁棒的数据增强和混合损失函数缓解类别不平衡。在内部验证集上,优化阈值0.4843下达到Dice相似系数0.8886。为评估泛化能力,在两个独立外部数据集(AZH伤口中心:n=278;Medetec:n=152)上进行零样本验证,未重新训练即分别获得Dice分数0.6209和0.7850,体现强大跨域迁移能力。临床效用分析显示,预测面积与真实面积呈高度相关(皮尔逊相关系数r=0.9749)。结果表明,该方法有效融合全局与局部特征,提供可靠、高效且可解释的足溃疡自动评估方案。

原文摘要 · Abstract (English)

Automated segmentation of diabetic foot ulcers (DFUs) plays a critical role in clinical diagnosis, therapeutic planning, and longitudinal wound monitoring. However, this task remains challenging due to the heterogeneous appearance, irregular morphology, and complex backgrounds associated with ulcer regions in clinical photographs. Traditional convolutional neural networks (CNNs), such as U-Net, provide strong localization capabilities but struggle to model long-range spatial dependencies due to their inherently limited receptive fields. To address this, we employ the TransUNet architecture, a hybrid framework that integrates the global attention mechanism of Vision Transformers (ViTs) into the U-Net structure. This combination allows the model to extract global contextual features while maintaining fine-grained spatial resolution. We trained the model on the public Foot Ulcer Segmentation Challenge (FUSeg) dataset using a robust augmentation pipeline and a hybrid loss function to mitigate class imbalance. On the internal validation set, the model achieved a Dice Similarity Coefficient (F1-score) of 0.8886 using an optimized threshold of 0.4843. Crucially, to assess generalizability, we performed external validation on two independent datasets: the AZH Wound Care Center dataset (n=278) and the Medetec dataset (n=152). Without any retraining, the model achieved Dice scores of 0.6209 and 0.7850, respectively, demonstrating robust zero-shot transferability to unseen clinical domains. Furthermore, clinical utility analysis revealed a strong correlation (Pearson r = 0.9749) between predicted and ground-truth wound areas. These outcomes demonstrate that our approach effectively integrates global and local feature extraction, offering a reliable, effective, and explainable solution for automated foot ulcer assessment.

医学图像分割Transformer可解释性糖尿病足

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。