Lightweight CNN Feature Extraction and PSO-Weighted Ensemble for Retinal OCT Classification
DOI:
https://doi.org/10.29207/resti.v10i3.7451Keywords:
ensemble learning, lightweight CNN, OCT, PSO, retinal diseases classificationAbstract
Retinal diseases are a primary cause of visual impairment, requiring accurate and efficient automated classification. This study proposes a multi-class classification approach for detecting retinal disease from Optical Coherence Tomography (OCT) images using the OCT-C8 dataset, which consists of 24,000 images across eight balanced classes. Pretrained lightweight convolutional neural networks (EfficientNet-B0, ShuffleNetV2, and RegNetY-400MF) are used as feature extractors to leverage transfer learning while reducing computational cost. Instead of end-to-end deep learning, extracted features are classified using traditional machine learning models (KNN, RF, SVM, and MLP), which are more efficient for moderate-sized datasets. To improve robustness, a weighted ensemble is applied, where classifier contributions are optimized using Particle Swarm Optimization (PSO). Experimental results show that the proposed method achieves a classification accuracy of 97.35%, outperforming conventional hard and soft voting methods, while maintaining a balance between computational efficiency and performance with potential for practical deployment.
Downloads
References
S. R. Flaxman et al., “Global causes of blindness and distance vision impairment 1990–2020: a systematic review and meta-analysis,” Lancet Global Health, vol. 5, no. 12, pp. e1221–e1234, 2017, doi: 10.1016/S2214-109X(17)30393-5.
G. F. Attizzani, L. Patrício, and H. G. Bezerra, “Optical coherence tomography assessment of calcified plaque modification after rotational atherectomy,” Catheterization and Cardiovascular Interventions, vol. 81, no. 3, pp. 558–561, 2013, doi: 10.1002/ccd.23385.
R. Chopra, S. K. Wagner, and P. A. Keane, “Optical coherence tomography in the 2020s—outside the eye clinic,” Eye (Basingstoke), vol. 35, no. 1, pp. 236–243, 2021, doi: 10.1038/s41433-020-01263-6.
N. Prakash, “Eye Anatomy,” Handbook of Optometry and Eye Disorders, pp. 1–1, 2006, doi: 10.5005/jp/books/10343_1.
D. S. Kermany et al., “Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep Learning,” Cell, vol. 172, no. 5, pp. 1122-1131.e9, 2018, doi: 10.1016/j.cell.2018.02.010.
A. Laouarem, C. Kara-Mohamed, E. B. Bourennane, and A. Hamdi-Cherif, “HTC-retina: A hybrid retinal diseases classification model using transformer-Convolutional Neural Network from optical coherence tomography images,” Computers in Biology and Medicine, vol. 178, no. June, p. 108726, 2024, doi: 10.1016/j.compbiomed.2024.108726.
N. Novely, S. M. Shuvo, and Md. F. Faruk, “Improving Pre-Trained CNNs with CBAM and Skip Connections for Multi-Class Retinal Diseases Classification using OCT Images,” in Proceedings of the 3rd International Conference on Computing Advancements, ACM, 2024, pp. 946–953, doi: 10.1145/3723178.3723304.
Ş. Aykat and S. Senan, “Using Machine Learning to Detect Different Eye Diseases from OCT Images,” International Journal of Computational and Experimental Science and Engineering, vol. 9, no. 2, pp. 62–67, 2023, doi: 10.22399/ijcesen.1297655.
Z. Qi et al., “MSLI-Net: retinal disease detection network based on multi-segment localization and multi-scale interaction,” Front. Cell Dev. Biol., vol. 13, no. June, pp. 1–18, 2025, doi: 10.3389/fcell.2025.1608325.
J. Yang et al., “HyReti-Net: hybrid retinal diseases classification and diagnosis network using optical coherence tomography,” Front. Med., vol. 12, no. September, pp. 1–13, 2025, doi: 10.3389/fmed.2025.1660920.
A. Zedadra, M. Y. Salah-Salah, O. Zedadra, and A. Guerrieri, “Multi-Modal AI for Multi-Label Retinal Disease Prediction Using OCT and Fundus Images: A Hybrid Approach,” Sensors, vol. 25, no. 14, pp. 1–20, 2025, doi: 10.3390/s25144492.
K. Pin, J. W. Han, and Y. Nam, “Retinal diseases classification based on hybrid ensemble deep learning and optical coherence tomography images,” Electronic Research Archive, vol. 31, no. 8, pp. 4843–4861, 2023, doi: 10.3934/era.2023248.
A. M. Mutawa, K. Sabti, B. S. Sundaram Thankaleela, and S. Raizada, “Bio-Inspired Optimization of Transfer Learning Models for Diabetic Macular Edema Classification,” AI, vol. 6, no. 10, pp. 1–30, 2025, doi: 10.3390/ai6100269.
N. Verma, N. Kumar, G. Kumar, and K. Singh, “A hybrid ensemble framework with particle swarm optimization for network anomaly detection,” Discover Applied Sciences, vol. 7, no. 8, Aug. 2025, doi: 10.1007/s42452-025-07419-x.
O. S. Naren, “Retinal OCT Image Classification - C8” [Dataset]. Kaggle, 2021. doi: 10.34740/KAGGLE/DSV/2736749.
J. Deng, W. Dong, R. Socher, L.-J. Li, Kai Li, and Li Fei-Fei, “ImageNet: A large-scale hierarchical image database,” in Conference on Computer Vision and Pattern Recognition, IEEE, Jun. 2009, pp. 248–255. doi: 10.1109/CVPR.2009.5206848.
M. Tan and Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” in Proceedings of the International Conference on Machine Learning, PMLR, 2019, pp. 6105–6114, doi: 10.48550/arXiv.1905.11946.
X. Long et al., “EfficientNetB0-Based End-to-End Diagnostic System for Diabetic Retinopathy Grading and Macular Edema Detection,” Diabetes, Metabolic Syndrome and Obesity, vol. Volume 18, pp. 1311–1321, Apr. 2025, doi: 10.2147/DMSO.S506494.
F. F. Jiven and R. Rumini, “MRI Classification of Brain Tumors Using EfficientNetB0 Feature Extraction and Machine Learning Methods,” Journal of Applied Informatics and Computing, vol. 9, no. 6, pp. 3394–3404, Dec. 2025, doi: 10.30871/jaic.v9i6.10363.
N. Ma, X. Zhang, H.-T. Zheng, and J. Sun, “ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design,” in Computer Vision – ECCV, Springer, 2018, pp. 122–138. doi: 10.1007/978-3-030-01264-9_8.
I. Radosavovic, R. P. Kosaraju, R. Girshick, K. He, and P. Dollar, “Designing Network Design Spaces,” in Conference on Computer Vision and Pattern Recognition, IEEE, Jun. 2020, pp. 10425–10433. doi: 10.1109/CVPR42600.2020.01044.
J. Kennedy and R. Eberhart, “Particle swarm optimization,” in Proceedings of ICNN’95 - International Conference on Neural Networks, IEEE, 1995, pp. 1942–1948. doi: 10.1109/ICNN.1995.488968.
Y. Shi and R. Eberhart, “A modified particle swarm optimizer,” in 1998 IEEE International Conference on Evolutionary Computation Proceedings. IEEE World Congress on Computational Intelligence (Cat. No.98TH8360), IEEE, 1998, pp. 69–73. doi: 10.1109/ICEC.1998.699146.
R. Poli, J. Kennedy, and T. Blackwell, “Particle swarm optimization,” Swarm Intelligence, vol. 1, no. 1, pp. 33–57, Oct. 2007, doi: 10.1007/s11721-007-0002-0.
J. Cao, S. Kwong, R. Wang, X. Li, K. Li, and X. Kong, “Class-specific soft voting based multiple extreme learning machines ensemble,” Neurocomputing, vol. 149, no. Part A, pp. 275–284, Feb. 2015, doi: 10.1016/j.neucom.2014.02.072.
S. A. Hicks et al., “On evaluation metrics for medical applications of artificial intelligence,” Sci. Rep., vol. 12, no. 1, p. 5979, Apr. 2022, doi: 10.1038/s41598-022-09954-8.
A. Tharwat, “Classification assessment methods,” Applied Computing and Informatics, vol. 17, no. 1, pp. 168–192, Jan. 2021, doi: 10.1016/j.aci.2018.08.003.
F. Marini and B. Walczak, “Particle swarm optimization (PSO). A tutorial,” Chemometrics and Intelligent Laboratory Systems, vol. 149, pp. 153–165, Dec. 2015, doi: 10.1016/j.chemolab.2015.08.020.
A. P. Piotrowski, J. J. Napiorkowski, and A. E. Piotrowska, “Population size in Particle Swarm Optimization,” Swarm and Evolutionary Computation, vol. 58, Nov. 2020, doi: 10.1016/j.swevo.2020.100718.
J. Kittler, M. Hatef, R. P. W. Duin, and J. Matas, “On combining classifiers,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 20, no. 3, pp. 226–239, Mar. 1998, doi: 10.1109/34.667881.
R. Delgado, “A semi-hard voting combiner scheme to ensemble multi-class probabilistic classifiers,” Applied Intelligence, vol. 52, no. 4, pp. 3653–3677, Mar. 2022, doi: 10.1007/s10489-021-02447-7.
K. Simonyan and A. Zisserman, “Very Deep Convolutional Networks for Large-Scale Image Recognition,” arXiv preprint arXiv:1409.1556, 2015. doi: 10.48550/arXiv.1409.1556.
K. He, X. Zhang, S. Ren, and J. Sun, “Deep Residual Learning for Image Recognition,” in Conference on Computer Vision and Pattern Recognition, IEEE, 2016, pp. 770–778, doi: 10.1109/CVPR.2016.90.
A. Khan, A. Sohail, U. Zahoora, and A. S. Qureshi, “A survey of the recent architectures of deep convolutional neural networks,” Artificial Intelligence Review, vol. 53, no. 8, pp. 5455–5516, Dec. 2020, doi: 10.1007/s10462-020-09825-6.
Downloads
Published
Issue
Section
License
Copyright in each article belongs to the author
- The author acknowledges that the RESTI Journal (System Engineering and Information Technology) is the first publisher to publish with a license Creative Commons Attribution 4.0 International License.
- Authors can enter writing separately, arrange the non-exclusive distribution of manuscripts that have been published in this journal into other versions (eg sent to the author's institutional repository, publication in a book, etc.), by acknowledging that the manuscript has been published for the first time in the RESTI (Rekayasa Sistem dan Teknologi Informasi) journal ;
