新数据集+82种模型评估,找到最优伤口组织分割方法
Deep Learning for Wound Tissue Segmentation: A Comprehensive Evaluation using A Novel Dataset
- 构建147张图像的六类组织标注数据集,支持三种标注格式
- FPN+VGG16在三种格式中表现最佳,最高达82.25%的Dice分数
- 公开数据集与完整评测框架,助力临床研究与模型开发
深度学习技术在医学伤口组织分割中展现出巨大潜力,但缺乏公开标注数据集和统一评估标准。本研究通过构建一个包含147张伤口图像的新数据集,涵盖六类组织(坏死、肉芽、渗出、水肿、骨、肌腱),并采用监督学习进行语义分割标注,提供全图、图像块和超像素三种标注格式。共评估82种分割模型,涵盖UNet、cGAN、FPN+VGG16等深度学习方法及基于深度特征的机器学习分类器(如AlexNet+RF)。结果表明,FPN+VGG16在各类别上表现最优,加权平均Dice得分达82.25%。本研究为伤口图像分割与分类提供了可复现的基准,所用数据集已开源。
原文摘要 · Abstract (English)
Deep learning (DL) techniques have emerged as promising solutions for medical wound tissue segmentation. However, a notable limitation in this field is the lack of publicly available labelled datasets and a standardised performance evaluation of state-of-the-art DL models on such datasets. This study addresses this gap by comprehensively evaluating various DL models for wound tissue segmentation using a novel dataset. We have curated a dataset comprising 147 wound images exhibiting six tissue types: slough, granulation, maceration, necrosis, bone, and tendon. The dataset was meticulously labelled for semantic segmentation employing supervised machine learning techniques. Three distinct labelling formats were developed -- full image, patch, and superpixel. Our investigation encompassed a wide array of DL segmentation and classification methodologies, ranging from conventional approaches like UNet, to generative adversarial networks such as cGAN, and modified techniques like FPN+VGG16. Also, we explored DL-based classification methods (e.g., ResNet50) and machine learning-based classification leveraging DL features (e.g., AlexNet+RF). In total, 82 wound tissue segmentation models were derived across the three labelling formats. Our analysis yielded several notable findings, including identifying optimal DL models for each labelling format based on weighted average Dice or F1 scores. Notably, FPN+VGG16 emerged as the top-performing DL model for wound tissue segmentation, achieving a dice score of 82.25%. This study provides a valuable benchmark for evaluating wound image segmentation and classification models, offering insights to inform future research and clinical practice in wound care. The labelled dataset created in this study is available at https://github.com/akabircs/WoundTissue.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。