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.

Cite this paper

@article{moozhippurath2024lung,
  title = {{Lung cancer prediction with advanced graph neural networks}},
  author = {Moozhippurath, Bineesh and Jayapandian, N.},
  journal = {Indonesian Journal of Electrical Engineering and Computer Science},
  volume = {37},
  number = {2},
  pages = {1077},
  year = {2024},
  doi = {10.11591/ijeecs.v37.i2.pp1077-1084}
}

Made from the index record. Check names, capitalisation and page numbers against the paper. Need another style? AI for Research has over 10,000.

Related papers