UPM Institutional Repository

SwinUNeLCsT: Global–local spatial representation learning with hybrid CNN–transformer for efficient tuberculosis lung cavity weakly supervised semantic segmentation


Citation

Zhuoyi, Tan and Hizmawati, Madzin and Bahari, Norafida and Rahmita, Wirza OK Rahmat and Fatimah, Khalid and Puteri, Suhaiza Sulaiman (2024) SwinUNeLCsT: Global–local spatial representation learning with hybrid CNN–transformer for efficient tuberculosis lung cavity weakly supervised semantic segmentation. Journal of King Saud University - Computer and Information Sciences, 36 (4). art. no. 102012. pp. 1-15. ISSN 1319-1578; ESSN: 2213-1248

Abstract

Radiological diagnosis of lung cavities (LCs) is the key to identifying tuberculosis (TB). Conventional deep learning methods rely on a large amount of accurate pixel-level data to segment LCs. This process is timeconsuming and laborious, especially for those subtle LCs. To address such challenges, firstly, we introduce a novel 3D TB LCs imaging convolutional neural network (CNN)-transformer hybrid model (SwinUNeLCsT). The core idea of SwinUNeLCsT is to combine local details and global dependencies for TB CT scan image feature representation to effectively improve the recognition ability of LCs. Secondly, to reduce the dependence on accurate pixel-level annotations, we design an end-to-end LCs weakly supervised semantic segmentation (WSSS) framework. Through this framework, radiologists need only to classify the number and the approximate location (e.g., left lung, right lung, or both) of LCs in the CT scan to achieve efficient segmentation of the LCs. This process eliminates the need for meticulously drawing boundaries, greatly reducing the cost of annotation. Extensive experimental results show that SwinUNeLCsT outperforms currently popular medical 3D segmentation methods in the supervised semantic segmentation paradigm. Meanwhile, our WSSS framework based on SwinUNeLCsT also performs best among the existing state-of-the-art medical 3D WSSS methods.


Download File

[img] Text
SwinUNeLCsT.pdf

Download (5MB)

Additional Metadata

Item Type: Article
Divisions: Faculty of Computer Science and Information Technology
Faculty of Medicine and Health Science
DOI Number: https://doi.org/10.1016/j.jksuci.2024.102012
Publisher: Elsevier
Keywords: Classification; CT tuberculosis imaging; Deep learning; Semantic segmentation; Weakly-supervised learning
Depositing User: Scopus 2024
Date Deposited: 27 Jun 2024 16:14
Last Modified: 27 Jun 2024 16:14
Altmetrics: http://www.altmetric.com/details.php?domain=psasir.upm.edu.my&doi=10.1016/j.jksuci.2024.102012
URI: http://psasir.upm.edu.my/id/eprint/111381
Statistic Details: View Download Statistic

Actions (login required)

View Item View Item