arXiv:2512.24214cs.CVcs.LG2025-12被引 1

用生成模型合成数据并优化网络,提升新冠影像分类准确率。

Medical Image Classification on Imbalanced Data Using ProGAN and SMA-Optimized ResNet: Application to COVID-19

  • 用ProGAN生成合成医学图像,缓解数据不平衡问题。
  • 在4类和2类分类任务中分别达到95.5%和98.5%准确率。
  • 适合处理疫情下样本稀缺的医疗图像分类场景。

医疗图像分类中的数据不平衡问题尤为突出,尤其在疫情期间,某类疾病(如新冠)的影像样本远少于其他类别。尽管人工智能和机器学习方法被广泛用于快速、准确检测感染者,但缺乏充足且平衡的数据仍是主要障碍。本研究提出一种渐进式生成对抗网络(ProGAN)来生成合成图像,补充真实数据,并采用加权融合策略将合成与真实数据结合后输入深度分类器。同时,引入多目标元启发式种群优化算法,自动优化分类器超参数。在大规模、不平衡的新冠胸部X光图像数据集上,该模型在4类和2类分类任务中分别达到95.5%和98.5%的准确率,显著优于现有方法。实验结果验证了该模型在疫情背景下处理不平衡医疗图像分类的有效性。

原文摘要 · Abstract (English)

The challenge of imbalanced data is prominent in medical image classification. This challenge arises when there is a significant disparity in the number of images belonging to a particular class, such as the presence or absence of a specific disease, as compared to the number of images belonging to other classes. This issue is especially notable during pandemics, which may result in an even more significant imbalance in the dataset. Researchers have employed various approaches in recent years to detect COVID-19 infected individuals accurately and quickly, with artificial intelligence and machine learning algorithms at the forefront. However, the lack of sufficient and balanced data remains a significant obstacle to these methods. This study addresses the challenge by proposing a progressive generative adversarial network to generate synthetic data to supplement the real ones. The proposed method suggests a weighted approach to combine synthetic data with real ones before inputting it into a deep network classifier. A multi-objective meta-heuristic population-based optimization algorithm is employed to optimize the hyper-parameters of the classifier. The proposed model exhibits superior cross-validated metrics compared to existing methods when applied to a large and imbalanced chest X-ray image dataset of COVID-19. The proposed model achieves 95.5% and 98.5% accuracy for 4-class and 2-class imbalanced classification problems, respectively. The successful experimental outcomes demonstrate the effectiveness of the proposed model in classifying medical images using imbalanced data during pandemics.

医学影像数据不平衡生成对抗网络新冠分类

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