arXiv:2502.17154cs.CVcs.AI2025-02被引 23

轻量级视觉变换器助力眼底图早期青光眼诊断,准确率达92%

MaxGlaViT: A novel lightweight vision transformer-based approach for early diagnosis of glaucoma stages from fundus images

  • 重构MaxViT结构并优化通道与块数,实现轻量化设计
  • 引入ECA注意力机制和ConvNeXtV2模块,准确率提升至92.03%
  • 适合医学影像分析、资源受限场景下的青光眼筛查应用

青光眼进展隐匿,早期发现对防止失明至关重要。本文提出MaxGlaViT,一种基于重构多轴视觉变换器(MaxViT)的轻量级模型,用于眼底图像中青光眼阶段的早期检测。通过优化网络深度与通道数、在主干中加入CBAM、ECA、SE注意力机制、替换原块中的MBConv为ConvNeXt、ConvNeXtV2和InceptionNeXt结构,逐步提升性能。在HDV1数据集上评估,相比40种CNN和40种ViT模型,最终模型达到92.03%准确率,92.33%精确率,92.03%召回率,92.13%F1分数,87.12%科恩κ系数,显著优于现有方法。

原文摘要 · Abstract (English)

Glaucoma is a prevalent eye disease that progresses silently without symptoms. If not detected and treated early, it can cause permanent vision loss. Computer-assisted diagnosis systems play a crucial role in timely and efficient identification. This study introduces MaxGlaViT, a lightweight model based on the restructured Multi-Axis Vision Transformer (MaxViT) for early glaucoma detection. First, MaxViT was scaled to optimize block and channel numbers, resulting in a lighter architecture. Second, the stem was enhanced by adding attention mechanisms (CBAM, ECA, SE) after convolution layers to improve feature learning. Third, MBConv structures in MaxViT blocks were replaced by advanced DL blocks (ConvNeXt, ConvNeXtV2, InceptionNeXt). The model was evaluated using the HDV1 dataset, containing fundus images of different glaucoma stages. Additionally, 40 CNN and 40 ViT models were tested on HDV1 to validate MaxGlaViT's efficiency. Among CNN models, EfficientB6 achieved the highest accuracy (84.91%), while among ViT models, MaxViT-Tiny performed best (86.42%). The scaled MaxViT reached 87.93% accuracy. Adding ECA to the stem block increased accuracy to 89.01%. Replacing MBConv with ConvNeXtV2 further improved it to 89.87%. Finally, integrating ECA in the stem and ConvNeXtV2 in MaxViT blocks resulted in 92.03% accuracy. Testing 80 DL models for glaucoma stage classification, this study presents a comprehensive and comparative analysis. MaxGlaViT outperforms experimental and state-of-the-art models, achieving 92.03% accuracy, 92.33% precision, 92.03% recall, 92.13% f1-score, and 87.12% Cohen's kappa score.

青光眼诊断视觉Transformer轻量模型眼底图像

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