Article Per Year (5 Year)
p-Index From 2021 - 2026
0.23
P-Index
This Author published in this journals
All Journal IAES International Journal of Artificial Intelligence (IJ-AI)
Poonsawad, Akarapon
Unknown Affiliation
Computer Science & IT Engineering

Published : 1 Documents Claim Missing Document
Claim Missing Document
Check
Articles

Found 1 Documents
Search

 1 
Lung cancer patients survival prediction using outlier detection and optimized XGBoost Yotsawat, Wirot; Suebpeng, Peetiphart; Purisangkaha, Saroch; Poonsawad, Akarapon; Phodong, Kanyalag
IAES International Journal of Artificial Intelligence (IJ-AI) Vol 14, No 3: June 2025
Publisher : Institute of Advanced Engineering and Science

Show Abstract | Download Original | Original Source | Check in Google Scholar | DOI: 10.11591/ijai.v14.i3.pp2146-2157

Abstract

This research aims to improve the prediction’s model for survival time of lung cancer patients by using outlier detection, hyper-parameter optimization, and machine learning technique. The research compares the performance of several methods including multilayer perceptron (MLP), decision tree (DT), linear regression (LR), Bagging, XGBoost, and random forest (RF). The dataset used for the experiment is obtained from the surveillance, epidemiology, and end result (SEER) cancer database, which contains diagnoses data from 2004 to 2015. The total number of records used is 196,031 with 22 features. 10-fold cross-validation is used for training and testing sets. The evaluation metrics are root mean square error (RMSE), mean squared error (MSE), R-squared (R2), and mean absolute error (MAE). The results show that the lung cancer patient survival prediction model using the optimized XGBoost (O-XGBoost) model performs the best with an RMSE of 13.74 and outperforms the baseline-XGBoost model as well as other models. This research will be useful for developing a clinical decision support system for the care of lung cancer patients. Physicians can use the developed model to assess the patient’s chance of survival in order to plan more effective treatment.
Co-Authors Phodong, Kanyalag Purisangkaha, Saroch Suebpeng, Peetiphart Yotsawat, Wirot