arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2309.12245eess.IVcs.CVcs.LG

自适应输入图像归一化用于解决基于GAN的X射线图像中的模式崩溃问题

Adaptive Input-image Normalization for Solving the Mode Collapse Problem in GAN-based X-ray Images

Department of Computer Science, Munster Technological University, Cork, Ireland(计算机科学系,穆斯廷技术大学,科克,爱尔兰)

Muhammad Muneeb Saad, Mubashir Husain Rehmani, Ruairi O'Reilly

AI总结:

本文提出通过自适应输入图像归一化技术缓解GAN生成X射线图像中模式崩溃问题,提升图像多样性和分类性能。

摘要:

生物医学图像数据集可能由于目标疾病罕见而失衡。生成对抗网络在解决这种失衡方面发挥着关键作用,通过生成合成图像来增强数据集。生成具有广泛特征范围的合成图像以准确表示训练图像中存在的特征分布非常重要。此外,合成图像中缺乏多样化的特征会降低机器学习分类器的性能。模式崩溃问题影响了生成对抗网络生成多样化图像的能力。模式崩溃有两种类型:类内和类间。在本文中,研究了这两种模式崩溃问题,并评估了它们对合成X射线图像多样性的后续影响。这项工作贡献了将自适应输入图像归一化与深度卷积GAN和辅助分类器GAN结合的实证演示,以缓解模式崩溃问题。合成生成的图像用于数据增强和训练视觉转换器模型。通过准确率、召回率和精确率分数评估模型的分类性能。结果表明,带有自适应输入图像归一化的DCGAN和ACGAN在多样性和分类分数上优于未归一化的X射线图像的DCGAN和ACGAN。

英文摘要:

Biomedical image datasets can be imbalanced due to the rarity of targeted diseases. Generative Adversarial Networks play a key role in addressing this imbalance by enabling the generation of synthetic images to augment datasets. It is important to generate synthetic images that incorporate a diverse range of features to accurately represent the distribution of features present in the training imagery. Furthermore, the absence of diverse features in synthetic images can degrade the performance of machine learning classifiers. The mode collapse problem impacts Generative Adversarial Networks' capacity to generate diversified images. Mode collapse comes in two varieties: intra-class and inter-class. In this paper, both varieties of the mode collapse problem are investigated, and their subsequent impact on the diversity of synthetic X-ray images is evaluated. This work contributes an empirical demonstration of the benefits of integrating the adaptive input-image normalization with the Deep Convolutional GAN and Auxiliary Classifier GAN to alleviate the mode collapse problems. Synthetically generated images are utilized for data augmentation and training a Vision Transformer model. The classification performance of the model is evaluated using accuracy, recall, and precision scores. Results demonstrate that the DCGAN and the ACGAN with adaptive input-image normalization outperform the DCGAN and ACGAN with un-normalized X-ray images as evidenced by the superior diversity scores and classification scores.

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

↑