Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray.
Thoracic diseases, including pneumonia, tuberculosis, lung cancer, and others, pose significant health risks and require timely and accurate diagnosis to ensure proper treatment. Thus, in this research, a model for thorax disease classification using Chest X-rays is proposed by considering deep lear...
Saved in:
| Main Authors: | , , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2025-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://doi.org/10.1371/journal.pone.0327099 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1849425016750342144 |
|---|---|
| author | Nadim Rana Yahaya Coulibaly Ayman Noor Talal H Noor Md Imran Alam Zeba Khan Ali Tahir Mohammad Zubair Khan |
| author_facet | Nadim Rana Yahaya Coulibaly Ayman Noor Talal H Noor Md Imran Alam Zeba Khan Ali Tahir Mohammad Zubair Khan |
| author_sort | Nadim Rana |
| collection | DOAJ |
| description | Thoracic diseases, including pneumonia, tuberculosis, lung cancer, and others, pose significant health risks and require timely and accurate diagnosis to ensure proper treatment. Thus, in this research, a model for thorax disease classification using Chest X-rays is proposed by considering deep learning model. The input is pre-processed by resizing, normalizing pixel values, and applying data augmentation to address the issue of imbalanced datasets and improve model generalization. Significant features are extracted from the images using an Enhanced Auto-Encoder (EnAE) model, which combines a stacked auto-encoder architecture with an attention module to enhance feature representation and classification accuracy. To further improve feature selection, we utilize the Chaotic Whale Optimization (ChWO) Algorithm, which optimally selects the most relevant attributes from the extracted features. Finally, the disease classification is performed using the novel Improved Swin Transformer (IMSTrans) model, which is designed to efficiently process high-dimensional medical image data and achieve superior classification performance. The proposed EnAE + ChWO+IMSTrans model for thorax disease classification was evaluated using extensive Chest X-ray datasets and the Lung Disease Dataset. The proposed method demonstrates enhanced Accuracy, Precision, Recall, F-Score, MCC and MAE of 0.964, 0.977, 0.9845, 0.964, 0.9647, and 0.184 respectively indicating the reliable and efficient solution for thorax disease classification. |
| format | Article |
| id | doaj-art-2fd520ec3e9c44ec9d2377376373a142 |
| institution | Kabale University |
| issn | 1932-6203 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-2fd520ec3e9c44ec9d2377376373a1422025-08-20T03:29:53ZengPublic Library of Science (PLoS)PLoS ONE1932-62032025-01-01206e032709910.1371/journal.pone.0327099Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray.Nadim RanaYahaya CoulibalyAyman NoorTalal H NoorMd Imran AlamZeba KhanAli TahirMohammad Zubair KhanThoracic diseases, including pneumonia, tuberculosis, lung cancer, and others, pose significant health risks and require timely and accurate diagnosis to ensure proper treatment. Thus, in this research, a model for thorax disease classification using Chest X-rays is proposed by considering deep learning model. The input is pre-processed by resizing, normalizing pixel values, and applying data augmentation to address the issue of imbalanced datasets and improve model generalization. Significant features are extracted from the images using an Enhanced Auto-Encoder (EnAE) model, which combines a stacked auto-encoder architecture with an attention module to enhance feature representation and classification accuracy. To further improve feature selection, we utilize the Chaotic Whale Optimization (ChWO) Algorithm, which optimally selects the most relevant attributes from the extracted features. Finally, the disease classification is performed using the novel Improved Swin Transformer (IMSTrans) model, which is designed to efficiently process high-dimensional medical image data and achieve superior classification performance. The proposed EnAE + ChWO+IMSTrans model for thorax disease classification was evaluated using extensive Chest X-ray datasets and the Lung Disease Dataset. The proposed method demonstrates enhanced Accuracy, Precision, Recall, F-Score, MCC and MAE of 0.964, 0.977, 0.9845, 0.964, 0.9647, and 0.184 respectively indicating the reliable and efficient solution for thorax disease classification.https://doi.org/10.1371/journal.pone.0327099 |
| spellingShingle | Nadim Rana Yahaya Coulibaly Ayman Noor Talal H Noor Md Imran Alam Zeba Khan Ali Tahir Mohammad Zubair Khan Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. PLoS ONE |
| title | Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. |
| title_full | Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. |
| title_fullStr | Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. |
| title_full_unstemmed | Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. |
| title_short | Improved swin transformer-based thorax disease classification with optimal feature selection using chest X-ray. |
| title_sort | improved swin transformer based thorax disease classification with optimal feature selection using chest x ray |
| url | https://doi.org/10.1371/journal.pone.0327099 |
| work_keys_str_mv | AT nadimrana improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT yahayacoulibaly improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT aymannoor improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT talalhnoor improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT mdimranalam improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT zebakhan improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT alitahir improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray AT mohammadzubairkhan improvedswintransformerbasedthoraxdiseaseclassificationwithoptimalfeatureselectionusingchestxray |