Classification of Corneal Pattern Based on Convolutional Neural Network

Document Type : Original Article

Authors

1 Dept. of Communications Faculty of Electronic Engineeing, Menofia University Menouf, Egypt

2 Dep. Of electrical engineering, Faculty of electronic Engineering, Menia University, Egypt.

3 Dep. of Ind. Electronics and Control Eng., Faculty of Electronic Engineering,Menoufia University, Egypt.

4 Dep. Of Computers Engineering. Faculty of Electronic Engineering, Menoufia University, Egypt.

Abstract

The early discovery of the disease is a great achievement in management of the cornea. This paper presents an efficient approach for the classification of normal and abnormal corneal patterns based on deep learning. Convolutional Neural Networks (CNNs) are utilized for this purpose. The CNN model built for this purpose comprises5 layers. The classification process is achieved through two stages. Automatic feature extraction based CNN is applied in the first stage, followed by sequence of processing layers includes: pooling layer, dropout layer and fully connected layer resulted in a diagnosis of the condition of the patient in terms of normal or abnormal. The proposed technique was tested and evaluated based MATLAB environment on a set of corneal images. These images were collected for patients based on confocal microscopy. The CNN classification results on corneal fundus images recorded an accuracy of 100 %.

Keywords



1. Introduction
Cornea is the transparent layer that face the light which enters the eye. Cornea and Sclera form the outer tunic of the eye and are mechanically strong. They protect the eye from any foreign object to enter it. Cornea has 5 main layers, namely: 1) Epithelium 2) Bowman’s layer 3) Stroma 4) Descemet’s membrane and 5) Endothelium. The main function of the endothelium is to pump the excess water out of the stroma to preserve its mechanical structure and optical clarity. In this paper we will use the corneal endothelium patterns for the classifications. The corneal endothelium is an important impact at the human vision because of the directly relationship between the cornea healthy and the endothelium cells [1-3]. M.Tang 2005, V. Girisha 2016, Iman Mohammed 2006 display the corneal disease that can cause some disaster cases such as clouding, distortion or blindness. The major types of the endothelium diseases are the following: 1) Keratoconus is weakening and thinning of the central cornea. 2) Corneal endothelial cells (CECs) are terminally differentiated cells and transparency. 3) Fuchs’ endothelial dystrophy is effect at the inner layer for the cornea which called the endothelium. 4) Bullouskeratopathy is a condition in which the cornea becomes permanently swollen. This occurs because the inner layer of the cornea, the endothelium, has been damaged and can’t pumping fluids out of the tissue, various techniques used at this article such as histogram, enhancement and segmentation for the images for enhance the quality of the images. The results from this classification with accuracy 83% for Fuchs ’dystrophy, 77% for Guttation, 82% for Iridocorneal, 82% for Posterior dystrophy [4-6]. G.Ayala 2001, R.Nadachi 1992, F.Sanchez 1999, the image preprocessing techniques are presented to improve the image quality. Several previous solutions for segmentation of endothelial image have already been introduced. Their aim is to delineate cell borders using such techniques as: Numerical measures for corneal cells using Wavelet transform, Local greyscalethre-sholding followed by scissoring and morphological thinning [7-9]. M. Mahzoun 1996, K. Habrat 2016 and A. piokowski 2017 used some techniques such as active contours, or analysis local pixel levels aimed at finding intensity borders between cells, calculating the corneal cells and their inter relationship, cell density, coefficient of variation, hexagonally, exploits the features that endothelium cells are approximately laid out as a regular tessellation of hexagonal shapes. These techniques estimate the inverse transpose of a matrix generating the cellular lattice [10-12].
1 0
Rajendra Acharya and kannathal N2006presented a comparison between three classifiers for corneal images, artificial neural network ANN with accuracy 89% and fuzzy classifier 92.94 % and adaptive neuro fuzzy classifier 92.94% and they trained 30 image normal and 50 image abnormal Magdalena Habrat, Adam Piorkowski, 2016based on automated analysis of cornea endothelial cell images. It presented a method for corneal endothelial images segmentation.A preprocessing step was done for edge regularization by using filtering technique. The cell’s borders are then determined using the directional filter [13-14]. Alferdo Ruggeri, Enrico Grisan, 2005 and Marco 2002 presented a solution to the problem of automatic estimation of endothelial sell density from corneal images. They proposed methods that the spatial frequency existed in digital endothelium images is extracted with a 2-D discrete fourier transform (DFT) technique. The cell density is related to a circular band in the DFT of the images which contain the frequency information and the goal of Gavet article,2008 to compute the borders of the cornea cells based on NN technique. Greyscale specular microscopic images of corneal endothelial cells are used for evaluating this algorithm. The disadvantage of the neural network technique is the use of fixed-size mask which needs to expert to perform the necessary corrections while Gavet article compute the borders of the cornea cells automatically [15-17]. Enrico Grisan and Anna Paviotti2005 proposed a new method to calculate endothelium cell density which is one of the main indicators of cornea health state and quality. This article exploits the property that endothelium cells have approximately a regular tessellation of hexagonal shapes. This approach evaluates the inverse transpose of a matrix creating this cellular lattice and then from this cellular lattice the cell density can be easily obtained. This information could be easily extracted if the cell contours are well defined in the image. There are other methods to classify the corneal images based on the corneal curvature [18-20]. Fabijańska 2019, introduced an approach to study the corneal health status based efficient automatic segmentation of corneal endothelial images. They used a combination of neural network to determine the pixel location at the cell boundaries with post processing stage to obtain the edginess map. They exploited morphological operations with local thresholding for images segmentation. They achieved acceptable results compared with ground truths [21]. Table 1 shows a comparison between different classification techniques according the accuracy:
Table 1: Comparison between different classification techniques Classification methods accuracy
ANN [13]
89%
Fuzzy classifier [13]
92.94
Neuro Fuzzy classifier[13]
92.94
Automated neural networks model [22]
80%
Segmentation Cell density [18]
90%
NN[16]
90%
This paper gives an efficient approach for automatic corneal image classification based on deep learning. The objective is to detect anomalies in corneal patterns for efficient diagnosis of different diseases. Also, the grading of diseases is of large importance for patients and technicians.
The paper is organized as follows. Section 2 gives the traditional approach based on automatic feature extraction. Section3 shows the proposed approach based on deep learning which estimated by equations matrices in section 4.Section 5 presents the simulation results. In section 6, the concluding remarks are spotted.
2. Approach Based on Automatic Feature Extraction
In the traditional method for identifying the corneal disease, an image enhancement of fundus images is applied to enhance contrast for better representation of objects. Fig.1 shows samples of normal and abnormal corneal images. Fig.2 displays a simplified diagram of traditional method to classify the corneal images. Fig.3 shows a block diagram of the proposed CNN system to classify between the normal and abnormal images. The CNN is a deep neural network for image recognition. CNN includes the feature extractor in the training process. The convolutional layers followed as an activation function, pooling layers, dropout layers and fully connected layers. The convolutional layer is contained filters that are used to perform a two dimensional (2D) convolution with the input image. Feature maps are generated from the convolution layer. The pooling layer decreases the size of the image. The used value is the maximum or mean value of the pixels. The max-pooling layer is used in this technique. The inclusion of a dropout layer is used a regularization technique for reducing over fitting [22-24].
1 1
Fig.1Sample frames of corneal images. a, b are normal images. c,d are abnormal images [5].
Fig.2 Block diagram of traditional method for identification of corneal status [5].
Fig.3 Block diagram of the proposed system
Fig.4 Layers of the Proposed Deep Learning Model.
Conv.
Max. Pooling
Conv.
Max. Pooling
Conv.
Max. Poolig
Conv.
Max. Pooling
Conv.
Max. Pooling
Fully Connected
Dense
1 2
3. Proposed Approach Based on Deep Learning This approach is based on building an efficient deep learning model that can distinguish between both normal and abnormal states of person’s cornea. The proposed model consists of 5 Conv. layers followed by 5 max pooling layers. Finally a global average pooling is used. Table 1 shows the model summary for each layer and its output shape. Images are input in 224 ×224. Layers have number filters of 16, 32, 64, 128 and 256 for layers 1, 2, 3, 4, 5 respectively. Finally two dense layer will be used. Figure 4 illustrates the steps of the proposed model. This model is implemented for 50 corneal images dataset. The dataset is divided into 80% training and 20% for testing. The proposed convolutional neural network (CNN) is summarized in table 2. This table shows the output of each layer of the proposed deep learning CNN model. Table 2: Classifications of Proposed Deep Learning Model.
Layer Type
Output Shape
CNV
(222, 222, 16)
Pooling
(111, 111, 16)
CNV
(109, 109, 32)
Pooling
(54, 54, 32)
CNV
(52, 52, 64)
Pooling
(26, 26, 64)
CNV
(24, 24, 128)
Pooling
(12, 12, 128)
CNV
(10, 10, 256)
Pooling
(5, 5, 256)
Global Average Layer
(256)
Dense
(2)
4. Evaluation metrics:
A. In this proposed approach, accuracy and cross entropy are used to estimate the strength of the CNN model. Accuracy knowing as recognition rate is calculated as follows:
 Loss (cross entropy)% = Log (estimated) – Log (target)
4. Results
The proposed approach is implemented for the 50 corneal images and the accuracy and loss are estimated as the following: figure5 and 6 show the accuracy and loss during the training phase of the benchmark model using one Max pooling layer, figures 7 and 8 show the accuracy and loss during the training phase. It can be observed that the accuracy reached 100 % and the loss decreased near to be zero.
Fig.5. Resulting accuracy of the benchmark model.
Fig.6. Resulting loss of the benchmark model.
Fig.7. Resulting accuracy of the proposed deep learning model.
1 3
Fig.8. Resulting loss of the proposed deep learning model. Table 3 shows the difference between ANN, SVM and the proposed solution 6. Conclusion This paper represented an automatic feature extraction and classification of corneal images based on deep learning using convolutional neural networks. The proposed approach presents high performance classification of normal and abnormal states of cornea. The evaluation results reveal that the proposed approach realizes an accuracy of 100 % and zero % loss.

[1] M. Balasubramian and A. Louise and Roger W“Fractal Dimension Based Corneal Fungal Diagnosis”,Proceedings of the SPIE, Volume 6312, pp. 200-211, 2006. [2]Anna Fabijan “Corneal Endothelium Image Segmentation Using Feedforward Neural Network”,proceeding of the Federated conference on computer science and information systems, pp. 629-637, IEEE Cataloge Number, 2017. [3] Curry Bucht, “Fully Automated Corneal Endothelial MorphometricOf Images Captured By Clinical Specular Microscopy”, Courant Institute of Mathematical Sciences, pp. 125-1362009. [4] Maolong Tang, Raj Shekhar, And David Huang “ Curvature Mapping For Detection Of Corneal Shape Abnormality”, IEEE Transactions On Medical Imageing, pp. 16-25, March 2005. [5] V. Girisha and K.Vijay Chandra“Machine Intelligence On Confocal Microscope For Detection Of Endothelium Layer Of Corneal Disease”, Proceeding of IJCTER,pp. 148-211Auguest 2016. [6] Iman Mohammed Rezazadeh and Keivan Maghooli “Investigating And Validation Numerical Measures Of Corneal Topographic Data Is Lasik Surgery Outcome Using Wavelet Transform”, Proceeding ofIEEE, pp. 153-159,2006. [7] G.Ayala, M. Diaz And L.Martines-Costa “Granulometric Moments And Corneal Endothelium Status”, Pattern Recognition , pp.1219-12222001. [8] R. Nadachi “Automated Corneal Endothelial Cell Analysis In Fifth Annual”, IEEE Symposium On Computer Based Medical Systems, ppt 14-17, 1992. [9] F. Sanchez“ Automatic Segmentation Of Contours Of Corneal Cells”, Computers in Biology and Medicine, pp. 243-155, 1999.10] M. Mahzoun, K. Okazaki, H.Kawai, Y.Sato, S.Tamura, And K.Kani “ Detection And Complment Of Hexagonal Borders In Corneal Endothelial Cell Image”, Medical Imaging Technology, pp. 56-65,1996. [11] K.Habrat, M. Habrat“ Cell Detection In Corneal Endothelial Images Using Directional Filters”, In Image Processing And Communications Challenges ,pp. 113-123, 2016. [12] A. Piorkowski, K. Nurzynska, J. Gronkowska-Serafin, B.Selig, C.Boldak, And D. Reska “ Influence Of Applied Corneal Endothelium Image Segmentation Technigues On The Clinical Parameters, “ Comp. Med. Imag. Grap, pp. 13-27, 2017. [13] Acharya, U Rajendra “Computer-Based Classification Of Eye Diseases”, Proceeding of the28th IEEE EMBS annual international conferenece, New York city, USA , Aug 30-Sept 3,2006. [14] M. Habrat, A. Piorkowski “Cell detection in corneal Endothelial images using directional filters”, October`, pp. 113-123, 2016. [15] Alferdo Ruggeri And Enrico Grisan “Anew System For The Automatic Estimation Of Endothelial Cell Density In Donor Corneas”, Br J Ophthalmol, Ppt 306-311, 2005. [16] Marco “Estimating Cell Density In Corneal Endothelium By Means Of Fourier Analysis”, Proceeding of IEEE, ppt 23-26, 2002. [17] Gavet“Visual Perception Based Automatic Recognition Of Cell Mosaics In Human Corneal Endothelium Microscopy Images”, Image Anal Stereol, pp. 53-62, 2008. [18] Enrico Grisan, Anna Paviotti “A lattice estimation approach for the automatic evaluation of corneal endothelium density” Proceeding of IEEE, ppt 150-168, 2005. [19] G.W.Griffiths “Analysis of cornea curvature using radial basis functions-Part1-Methadology”, Computers in Biologyand Medicine, Elsevier, pp.272-284, 2016. [20] D.klyce “Computer Assisted corneal topography”, Avro-journal, pp. 210-230, 2018. [21] A. Fabijanska, " Automatic segmentation of corneal endothelial cells from microscopyimages", Biomedical Signal Processing and Control, Vol. 47, pp. 145-158, 2019. [22] LeCun, and Y Bottou: ''Gradient-based learning applied to document recognition'', Proceedings of the IEEE, pp. 482-456, 2008. [23] Ranzato, M. A “Unsupervised learning of invariant feature hierarchies with applications to object recognition”, IEEE Computer Vision and Pattern Recognition, pp. 845-850,2007. [24] Srivastava And N. HintonDropout“A Simple Way To Prevent Neural Networks From Over Fitting'', The Journal Of Machine Learning Research, Pp. 1929-1958, 2014. [25] Shen, Yi. Loss functions for binary classification and class probability estimation. Diss. University of Pennsylvania, 2005.