Skip to main content

ORIGINAL RESEARCH article

Front. Robot. AI
Sec. Biomedical Robotics
Volume 11 - 2024 | doi: 10.3389/frobt.2024.1445565

SwAV-Driven Diagnostics: New Perspectives on Grading Diabetic Retinopathy from Retinal Photography

Provisionally accepted
  • 1 International Islamic University, Chittagong, Chittagong, Chittagong, Bangladesh
  • 2 Daffodil International University, Dhaka, Dhaka, Bangladesh
  • 3 University of Oslo, Oslo, Norway
  • 4 SINTEF, Trondheim, Sør-Trøndelag, Norway

The final, formatted version of the article will be published soon.

    Diabetic Retinopathy (DR) is a serious eye condition that occurs due to high blood sugar levels in patients with Diabetes Mellitus. If left untreated, DR can potentially result in blindness.Using automated neural network-based methods to grade DR shows potential for early detection.However, the uneven and non-quadrilateral forms of DR lesions provide difficulties for traditional Convolutional Neural Network (CNN)-based architectures. To address this challenge and explore a novel algorithm architecture, this work delves into the usage of contrasting cluster assignments in retinal fundus images with the Swapping Assignments between multiple Views (SwAV) algorithm for DR grading. An ablation study was made where SwAV outperformed other CNN and Transformer-based models, independently and in ensemble configurations with an accuracy of 87.00% despite having fewer parameters and layers. The proposed approach outperforms existing state-of-the-art models regarding classification metrics, complexity, and prediction time.The findings offer great potential for medical practitioners, allowing for more accurate diagnosis of DR and earlier treatments to avoid visual loss.

    Keywords: Diabetic Retinopathy, Contrasting Clustering, SwAV, Convolutional Neural Network, ensemble learning, transformer, early diagnosis

    Received: 07 Jun 2024; Accepted: 29 Aug 2024.

    Copyright: © 2024 Ul Alam, Bahadur, Masum, Noori and Uddin. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Farzan Majeed Noori, University of Oslo, Oslo, Norway

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.