SwinECAT提升9类眼底病分类精度,兼顾细节与全局特征。
SwinECAT: A Transformer-based fundus disease classification model with Shifted Window Attention and Efficient Channel Attention
- 结合移位窗口与高效通道注意力,捕捉眼底图像局部与长程依赖。
- 在16,140张图像上达88.29%准确率,宏平均F1达0.90。
- 首次实现9类眼底病高精度分类,适合临床辅助诊断场景。
近年来,人工智能在医学影像领域应用日益广泛。其中,眼底图像分析面临特定挑战,如某些眼底疾病病灶面积小、不同疾病间差异细微,易导致模型预测精度下降和过拟合。为此,本文提出基于Transformer的SwinECAT模型,融合移位窗口(Swin)注意力与高效通道注意力(ECA)机制。SwinECAT利用Swin Transformer主干中的Swin注意力机制,有效捕捉眼底图像内的局部空间结构与长程依赖关系;同时引入轻量级ECA机制,引导模型关注关键特征通道,实现更具区分性的特征表示。相较于以往通常将眼底图像分为4至6类的研究,本工作将分类扩展至9种不同疾病类型,提升诊断粒度。我们在包含16,140张眼底图像的公开数据集Eye Disease Image Dataset (EDID) 上进行评估,实验结果表明,SwinECAT在9类分类任务中达到88.29%准确率,加权F1-score为0.88,宏平均F1-score为0.90。其分类性能显著优于基线Swin Transformer及多个对比模型,是目前在该数据集上9类分类的最高报告表现。
原文摘要 · Abstract (English)
In recent years, artificial intelligence has been increasingly applied in the field of medical imaging. Among these applications, fundus image analysis presents special challenges, including small lesion areas in certain fundus diseases and subtle inter-disease differences, which can lead to reduced prediction accuracy and overfitting in the models. To address these challenges, this paper proposes the Transformer-based model SwinECAT, which combines the Shifted Window (Swin) Attention with the Efficient Channel Attention (ECA) Attention. SwinECAT leverages the Swin Attention mechanism in the Swin Transformer backbone to effectively capture local spatial structures and long-range dependencies within fundus images. The lightweight ECA mechanism is incorporated to guide the SwinECAT's attention toward critical feature channels, enabling more discriminative feature representation. In contrast to previous studies that typically classify fundus images into 4 to 6 categories, this work expands fundus disease classification to 9 distinct types, thereby enhancing the granularity of diagnosis. We evaluate our method on the Eye Disease Image Dataset (EDID) containing 16,140 fundus images for 9-category classification. Experimental results demonstrate that SwinECAT achieves 88.29\% accuracy, with weighted F1-score of 0.88 and macro F1-score of 0.90. The classification results of our proposed model SwinECAT significantly outperform the baseline Swin Transformer and multiple compared baseline models. To our knowledge, this represents the highest reported performance for 9-category classification on this public dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。