Optical Coherence Tomography Image Classification Using Hybrid Deep Learning and Ant Colony Optimizationopen access
- Authors
- Khan, Awais; Pin, Kuntha; Aziz, Ahsan; Han, Jung Woo; Nam, Yunyoung
- Issue Date
- Aug-2023
- Publisher
- MDPI
- Keywords
- optical coherence tomography; deep learning; convolutional neural network; feature selection; ant colony optimization; machine learning; age-related macular degeneration; branch retinal vein occlusion; central retinal vein occlusion; central serous chorioretinopathy; diabetic macular edema
- Citation
- SENSORS, v.23, no.15
- Journal Title
- SENSORS
- Volume
- 23
- Number
- 15
- URI
- https://scholarworks.bwise.kr/sch/handle/2021.sw.sch/25383
- DOI
- 10.3390/s23156706
- ISSN
- 1424-8220
1424-3210
- Abstract
- Optical coherence tomography (OCT) is widely used to detect and classify retinal diseases. However, OCT-image-based manual detection by ophthalmologists is prone to errors and subjectivity. Thus, various automation methods have been proposed; however, improvements in detection accuracy are required. Particularly, automated techniques using deep learning on OCT images are being developed to detect various retinal disorders at an early stage. Here, we propose a deep learning-based automatic method for detecting and classifying retinal diseases using OCT images. The diseases include age-related macular degeneration, branch retinal vein occlusion, central retinal vein occlusion, central serous chorioretinopathy, and diabetic macular edema. The proposed method comprises four main steps: three pretrained models, DenseNet-201, InceptionV3, and ResNet-50, are first modified according to the nature of the dataset, after which the features are extracted via transfer learning. The extracted features are improved, and the best features are selected using ant colony optimization. Finally, the best features are passed to the k-nearest neighbors and support vector machine algorithms for final classification. The proposed method, evaluated using OCT retinal images collected from Soonchunhyang University Bucheon Hospital, demonstrates an accuracy of 99.1% with the incorporation of ACO. Without ACO, the accuracy achieved is 97.4%. Furthermore, the proposed method exhibits state-of-the-art performance and outperforms existing techniques in terms of accuracy.
- Files in This Item
- There are no files associated with this item.
- Appears in
Collections - College of Engineering > Department of Computer Science and Engineering > 1. Journal Articles
- College of Medicine > Department of Ophthalmology > 1. Journal Articles
![qrcode](https://api.qrserver.com/v1/create-qr-code/?size=55x55&data=https://scholarworks.bwise.kr/sch/handle/2021.sw.sch/25383)
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.