arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2301.04802cs.LGcs.CVeess.IV
收藏

用于皮肤病分类的基于扩散的数据增强:对原始医学数据集和全合成图像的影响

Diffusion-based Data Augmentation for Skin Disease Classification: Impact Across Original Medical Datasets to Fully Synthetic Images

AIP Labs(AIP实验室);Semmelweis University(塞麦尔维斯大学)

Mohamed Akrout, Bálint Gyepesi, Péter Holló, Adrienn Poór, Blága Kincső, Stephen Solis, Katrina Cirone, Jeremy Kawahara, Dekker Slade, Latif Abid, Máté Kovács, István Fazekas

摘要:

尽管近年来不断进步,但深度神经网络仍然依赖于大量的训练数据来避免过拟合。然而,由于长期的隐私和严格的数据共享策略,用于诸如医疗保健的真实世界应用的标记的训练数据是有限的并且难以访问。通过在像素或特征空间中操作图像数据集,现有的数据增强技术代表了提高训练数据的数量和多样性的有效途径之一。在这里,我们希望通过建立在文本到图像扩散概率模型在增强我们的宏观皮肤病数据集的训练样本方面的新兴成功来推进增强技术。我们通过输入文本提示来实现对图像生成过程的细粒度控制。我们证明了这种生成数据增强方法即使在完全合成的皮肤病数据集上训练时,也成功地保持了视觉分类器的类似分类精度。与生成模型的最新应用类似,我们的研究表明,扩散模型在生成不牺牲分类器性能的高质量皮肤图像方面确实有效,并且可以在策展后改善训练数据集的增强。

英文摘要:

Despite continued advancement in recent years, deep neural networks still rely on large amounts of training data to avoid overfitting. However, labeled training data for real-world applications such as healthcare is limited and difficult to access given longstanding privacy, and strict data sharing policies. By manipulating image datasets in the pixel or feature space, existing data augmentation techniques represent one of the effective ways to improve the quantity and diversity of training data. Here, we look to advance augmentation techniques by building upon the emerging success of text-to-image diffusion probabilistic models in augmenting the training samples of our macroscopic skin disease dataset. We do so by enabling fine-grained control of the image generation process via input text prompts. We demonstrate that this generative data augmentation approach successfully maintains a similar classification accuracy of the visual classifier even when trained on a fully synthetic skin disease dataset. Similar to recent applications of generative models, our study suggests that diffusion models are indeed effective in generating high-quality skin images that do not sacrifice the classifier performance, and can improve the augmentation of training datasets after curation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑