Enhancing Lung Cancer Screening with Bidirectional LSTM and GRU Models
Abstract. This study aims to enhance lung cancer patient screening by developing and evaluating bidirectional Long Short-Term Memory (LSTM) and bidirectional Gated Recurrent Unit (GRU) models using the Lung Cancer dataset from Kaggle. The dataset includes 16 features related to patient symptoms and lung cancer status, providing a broad spectrum of symptoms to improve model accuracy. The research advances Artificial Intelligence (AI)-driven healthcare by integrating these sophisticated machine learning techniques into diagnostic processes. The methodology involves four main steps: preprocessing the dataset for model compatibility, defining the model architecture with bidirectional LSTM and GRU layers and evaluating its performance. The results show an overall accuracy of 52.17%, with accuracy, recall, and F1 scores for both cancerous and non-cancerous categories around 50%. Despite the hybrid model's average performance, it establishes a basis for future enhancements. Optimizing model parameters and exploring additional other techniques to improve prediction accuracy and clinical applicability will be done in the future.
Authors
- Ang Ji (ORCID: https://orcid.org/0000-0002-7943-7461)
Institutions
- South China University of Technology (CN)
Publication Details
- Journal
- Applied and Computational Engineering
- Published
- 2024-11-08
- DOI
- https://doi.org/10.54254/2755-2721/104/20241187
- Citations
- 2
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.81