arXiv:2502.05459cs.CVcs.AI2025-02被引 9

用集成卷积网络提升白细胞分类准确率并实现可解释性分析

DCENWCNet: A Deep CNN Ensemble Network for White Blood Cell Classification with LIME-Based Explainability

  • 三模型集成,通过不同丢弃率与池化配置增强特征学习
  • 在Rabbin-WBC数据集上达到最高平均准确率及各项指标最优
  • 结合LIME技术实现预测过程可视化,便于医生理解与信任

白血细胞是免疫系统的重要组成部分,其数量和类型能反映健康状况。传统卷积神经网络(CNN)可用于血细胞分类与目标识别,但现有模型常因随意设计导致冗余层、数据不平衡及数据增强不足。为此,本文提出一种新型集成方法——DCENWCNet,融合三个不同配置的CNN架构,分别采用差异化的丢弃率与最大池化层设置,以优化特征学习并平衡偏差-方差权衡。在广泛使用的Rabbin-WBC数据集上,该模型在所有类别中均取得最高的平均准确率,并在精确率、召回率、F1分数和受试者工作特征曲线下面积(AUC)方面表现优异。为进一步提升可解释性,引入局部可解释模型无关解释(LIME)技术,通过分析特征值与预测结果间的关系,揭示黑箱模型决策逻辑,使用户能够理解并验证诊断结果,增强对自动化诊断系统的信任。

原文摘要 · Abstract (English)

White blood cells (WBC) are important parts of our immune system, and they protect our body against infections by eliminating viruses, bacteria, parasites and fungi. The number of WBC types and the total number of WBCs provide important information about our health status. A traditional method, convolutional neural networks (CNN), a deep learning architecture, can classify the blood cell from a part of an object and perform object recognition. Various CNN models exhibit potential; however, their development often involves ad-hoc processes that neglect unnecessary layers, leading to issues with unbalanced datasets and insufficient data augmentation. To address these challenges, we propose a novel ensemble approach that integrates three CNN architectures, each uniquely configured with different dropout and max-pooling layer settings to enhance feature learning. This ensemble model, named DCENWCNet, effectively balances the bias-variance trade-off. When evaluated on the widely recognized Rabbin-WBC dataset, our model outperforms existing state-of-the-art networks, achieving highest mean accuracy. Additionally, it demonstrates superior performance in precision, recall, F1-score, and Area Under the ROC Curve (AUC) across all categories. To delve deeper into the interpretability of classifiers, we employ reliable post-hoc explanation techniques, including Local Interpretable Model-Agnostic Explanations (LIME). These methods approximate the behavior of a black-box model by elucidating the relationships between feature values and predictions. Interpretable results enable users to comprehend and validate the model's predictions, thereby increasing their confidence in the automated diagnosis.

白细胞分类深度学习可解释性集成学习

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