arXiv:2412.03629eess.IVcs.CV2024-12被引 4

用扩散模型生成假数据,提升青光眼检测准确率。

DiffuPT: Class Imbalance Mitigation for Glaucoma Detection via Diffusion Based Generation and Model Pretraining

  • 用扩散模型生成青光眼病变图像,缓解数据不平衡问题。
  • 测试集准确率提升至92.59%,比原来高3.5个百分点。
  • 适合医学影像诊断、数据少的疾病检测场景使用。

青光眼是一种进行性视神经病变,表现为视盘结构损伤和视野功能改变。早期检测对防止失明至关重要。然而,医学数据集常存在类别不平衡,导致深度学习模型性能下降。本文提出一种基于生成模型的框架,利用扩散模型生成合成数据以缓解类别不平衡。同时,构建了全国最大规模的青光眼检测数据集支持研究。通过结合扩散生成与模型预训练,提升了分类器训练鲁棒性。实验表明,该方法显著改善分类器的调和平均值(敏感性和特异性)与受试者工作特征曲线下面积(AUC)。在自建国家数据集测试集中,调和平均值从89.09%提升至92.59%。在AIROGS数据集上也获得类似提升。结果表明,基于扩散的生成方法在解决医疗数据不平衡问题、提升诊断性能方面具有重要意义。

原文摘要 · Abstract (English)

Glaucoma is a progressive optic neuropathy characterized by structural damage to the optic nerve head and functional changes in the visual field. Detecting glaucoma early is crucial to preventing loss of eyesight. However, medical datasets often suffer from class imbalances, making detection more difficult for deep-learning algorithms. We use a generative-based framework to enhance glaucoma diagnosis, specifically addressing class imbalance through synthetic data generation. In addition, we collected the largest national dataset for glaucoma detection to support our study. The imbalance between normal and glaucomatous cases leads to performance degradation of classifier models. By combining our proposed framework leveraging diffusion models with a pretraining approach, we created a more robust classifier training process. This training process results in a better-performing classifier. The proposed approach shows promising results in improving the harmonic mean (sensitivity and specificity) and AUC for the roc for the glaucoma classifier. We report an improvement in the harmonic mean metric from 89.09% to 92.59% on the test set of our national dataset. We examine our method against other methods to overcome imbalance through extensive experiments. We report similar improvements on the AIROGS dataset. This study highlights that diffusion-based generation can be of great importance in tackling class imbalances in medical datasets to improve diagnostic performance.

青光眼检测扩散模型数据增强医学影像

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