arXiv:2410.08797cs.CV2024-10被引 34

用混合模型提升白血病细胞识别准确率,解决数据不平衡与特征难辨问题。

CoTCoNet: An Optimized Coupled Transformer-Convolutional Network with an Adaptive Graph Reconstruction for Leukemia Detection

  • 融合Transformer与卷积网络,捕捉全局与局部细胞特征。
  • 引入图重构模块和生成器,提升对细微生物特征的识别与数据均衡性。
  • 在4个公开数据集上超越现有方法,支持可解释性可视化分析。

快速准确的血涂片分析是诊断白血病及其他血液恶性肿瘤的有效手段。然而,传统显微镜下的人工白细胞计数与形态学评估耗时且易出错;常规图像处理方法因恶性与良性细胞形态高度相似而难以区分,且训练数据分布不均进一步阻碍可靠特征提取。针对上述挑战,本文提出优化的耦合式变压器-卷积网络(CoTCoNet),通过深度融合变压器与深度卷积网络,有效捕获全局特征与可扩展的空间模式,实现复杂大规模血液学特征的识别。框架还引入基于图的特征重构模块,揭示白细胞中隐藏或难以观察的生物学特征,并采用基于种群的元启发式算法进行特征选择与优化。为缓解数据不平衡问题,使用合成白细胞生成器扩充数据。在包含16,982个标注细胞的数据集上,CoTCoNet达到0.9894的准确率和0.9893的F1分数。为进一步验证泛化能力,模型在四个公开多样性数据集上测试,表现优于当前最先进方法。同时,通过与细胞标注对齐的特征可视化方法增强可解释性,深化对模型决策的理解。

原文摘要 · Abstract (English)

Swift and accurate blood smear analysis is an effective diagnostic method for leukemia and other hematological malignancies. However, manual leukocyte count and morphological evaluation using a microscope is time-consuming and prone to errors. Conventional image processing methods also exhibit limitations in differentiating cells due to the visual similarity between malignant and benign cell morphology. This limitation is further compounded by the skewed training data that hinders the extraction of reliable and pertinent features. In response to these challenges, we propose an optimized Coupled Transformer Convolutional Network (CoTCoNet) framework for the classification of leukemia, which employs a well-designed transformer integrated with a deep convolutional network to effectively capture comprehensive global features and scalable spatial patterns, enabling the identification of complex and large-scale hematological features. Further, the framework incorporates a graph-based feature reconstruction module to reveal the hidden or unobserved hard-to-see biological features of leukocyte cells and employs a Population-based Meta-Heuristic Algorithm for feature selection and optimization. To mitigate data imbalance issues, we employ a synthetic leukocyte generator. In the evaluation phase, we initially assess CoTCoNet on a dataset containing 16,982 annotated cells, and it achieves remarkable accuracy and F1-Score rates of 0.9894 and 0.9893, respectively. To broaden the generalizability of our model, we evaluate it across four publicly available diverse datasets, which include the aforementioned dataset. This evaluation demonstrates that our method outperforms current state-of-the-art approaches. We also incorporate an explainability approach in the form of feature visualization closely aligned with cell annotations to provide a deeper understanding of the framework.

白血病检测图像识别Transformer数据均衡

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