Ensemble learning driven Kolmogorov-Arnold Networks-based Lung Cancer classification

Abstract

Early Lung Cancer (LC) detection is essential for reducing the global mortality rate. The limitations of traditional diagnostic techniques cause challenges in identifying LC using medical imaging data. In this study, we aim to develop a robust LC detection model. Positron Emission Tomography / Computed Tomography (PET / CT) images are utilized to comprehend the metabolic and anatomical data, leading to optimal LC diagnosis. In order to extract multiple LC features, we enhance MobileNet V3 and LeViT models. The weighted sum feature fusion technique is used to generate unique LC features. The extracted features are classified using spline functions, including linear, cubic, and B-spline of Kolmogorov–Arnold Networks (KANs). We ensemble the outcomes using the soft-voting approach. The model is generalized using the Lung—PET–CT–DX dataset. Five–fold cross-validation is used to evaluate the model. The proposed LC detection model achieves an impressive accuracy of 99.0% with a minimal loss of 0.07. In addition, limited resources are required to classify PET / CT images. The high performance underscores the potential of the proposed LC detection model in providing valuable and optimal results. The study findings can significantly improve clinical practice by presenting sophisticated and interpretable outcomes. The proposed model can be enhanced by integrating advanced feature fusion techniques.