mariotiara / rd-detection-ensemble-cnn

Ensemble of 3 convolutional neural network model with different convolutional network backbone (DensNet201, InceptionV3 and MobileNetV2) for diabetic retinopathy stage detection

Retinopathy Diabetic (RD) Classifier - Ensemble CNN

Detect the stage of diabetic retinopathy in human retina through fundus images. The dataset that used is from 4th Asia Pacific Tele-Ophthalmology Society (APTOS) in Kaggle . The total of images we used are 8000 images, which is 6000 for training, 1500 as data validation and 500 for testing. We used several proprocessing method for getting hight performance of our model. Our final model is an ensemble of 3 convolutional neural network model with different convolutional network backbone (DensNet201, InceptionV3 and MobileNetV2). Complete project with a simple GUI made with python and PyQt5. Image below show our project framework looks like.

Experimental Results

Backbone performance in SGD and Adam Optimizer with learning rate variance

Best accuracy of DensNet201 is 0.94 using Adam optimizer with 1.125e-5 lr, InceptionV3 is 0.93 using Adam with 1.25e-5 lr and MobileNet is 0.9 using Adam and 2.5e-5 lr.

Comparing Single model with Ensemble model in different input

We evaluate our model with 5 input variances, original and 4 variances of K of K-meanse clustering.

Detail Performance of Ensemble Model Using Confusion Matrix

Demo

The Research paper "NON-PROLIFERATIF RETINOPATHY CLASSIFICATION USING ENSEMBLE CONVOLUTIONAL NEURAL NETWORK" is publised in Pseudoce jurnal Univerity of Bengkulu paper link