EM-AWFNet: dual-branch adaptive weight fusion for lung cancer PET/CT classification.
Authors
Affiliations (2)
Affiliations (2)
- Henan University of Science and Technology, School of Medical Technology and Engineering, Luoyang, 471023, China.
- Henan University of Science and Technology, School of Medical Technology and Engineering, Luoyang, 471003, China.
Abstract
Lung cancer (LC) is a malignant tumor that poses a serious threat to human health. Positron emission tomography/computed tomography (PET/CT) imaging, which integrates both anatomical structure and metabolic activity, plays a crucial role in the auxiliary diagnosis of lung cancer. Despite its advantages, the automated classification of lung cancer subtypes is challenged by the overlapping imaging features among different subtypes and the complex morphological characteristics of the lesions. To address the aforementioned challenges, this paper proposes EM-AWFNet, a dual-branch adaptive weight fusion network based on deep learning for lung cancer classification using PET/CT images. The method employs a pretrained EfficientNet-B1 as the local path to extract detailed structural features, while MobileViT serves as the global path to capture long-range contextual semantic information. A Transformer module is simultaneously introduced to facilitate cross-branch feature interaction. Furthermore, an adaptive weight fusion mechanism is designed to dynamically adjust the contribution of each feature branch during the classification process. Experimental results demonstrate that the proposed method achieves a classification accuracy of 0.9754 on the Lung-PET-CT-DX dataset for the three-class classification task of lung adenocarcinoma (ADC), small cell carcinoma (SCC), and squamous cell carcinoma (SQC). It provides new insights for research on intelligent analysis and computer-aided diagnosis of lung cancer using PET/CT images.