arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.08982cs.CV
收藏

血细胞形态的深度生成式分类

Deep Generative Classification of Blood Cell Morphology

University of Cambridge(剑桥大学);UZ Leuven(鲁汶大学医院);University of Leuven(鲁汶大学);KU Leuven(鲁汶大学);Zuyderland Medical Center(宰德兰医学中心);Barts Health NHS Trust(巴茨健康国民保健服务信托);University College London(伦敦大学学院);Queen Mary University of London(伦敦玛丽女王大学)
Simon Deltadahl, Julian Gilbey, Christine Van Laer, Nancy Boeckx, Mathie Leers, Tanya Freeman, Laura Aiken, Timothy Farren, Matthew Smith, Mohamad Zeina, BloodC… 展开作者

Simon Deltadahl, Julian Gilbey, Christine Van Laer, Nancy Boeckx, Mathie Leers, Tanya Freeman, Laura Aiken, Timothy Farren, Matthew Smith, Mohamad Zeina, BloodCounts consortium, James HF Rudd, Concetta Piazzese, Joseph Taylor, Nicholas Gleadall, Carola-Bibiane Schönlieb, Suthesh Sivapalaratnam, Michael Roberts, Parashkev Nachev

AI总结:

本文提出基于扩散模型的CytoDiffusion血细胞分类器,在异常检测、抗分布偏移和低数据场景下均优于现有判别模型,并具备生成逼真图像和可解释反事实热图的能力。

摘要:

血液细胞的准确分类对于诊断血液疾病至关重要,但由于细胞形态的复杂性、生物学、病理学和成像特征的异质性以及细胞类型频率的不平衡,机器自动化面临重大挑战。我们引入了CytoDiffusion,这是一种基于扩散的分类器,能够有效建模血细胞形态,将准确分类与稳健的异常检测、对分布偏移的抵抗能力、可解释性、数据效率以及超越人类的 uncertainty quantification 相结合。我们的方法在异常检测(AUC 0.990 对比 0.918)、对领域偏移的抵抗能力(平衡准确率 85.85% 对比 74.38%)以及低数据场景下的表现(平衡准确率 95.88% 对比 94.95%)方面均优于最先进的判别模型。值得注意的是,我们的模型生成的合成血细胞图像几乎与真实图像无法区分,这一点通过一项真实性测试得到证明:在该测试中,血液学专家在区分真实图像与生成图像时仅达到 52.3% 的准确率(95% CI: [50.5%, 54.2%])。此外,我们通过生成可直接解释的反事实热图来增强模型的可解释性。我们的综合评估框架涵盖这些多个性能维度,为血液学中的医学图像分析建立了新的基准,最终能够在临床环境中提高诊断准确性。我们的代码可在 https://github.com/CambridgeCIA/CytoDiffusion 获取。

英文摘要:

Accurate classification of haematological cells is critical for diagnosing blood disorders, but presents significant challenges for machine automation owing to the complexity of cell morphology, heterogeneities of biological, pathological, and imaging characteristics, and the imbalance of cell type frequencies. We introduce CytoDiffusion, a diffusion-based classifier that effectively models blood cell morphology, combining accurate classification with robust anomaly detection, resistance to distributional shifts, interpretability, data efficiency, and superhuman uncertainty quantification. Our approach outperforms state-of-the-art discriminative models in anomaly detection (AUC 0.990 vs. 0.918), resistance to domain shifts (85.85% vs. 74.38% balanced accuracy), and performance in low-data regimes (95.88% vs. 94.95% balanced accuracy). Notably, our model generates synthetic blood cell images that are nearly indistinguishable from real images, as demonstrated by an authenticity test in which expert haematologists achieved only 52.3% accuracy (95% CI: [50.5%, 54.2%]) in distinguishing real from generated images. Furthermore, we enhance model explainability through the generation of directly interpretable counterfactual heatmaps. Our comprehensive evaluation framework, encompassing these multiple performance dimensions, establishes a new benchmark for medical image analysis in haematology, ultimately enabling improved diagnostic accuracy in clinical settings. Our code is available at https://github.com/CambridgeCIA/CytoDiffusion.

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

↑