A Tailored Complex Medical Decision Analysis Model for Diabetic Retinopathy Classification Based on Optimized Un-Supervised Feature Learning Approach

Manaswini Jena,Debahuti Mishra,Smita Prava Mishra,Pradeep Kumar Mallick
DOI: https://doi.org/10.1007/s13369-022-07057-0
IF: 2.807
2022-08-04
Arabian Journal for Science and Engineering
Abstract:Diabetic Retinopathy has become a major medical concern all over the world. The only way to reduce the chances of vision loss due to diabetic retinopathy is through timely treatment. The increasing prevalence of this disease leads us toward the development of an automatic identification tool. Various machine learning algorithms have been introduced to solve different types of problems using deep architecture models and showed very impressive performance. But, the deep architectures like Convolution Neural Network (CNN) suffers from overfitting problem while dealing with small datasets. In this paper, a remedy to this problem is attempted for classification using CNN by making the feature learning and classification tasks separated. Feature learning is a major part of any classification model. Classification models depending upon error gradient may have limitations like vanishing gradient problems. For this, an independent non-hand crafted unsupervised feature learning method is chosen based on unsupervised clustering. The idea behind the proposed approach is that if a feature performs well in unsupervised clustering, it will perform well in the supervised scenario as well. Based on this idea this work proposes a feature extraction model that updates its parameters based on the Normalized Mutual Information performance of k -means clustering. The parameters of the feature extractor are updated by the nature-inspired Grey Wolf Optimization algorithm. The experiments show that the learned features can be useful in several classifiers. Individual classifiers such as k- NN, ELM, Kernel-ELM, SVM and ANN are used as classifiers for classification and finally, an ensemble of the classifiers is applied at the end which improves the performance.
multidisciplinary sciences
What problem does this paper attempt to address?