用注意力机制增强深度特征,提升青光眼检测准确率与可解释性。
Attention-Enhanced Deep Features with Heterogeneous Ensemble Learning for Glaucoma Detection

- 引入CBAM模块优化InceptionV3提取的视网膜特征。
- 融合SMOTE+TL处理数据不平衡,双层集成提升分类鲁棒性。
- 模型聚焦临床相关区域,适合医疗AI研发与临床辅助诊断。
青光眼是一种进行性视神经病变,导致不可逆视力损伤,早期诊断对防止失明至关重要。尽管深度学习在自动青光眼检测中表现良好,但现有方法常忽视特征精炼、存在类别不平衡问题,且依赖单一分类器限制预测稳定性。本文提出一种混合检测框架,结合注意力增强的深度特征提取与异构集成学习。首先使用InceptionV3提取深度特征,并通过卷积块注意力模块(CBAM)增强其判别能力;随后采用多种机器学习模型,配合单层(SLE)与双层(DLE)集成策略提升分类鲁棒性;同时利用SMOTE结合Tomek Links(SMOTE+TL)缓解类别不平衡。在两个公开视网膜眼底图像数据集上的实验表明,基于深度特征的方法持续优于手工特征方法,所提注意力增强框架整体性能最优。Grad-CAM可视化显示模型聚焦于临床相关视网膜区域,为预测过程提供可解释证据。
原文摘要 · Abstract (English)
Glaucoma is a progressive optic neuropathy characterized by irreversible damage to the optic nerve, making timely diagnosis critical to prevent permanent vision loss. Although deep learning has demonstrated promising performance in automated glaucoma detection, existing approaches often overlook feature refinement, suffer from class imbalance, and rely on individual classifiers that limit prediction robustness. To address these challenges, this paper proposes a hybrid glaucoma detection framework that integrates attention-enhanced deep feature extraction with heterogeneous ensemble learning. Specifically, deep representations are extracted using InceptionV3 and subsequently refined by incorporating the Convolutional Block Attention Module (CBAM) to enhance discriminative retinal features. To improve classification robustness, the extracted features are classified using multiple machine learning models together with Single-Level Ensemble (SLE) and Double-Level Ensemble (DLE) strategies, while SMOTE combined with Tomek Links (SMOTE+TL) is employed to alleviate class imbalance. Furthermore, a systematic comparison of handcrafted, deep, and attention-enhanced deep feature representations is conducted. Experimental evaluation on two public retinal fundus datasets demonstrates that deep feature-based methods consistently outperform handcrafted feature-based methods, while the proposed attention-enhanced framework achieves the best overall performance. Furthermore, Grad-CAM visualizations confirm that the proposed model focuses on clinically relevant retinal regions, providing interpretable evidence on the model's prediction process.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。