用生成模型扩充罕见心肌排斥图像,提升检测准确率。
Improving Heart Rejection Detection in XPCI Images Using Synthetic Data Augmentation
- 用StyleGAN生成1万张真实感强的排斥图像,解决数据稀少问题。
- 混合真实与合成数据训练的模型在两类样本上均达高精度和高召回。
- 适合医学影像领域数据稀缺场景,尤其适用于移植排斥检测研究。
准确识别心肌活检中急性细胞排斥(ACR)对心脏移植患者管理至关重要。然而,高分级排斥病例(3R)极为罕见,给深度学习模型训练带来挑战。本文通过使用StyleGAN生成合成数据来缓解类别不平衡问题。在训练GAN前,采用直方图均衡化标准化图像外观,提升组织表征一致性。基于可用的3R活检切片训练StyleGAN,生成10,000张逼真的合成图像。将这些合成图像与真实0R(无排斥)样本以不同组合方式结合,训练ResNet-18二分类器。评估了三种变体:仅用真实0R和合成3R训练、融合合成与额外真实样本、以及仅用真实数据训练。所有模型在独立真实活检集上测试。结果表明,合成数据显著提升分类性能,尤其在与真实样本结合时效果更佳。最优模型在两类样本上均实现高精度与高召回。研究证实混合训练策略的有效性,并凸显基于GAN的数据增强在生物医学图像分析中的潜力,特别是在标注数据受限的领域。
原文摘要 · Abstract (English)
Accurate identification of acute cellular rejection (ACR) in endomyocardial biopsies is essential for effective management of heart transplant patients. However, the rarity of high-grade rejection cases (3R) presents a significant challenge for training robust deep learning models. This work addresses the class imbalance problem by leveraging synthetic data generation using StyleGAN to augment the limited number of real 3R images. Prior to GAN training, histogram equalization was applied to standardize image appearance and improve the consistency of tissue representation. StyleGAN was trained on available 3R biopsy patches and subsequently used to generate 10,000 realistic synthetic images. These were combined with real 0R samples, that is samples without rejection, in various configurations to train ResNet-18 classifiers for binary rejection classification. Three classifier variants were evaluated: one trained on real 0R and synthetic 3R images, another using both synthetic and additional real samples, and a third trained solely on real data. All models were tested on an independent set of real biopsy images. Results demonstrate that synthetic data improves classification performance, particularly when used in combination with real samples. The highest-performing model, which used both real and synthetic images, achieved strong precision and recall for both classes. These findings underscore the value of hybrid training strategies and highlight the potential of GAN-based data augmentation in biomedical image analysis, especially in domains constrained by limited annotated datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。