arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.02456cs.CVeess.IV
收藏

利用数据增强技术克服深度学习伤口分类中数据稀缺问题的研究

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

BASIS Independent Fremont(贝赛思独立弗里蒙特校区);AIClub Research Institute(AI俱乐部研究院)

Harini Narayanan, Sindhu Ghanta

AI总结:

本研究探索几何变换与GAN等数据增强技术,以克服伤口分类中数据稀缺问题,在先进模型基础上将F1分数提升达11%,并验证DE-GANs生成多样伤口图像的可行性。

摘要:

慢性伤口对个人和医疗系统构成重大负担,影响数百万人并产生高昂费用。使用深度学习技术的伤口分类是实现更快诊断和启动治疗的一种有前景的方法。然而,缺乏高质量数据来训练机器学习模型,是发挥机器学习在伤口护理中潜力的主要挑战。事实上,数据限制是当今使用医学或法医影像的研究中最大的挑战。我们研究了可用于克服数据稀缺限制并释放基于深度学习解决方案潜力的数据增强技术。在我们的研究中,我们探索了一系列数据增强技术,从伤口图像的几何变换到先进的生成对抗网络(GANs),以丰富和扩展数据集。使用Keras、Tensorflow和Pandas库,我们实现了能够生成逼真伤口图像的数据增强技术。我们表明,几何数据增强可以在最先进模型的基础上,将多个关键伤口类别的分类性能(F1分数)提高多达11%。我们基于GAN的增强实验证明了使用DE-GANs生成具有更丰富变异的伤口图像的可行性。我们的研究和结果表明,数据增强是一种有价值的隐私保护工具,具有克服数据稀缺限制的巨大潜力,我们相信它将成为任何现实世界中基于机器学习的伤口护理系统的一部分。

英文摘要:

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分析而来,可能存在错误,请以论文正文为准。

↑