Muhammad Ghulam Abbas Malik , Adnan Saeed , Khurram Shehzad , Muddesar Iqbal
{"title":"DEF-SwinE2NET: Dual enhanced features guided with multi-model fusion for brain tumor classification using preprocessing optimization","authors":"Muhammad Ghulam Abbas Malik , Adnan Saeed , Khurram Shehzad , Muddesar Iqbal","doi":"10.1016/j.bspc.2024.107079","DOIUrl":null,"url":null,"abstract":"<div><div>Brain tumors exhibit significant variability in shape, size, and location, making it difficult to achieve consistent and accurate classification. It requires advanced algorithms for handling diverse tumor presentations. To solve this issue, we propose a Dual-Enhanced Features Scheme (DEFS) with a Swin-Transformer model based on the EfficientNetV2S to improve the classification and reuse parameters. In DEFS, the dense-block with dilation enables to uncovering of hidden details and spatial relationships across varying scales in the model which are typically obscured by traditional convolutional-layers. This module is particularly crucial in medical imaging, where tumors and anomalies can present in various sizes and shapes. Further, the dual-attention mechanism in the enhanced Featured scheme enhances the explainability and interpretability of the model by using spatial and channel-wise information. Additionally, the Swin-Transformer-block improves the model’s capabilities to capture global patterns in brain-tumor images, which is highly advantageous in medical-imaging where the location and extent of abnormalities, such as tumors, can vary significantly. To strengthen the proposed DEF-SwinE2NET, we used EfficientNetV2S as a baseline-model due to its effectiveness and accurate classification compared to its predecessors. We evaluated DEFSwinE2NET using three benchmark datasets: two were sourced from Kaggle and one from a Figshare repositories. Several preprocessing-steps were applied to enhance the MRI-images before training including image cropping, median-filter noise-reduction, contrast-limited adaptive histogram equalization (CLAHE) for local-contrast enhancement, Laplacian-edge enhancement to highlight critical features, and data augmentation to improve model robustness and generalization. The DEF-SwinE2NET model achieves remarkable results with an accuracy of 99.43 %, a sensitivity of 99.39 %, and an F1-score of 99.41 %.</div></div>","PeriodicalId":55362,"journal":{"name":"Biomedical Signal Processing and Control","volume":null,"pages":null},"PeriodicalIF":4.9000,"publicationDate":"2024-10-28","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"Biomedical Signal Processing and Control","FirstCategoryId":"5","ListUrlMain":"https://www.sciencedirect.com/science/article/pii/S1746809424011376","RegionNum":2,"RegionCategory":"医学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"ENGINEERING, BIOMEDICAL","Score":null,"Total":0}
引用次数: 0
Abstract
Brain tumors exhibit significant variability in shape, size, and location, making it difficult to achieve consistent and accurate classification. It requires advanced algorithms for handling diverse tumor presentations. To solve this issue, we propose a Dual-Enhanced Features Scheme (DEFS) with a Swin-Transformer model based on the EfficientNetV2S to improve the classification and reuse parameters. In DEFS, the dense-block with dilation enables to uncovering of hidden details and spatial relationships across varying scales in the model which are typically obscured by traditional convolutional-layers. This module is particularly crucial in medical imaging, where tumors and anomalies can present in various sizes and shapes. Further, the dual-attention mechanism in the enhanced Featured scheme enhances the explainability and interpretability of the model by using spatial and channel-wise information. Additionally, the Swin-Transformer-block improves the model’s capabilities to capture global patterns in brain-tumor images, which is highly advantageous in medical-imaging where the location and extent of abnormalities, such as tumors, can vary significantly. To strengthen the proposed DEF-SwinE2NET, we used EfficientNetV2S as a baseline-model due to its effectiveness and accurate classification compared to its predecessors. We evaluated DEFSwinE2NET using three benchmark datasets: two were sourced from Kaggle and one from a Figshare repositories. Several preprocessing-steps were applied to enhance the MRI-images before training including image cropping, median-filter noise-reduction, contrast-limited adaptive histogram equalization (CLAHE) for local-contrast enhancement, Laplacian-edge enhancement to highlight critical features, and data augmentation to improve model robustness and generalization. The DEF-SwinE2NET model achieves remarkable results with an accuracy of 99.43 %, a sensitivity of 99.39 %, and an F1-score of 99.41 %.
期刊介绍:
Biomedical Signal Processing and Control aims to provide a cross-disciplinary international forum for the interchange of information on research in the measurement and analysis of signals and images in clinical medicine and the biological sciences. Emphasis is placed on contributions dealing with the practical, applications-led research on the use of methods and devices in clinical diagnosis, patient monitoring and management.
Biomedical Signal Processing and Control reflects the main areas in which these methods are being used and developed at the interface of both engineering and clinical science. The scope of the journal is defined to include relevant review papers, technical notes, short communications and letters. Tutorial papers and special issues will also be published.