Affiliation:
1. Department of Computer Engineering, North Tehran Branch Islamic Azad University Tehran Iran
2. Department of Statistics, North Tehran Branch Islamic Azad University Tehran Iran
3. Department of Mathematics, North Tehran Branch Islamic Azad University Tehran Iran
Abstract
AbstractCOVID‐19 has had a profound global impact, necessitating the development of infection detection systems based on machine learning. This paper presents a Multi‐task architecture that addresses the classification and segmentation tasks for COVID‐19 detection. The model comprises an encoder for feature representation, a decoder for segmentation, and a multi‐layer perceptron for classification. Evaluations conducted on two datasets demonstrate the model's performance in both classification and segmentation. To enhance efficiency and diagnosis accuracy, CT‐scan images undergo pre‐processing using image processing algorithms like histogram equalization, median filtering, and mathematical morphology operations. The combination of the median filter pre‐processing and the proposed model yields impressive results in the classification task, achieving high accuracy, sensitivity, and specificity, with values of 0.97, 0.97, and 0.96, respectively, for dataset 1, and 0.96 in mentioned metrics for dataset 2. For segmentation, the proposed model, particularly with the average morphology pre‐processing, exhibits excellent performance with high accuracy, low mean squared error, high peak signal‐to‐noise ratio, high structural similarity index, and a mean dice coefficient of 88.86 ± 0.05 for dataset 1, and 87.97 ± 0.02 for dataset 2. Furthermore, the pre‐trained models consistently demonstrate the superiority of the median filter and proposed model in the classification task on the same datasets. In conclusion, the proposed multi‐task model, incorporating image processing techniques, achieves remarkable results in both classification and segmentation. The utilization of pre‐processing algorithms and the multi‐task framework significantly contribute to superior performance metrics. This study encourages further exploration of combining diverse image processing algorithms to advance infection diagnosis and treatment.
Publisher
Institution of Engineering and Technology (IET)
Subject
Electrical and Electronic Engineering,Computer Vision and Pattern Recognition,Signal Processing,Software
Cited by
2 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
1. Pioneering Advanced Deep Multi-Task Learning Model for Precise COVID-19 Diagnosis;2023 2nd International Engineering Conference on Electrical, Energy, and Artificial Intelligence (EICEEAI);2023-12-27
2. Cutting-edge multi-task model: unveiling COVID-19 through fusion of image processing algorithms;Computer Methods in Biomechanics and Biomedical Engineering: Imaging & Visualization;2023-12-05