RetinaLiteNet: a lightweight transformer based CNN for retinal feature segmentation

Mehwish Mehmood*, Majed Alsharari, Shahzaib Iqbal, Ivor Spence, Muhammad Fahim

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Retinal image analysis plays a pivotal role in diagnosing diseases like glaucoma, diabetic retinopathy, neurodegenerative disorders, and cardiovascular diseases. The recent advancement of artificial intelligence (AI) can assist the practitioners to analyze the images accurately. In this research, a lightweight deep learning model is proposed which is based on multitask learning to segment the retinal images including retinal vessels and optic disc for further analysis by clinicians. The proposed model has encoder-decoder framework, where the encoder has convolutional layers with multi-head attention that captures both local details and long-range dependencies effectively. The resulting features from convolutional layers and multi-head attention are fused together to make the model more efficient and resilient for segmentation tasks. To further refine the features, the skip connections are implemented along with the convolutional block attention module (CBAM) in the decoder. The model's efficiency is validated on two publicly available datasets (i.e., IOSTAR and DRIVE) to confirm the lightweight aspects and robustness. It achieved the dice scores of 80.6% and 93.3% on DRIVE and 80.1% and 85.4% on IOSTAR dataset for simultaneous segmentation of blood vessels and optic disc, respectively. The empirical evaluations show 0.25 MB of memory, 0.066 million parameters, and a FLOPs estimation of 2.46 GFLOPs, which is better than existing models.
Original languageEnglish
Title of host publicationProceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2024: Women in Computer Vision workshop
PublisherInstitute of Electrical and Electronics Engineers Inc.
Publication statusAccepted - 06 Apr 2024
EventThe IEEE/CVF Conference on Computer Vision and Pattern Recognition 2024 - Seattle, United States
Duration: 17 Jun 202421 Jun 2024

Publication series

NameIEEE/CVF Conference on Computer Vision and Pattern Recognition: Proceedings
ISSN (Print)1063-6919
ISSN (Electronic)2575-7075

Conference

ConferenceThe IEEE/CVF Conference on Computer Vision and Pattern Recognition 2024
Country/TerritoryUnited States
CitySeattle
Period17/06/202421/06/2024

Fingerprint

Dive into the research topics of 'RetinaLiteNet: a lightweight transformer based CNN for retinal feature segmentation'. Together they form a unique fingerprint.

Cite this