1. Transfer Learning in Keratoconus Classification.
- Author
-
Aatila, Mustapha, Lachgar, Mohamed, Hrimech, Hamid, and Kartit, Ali
- Subjects
KERATOCONUS ,CONVOLUTIONAL neural networks - Abstract
Early detection of keratoconus will provide more treatment choices, avoid heavy treatments, and help stop the rapid progression of the disease. This study presents a machine learning-based keratoconus classification approach, using transfer learning, applied on corneal topographic images. Classi- fication is performed considering the three corneal classes namely: normal, suspicious and keratoconus. Keratoconus classification is carried out using transfer learning, by the adoption of six different pretrained convolutional neural networks (CNN) VGG16, InceptionV3, MobileNet, DenseNet201, Xception and EfficientNetB0, which already have knowledge from solving previous specific problems. Each of these different classifiers is trained individually on five different datasets, generated from an original dataset of 2924 corneal topographic images. Original corneal topographic images have been subjected to a special preprocessing before their use by different models in the learning phase. Images of corneal maps are separated in five different datasets while removing noise and textual annotation from images. Most of models used in the classification allow good discrimination between normal cornea, suspicious and keratoconus one. Obtained results reached classification accuracy of 99.31% and 98.51% by DenseNet201 and VGG16 respectively. Obtained results indicate that transfer learning technique could well improve performance of keratoconus classification systems. [ABSTRACT FROM AUTHOR]
- Published
- 2022
- Full Text
- View/download PDF