Optimized Deep Learning Framework with H2O for Lung Cancer Prediction

Authors

Walaa Hassan Ibrahim, Computer Engineering and Control Systems Department, Faculty of Engineering, Mansoura University, Mansoura 35516, EgyptFollow
Mohamed S. Saraya, Computer Engineering and Control Systems Department, Faculty of Engineering, Mansoura University, Mansoura 35516, Egypt
Sally M. Elghamrawy, Computer Engineering Department, MISR Higher Institute for Engineering and Technology, Mansoura 35516, Egypt
Ali I. Eldesouky, Computer Engineering and Control Systems Department, Faculty of Engineering, Mansoura University, Mansoura 35516, Egypt

Corresponding Author

Walaa Hassan Ibrahim

Subject Area

Computer and Control Systems Engineering

Article Type

Original Study

Abstract

The automatic diagnosis of lung cancer using chest X-ray (CXR) images has significantly advanced with progress in computing, machine learning, and deep learning. However, detecting lesions and nodules remains challenging due to CXR limitations. Early lung cancer detection is critical for successful treatment, but current AI algorithms often rely on large annotated datasets, which are not always available. To address this, a novel multi-classification deep learning framework is proposed that combines CXR and CT images. This approach leverages the detailed feature detection capabilities of CT scans alongside the complementary views from CXRs, improving early-stage lung cancer detection and classification precision. The proposed framework integrates the Binary Adaptive Crossover Particle Whale Optimization Algorithm-S (BACP-WOA-S) to efficiently address high-dimensionality challenges. The proposed framework includes key contributions: Employing a Binary Whale Optimization Algorithm variation (BWOA) is used in the feature selection (FS) layer to reduce computational complexity while preserving key features. Additionally, a Feed-Forward Neural Network (FFNN) is utilized in the Deep Learning (DL) layer to optimize characteristics like layers, neurons, and activation functions. The framework, powered by H2O for scalable deep learning processes, achieves a 97.6% accuracy rate, showcasing its effectiveness in early lung cancer detection.

Keywords

Lung cancer, chest x-ray, deep learning framework, Whale Optimization Algorithm, CT images, H2O

Creative Commons License

This work is licensed under a Creative Commons Attribution 4.0 License.

Recommended Citation

Ibrahim, Walaa Hassan; Saraya, Mohamed S.; Elghamrawy, Sally M.; and Eldesouky, Ali I. (2026) "Optimized Deep Learning Framework with H2O for Lung Cancer Prediction," Mansoura Engineering Journal: Vol. 51 : Iss. 2 , Article 2.
Available at: https://doi.org/10.58491/2735-4202.3408