Machine Learning in Prediction of Bladder Cancer on Clinical Laboratory Data

• Published in: Diagnostics (Basel) Vol. 12; no. 1; p. 203
• Main Authors: Tsai, I-Jung, Shen, Wen-Chi, Lee, Chia-Ling, Wang, Horng-Dar, Lin, Ching-Yu
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 14.01.2022; MDPI
• Subjects: Accuracy; Algorithms; Analytical chemistry; Artificial intelligence; Bladder cancer; Cardiovascular disease; clinical laboratory data; Creatinine; Decision trees; DNA methylation; Feature selection; Hospitals; Kidney diseases; Liver diseases; Lung cancer; Machine learning; Medical laboratories; Medical research; Prostate cancer; Support vector machines; Urinalysis; Urine; Urogenital system; machine learning; clinical laboratory data; bladder cancer; feature selection
• ISSN: 2075-4418; 2075-4418
• DOI: 10.3390/diagnostics12010203

Bladder cancer has been increasing globally. Urinary cytology is considered a major screening method for bladder cancer, but it has poor sensitivity. This study aimed to utilize clinical laboratory data and machine learning methods to build predictive models of bladder cancer. A total of 1336 patients with cystitis, bladder cancer, kidney cancer, uterus cancer, and prostate cancer were enrolled in this study. Two-step feature selection combined with WEKA and forward selection was performed. Furthermore, five machine learning models, including decision tree, random forest, support vector machine, extreme gradient boosting (XGBoost), and light gradient boosting machine (GBM) were applied. Features, including calcium, alkaline phosphatase (ALP), albumin, urine ketone, urine occult blood, creatinine, alanine aminotransferase (ALT), and diabetes were selected. The lightGBM model obtained an accuracy of 84.8% to 86.9%, a sensitivity 84% to 87.8%, a specificity of 82.9% to 86.7%, and an area under the curve (AUC) of 0.88 to 0.92 in discriminating bladder cancer from cystitis and other cancers. Our study provides a demonstration of utilizing clinical laboratory data to predict bladder cancer.