GS-Net: Global Self-Attention Guided CNN for Multi-Stage Glaucoma Classification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Das, Dipankar, Nayak, Deepak Ranjan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929512896266240
author Das, Dipankar
Nayak, Deepak Ranjan
author_facet Das, Dipankar
Nayak, Deepak Ranjan
contents Glaucoma is a common eye disease that leads to irreversible blindness unless timely detected. Hence, glaucoma detection at an early stage is of utmost importance for a better treatment plan and ultimately saving the vision. The recent literature has shown the prominence of CNN-based methods to detect glaucoma from retinal fundus images. However, such methods mainly focus on solving binary classification tasks and have not been thoroughly explored for the detection of different glaucoma stages, which is relatively challenging due to minute lesion size variations and high inter-class similarities. This paper proposes a global self-attention based network called GS-Net for efficient multi-stage glaucoma classification. We introduce a global self-attention module (GSAM) consisting of two parallel attention modules, a channel attention module (CAM) and a spatial attention module (SAM), to learn global feature dependencies across channel and spatial dimensions. The GSAM encourages extracting more discriminative and class-specific features from the fundus images. The experimental results on a publicly available dataset demonstrate that our GS-Net outperforms state-of-the-art methods. Also, the GSAM achieves competitive performance against popular attention modules.
format Preprint
id arxiv_https___arxiv_org_abs_2409_16082
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GS-Net: Global Self-Attention Guided CNN for Multi-Stage Glaucoma Classification
Das, Dipankar
Nayak, Deepak Ranjan
Computer Vision and Pattern Recognition
Glaucoma is a common eye disease that leads to irreversible blindness unless timely detected. Hence, glaucoma detection at an early stage is of utmost importance for a better treatment plan and ultimately saving the vision. The recent literature has shown the prominence of CNN-based methods to detect glaucoma from retinal fundus images. However, such methods mainly focus on solving binary classification tasks and have not been thoroughly explored for the detection of different glaucoma stages, which is relatively challenging due to minute lesion size variations and high inter-class similarities. This paper proposes a global self-attention based network called GS-Net for efficient multi-stage glaucoma classification. We introduce a global self-attention module (GSAM) consisting of two parallel attention modules, a channel attention module (CAM) and a spatial attention module (SAM), to learn global feature dependencies across channel and spatial dimensions. The GSAM encourages extracting more discriminative and class-specific features from the fundus images. The experimental results on a publicly available dataset demonstrate that our GS-Net outperforms state-of-the-art methods. Also, the GSAM achieves competitive performance against popular attention modules.
title GS-Net: Global Self-Attention Guided CNN for Multi-Stage Glaucoma Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.16082