arXiv:2411.02456cs.CVeess.IV2024-11被引 1

通过数据增强提升慢性伤口分类模型性能,解决医疗数据稀缺难题。

A Study of Data Augmentation Techniques to Overcome Data Scarcity in Wound Classification using Deep Learning

  • 采用几何变换与DE-GAN生成真实伤口图像以扩充数据集。
  • 几何增强使关键类别分类F1分数最高提升11%。
  • 成果适合医疗AI开发者及临床研究者参考应用。

慢性伤口对个人和医疗系统构成重大负担,影响数百万人并带来高昂成本。基于深度学习的伤口分类可加速诊断与治疗启动,但高质量训练数据匮乏是实现机器学习潜力的主要障碍。本文研究多种数据增强技术,涵盖伤口图像的几何变换与先进GAN生成,以缓解数据稀缺问题。利用Keras、TensorFlow和Pandas库,我们实现了能生成逼真伤口图像的数据增强方法。实验表明,几何增强可在多个关键伤口类别上将分类性能(F1分数)提升高达11%,超越现有最优模型。基于GAN的增强验证了DE-GAN在生成更具多样性伤口图像方面的可行性。研究结果证明,数据增强是一种隐私保护性强、潜力巨大的工具,有望成为真实世界医疗机器学习系统的重要组成部分。

原文摘要 · Abstract (English)

Chronic wounds are a significant burden on individuals and the healthcare system, affecting millions of people and incurring high costs. Wound classification using deep learning techniques is a promising approach for faster diagnosis and treatment initiation. However, lack of high quality data to train the ML models is a major challenge to realize the potential of ML in wound care. In fact, data limitations are the biggest challenge in studies using medical or forensic imaging today. We study data augmentation techniques that can be used to overcome the data scarcity limitations and unlock the potential of deep learning based solutions. In our study we explore a range of data augmentation techniques from geometric transformations of wound images to advanced GANs, to enrich and expand datasets. Using the Keras, Tensorflow, and Pandas libraries, we implemented the data augmentation techniques that can generate realistic wound images. We show that geometric data augmentation can improve classification performance, F1 scores, by up to 11% on top of state-of-the-art models, across several key classes of wounds. Our experiments with GAN based augmentation prove the viability of using DE-GANs to generate wound images with richer variations. Our study and results show that data augmentation is a valuable privacy-preserving tool with huge potential to overcome the data scarcity limitations and we believe it will be part of any real-world ML-based wound care system.

数据增强伤口分类深度学习医疗AI

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