Lung cancer prediction with advanced graph neural networks

Abstract

This research aims to enhance lung cancer prediction using advanced machine learning techniques. The major finding is that integrating graph convolutional networks (GCNs) with graph attention networks (GATs) significantly improves predictive accuracy. The problem addressed is the need for early and accurate detection of lung cancer, leveraging a dataset from Kaggle's "Lung Cancer Prediction Dataset," which includes 309 instances and 16 attributes. The proposed A-GCN with GAT model is meticulously engineered with multiple layers and hidden units, optimized through hyperparameter adjustments, early stopping mechanisms, and Adam optimization techniques. Experimental results demonstrate the model's superior performance, achieving an accuracy of 0.9454, precision of 0.9213, recall of 0.9743, and an F1 score of 0.9482. These findings highlight the model's efficacy in capturing intricate patterns within patient data, facilitating early interventions and personalized treatment plans. This research underscores the potential of graph-based methodologies in medical research, particularly for lung cancer prediction, ultimately aiming to improve patient outcomes and survival rates through proactive healthcare interventions.