Affiliation:
1. Research Centre, African Institute for Mathematical Sciences (AIMS) South Africa, Cape Town 7945, South Africa
2. Department of Computer Science and Informatics, University of the Free State, Phuthaditjhaba 9866, South Africa
3. Department of Mathematical Sciences, Stellenbosch University, Cape Town 7945, South Africa
Abstract
This study proposes a deep-learning-based solution (named CapsNetCovid) for COVID-19 diagnosis using a capsule neural network (CapsNet). CapsNets are robust for image rotations and affine transformations, which is advantageous when processing medical imaging datasets. This study presents a performance analysis of CapsNets on standard images and their augmented variants for binary and multi-class classification. CapsNetCovid was trained and evaluated on two COVID-19 datasets of CT images and X-ray images. It was also evaluated on eight augmented datasets. The results show that the proposed model achieved classification accuracy, precision, sensitivity, and F1-score of 99.929%, 99.887%, 100%, and 99.319%, respectively, for the CT images. It also achieved a classification accuracy, precision, sensitivity, and F1-score of 94.721%, 93.864%, 92.947%, and 93.386%, respectively, for the X-ray images. This study presents a comparative analysis between CapsNetCovid, CNN, DenseNet121, and ResNet50 in terms of their ability to correctly identify randomly transformed and rotated CT and X-ray images without the use of data augmentation techniques. The analysis shows that CapsNetCovid outperforms CNN, DenseNet121, and ResNet50 when trained and evaluated on CT and X-ray images without data augmentation. We hope that this research will aid in improving decision making and diagnostic accuracy of medical professionals when diagnosing COVID-19.
Reference54 articles.
1. COVID-19 diagnosis using deep learning neural networks applied to CT images;Akinyelu;Front. Artif. Intell.,2022
2. Sabour, S., Frosst, N., and Hinton, G.E. (2017). Dynamic routing between capsules. arXiv.
3. LaLonde, R., and Bagci, U. (2018). Capsules for object segmentation. arXiv.
4. A pre-trained convolutional neural network with optimized capsule networks for chest X-rays COVID-19 diagnosis;Darwish;Cluster Comput.,2022
5. SMOTE: Synthetic minority over-sampling technique;Chawla;J. Artif. Intell. Res.,2002
Cited by
4 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献