EyeAI

AIM - Training, deploying deep learning models for ophthalmology diagnosis on android app

Goals -

• Train models
• Test models
• Android app framework to deploy model
• On device model prediction
• Capture image / Upload image from device
• Image testing before prediction(Is it a Retina image or not?)
• Report False predictions by model
• Add "Share AI's Report" button
  • Without images
  • With images
• Suggestions to improve app button
• Prediction probabilities
• Choosing model function - Multiple models for different diseases
• Annonymization of files before sending to model
• Image augmentation before making prediction

Frameworks used -

• Model training - Tensorflow lite
• Android deployment - Kotlin and Android Studio

App size - 20 mb Coverage of Android devices - more than 98%

Screenshots

Normal	Diabetic retinopathy	Maculopathy	Retinitis pigmentosa

False image type prediction

CT scans	Xrays	Random faces	Screenshots

Training

Training loss	Training accuracy	Confusion matrix

Maybe sometime in future

• Uploading folder of images at once
• Cloud based interpretation on bigger/better models
• Links to model training process and accuracy statistics
• GradCam/Explainable AI components