Abstract
Lung cancer is the most common cause of cancer-related death in men worldwide and the second most common cause in women. Accurate prediction of lung cancer outcomes can help guide patient care and decision making. The amount and variety of available data on lung cancer cases continues to increase, which provides an opportunity to apply machine learning methods to predict lung cancer outcomes. Traditional population-wide machine learning methods for predicting clinical outcomes involve constructing a single model from training data and applying it to predict the outcomes for each future patient case. In contrast, instance-specific methods construct a model that is optimized to predict well for a given patient case. In this paper, we first describe an instance-specific method for learning Bayesian networks that we developed. We then use the Markov blanket of the outcome variable to predict 1-year survival in a cohort of 261 lung cancer patient cases containing clinical and omics variables. We report the results using AUROC as the evaluation measure. In leave-one-out testing, the instance-specific Bayesian network method achieved higher AUROC on average, compared to the population-wide Bayesian network method.
Keywords
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Agrawal, A., Misra, S., Narayanan, R., et al.: Lung cancer survival prediction using ensemble data mining on SEER data. Sci. Program. 20(1), 29–42 (2012)
American Cancer Society: Cancer Facts & Figures 2020. American Cancer Society, Atlanta (2020)
Boutilier, C., Friedman, N., Goldszmidt, M., Koller, D.: Context-specific independence in Bayesian networks. In: Proceedings of Uncertainty in Artificial Intelligence (UAI), pp. 115–123 (1996)
Bray, F., Ferlay, J., Soerjomataram, I., et al.: Global cancer statistics 2018: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries. CA Cancer J. Clin. 68(6), 394–424 (2018)
Cai, C., Cooper, G.F., Lu, K.N., et al.: Systematic discovery of the functional impact of somatic genome alterations in individual tumors through tumor-specific causal inference. PLoS Comput. Biol. 15(7), e1007088 (2019)
Chickering, D.M.: Learning Bayesian networks is NP-complete. In: Learning from Data, pp. 121–130. Springer (1996). https://doi.org/10.1007/978-1-4612-2404-4_12
Chickering, D.M.: Optimal structure identification with greedy search. J. Mach. Learn. Res. 3, 507–554 (2003)
Deist, T.M., Dankers, F.J., Valdes, G., et al.: Machine learning algorithms for outcome prediction in (chemo) radiotherapy: An empirical comparison of classifiers. Med. Phys. 45(7), 3449–3459 (2018)
DeLong, E.R., DeLong, D.M., Clarke-Pearson, D.L.: Comparing the areas under two or more correlated receiver operating characteristic curves: A nonparametric approach. Biometrics 44(3), 837–845 (1988)
Ferreira, A., Cooper, G.F., Visweswaran, S.: Decision path models for patient-specific modeling of patient outcomes. In: AMIA Annual Symposium Proceedings, pp. 413–421 (2013)
Heckerman, D.: A tutorial on learning with Bayesian networks. In: Learning in Graphical Models, pp. 301–354. Springer (1998)
Heckerman, D., Geiger, D., Chickering, D.M.: Learning Bayesian networks: The combination of knowledge and statistical data. Mach. Learn. 20(3), 197–243 (1995)
Jabbari, F., Cooper, G.F.: An instance-specific algorithm for learning the structure of causal Bayesian networks containing latent variables. In: Proceedings of the SIAM International Conference on Data Mining (SDM), pp. 433–441 (2020)
Jabbari, F., Visweswaran, S., Cooper, G.F.: Instance-specific Bayesian network structure learning. Proc. Mach. Learn. Res. 72, 169–180 (2018)
Jabbari, F., Visweswaran, S. and Cooper, G.F.: An empirical investigation of instance-specific causal Bayesian network learning. In: IEEE International Conference on Bioinformatics and Biomedicine (BIBM), pp. 2582–2585. IEEE (2019)
Kourou, K., Exarchos, T.P., Exarchos, K.P., et al.: Machine learning applications in cancer prognosis and prediction. Comput. Struct. Biotechnol. J. 13, 8–17 (2015)
Lengerich, B.J., Aragam, B., Xing, E.P.: Personalized regression enables sample-specific pan-cancer analysis. Bioinformatics 34(13), i178–i186 (2018)
Ramsey, J.D.: Scaling up greedy equivalence search for continuous variables. arXiv preprint arXiv:1507.07749 (2015)
Sesen, M.B., Kadir, T., Alcantara, R.B., et al.: Survival prediction and treatment recommendation with Bayesian techniques in lung cancer. In: AMIA Annual Symposium Proceedings, pp. 838–847 (2012)
Travis, W.D., Brambilla, E., Noguchi, M., et al.: International association for the study of lung cancer/American Thoracic Society/European Respiratory Society international multidisciplinary classification of lung adenocarcinoma. J. Thorac. Oncol. 6(2), 244–285 (2011)
Visweswaran, S., Cooper, G.F., Chickering, M.: Learning instance-specific predictive models. J. Mach. Learn. Res. 11(Dec), 3333–3369 (2010)
Zheng, Z., Webb, G.I.: Lazy learning of Bayesian rules. Mach. Learn. 41(1), 53–84 (2000)
Acknowledgement
The research reported in this paper was supported by grant #4100070287 from the Pennsylvania Department of Health (DOH), grant U54HG008540 from the National Human Genome Research Institute of the National Institutes of Health (NIH), and grant R01LM012095 from the National Library of Medicine of the NIH. The content of the paper is solely the responsibility of the authors and does not necessarily represent the official views of the Pennsylvania DOH or the NIH.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2020 Springer Nature Switzerland AG
About this paper
Cite this paper
Jabbari, F., Villaruz, L.C., Davis, M., Cooper, G.F. (2020). Lung Cancer Survival Prediction Using Instance-Specific Bayesian Networks. In: Michalowski, M., Moskovitch, R. (eds) Artificial Intelligence in Medicine. AIME 2020. Lecture Notes in Computer Science(), vol 12299. Springer, Cham. https://doi.org/10.1007/978-3-030-59137-3_14
Download citation
DOI: https://doi.org/10.1007/978-3-030-59137-3_14
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-59136-6
Online ISBN: 978-3-030-59137-3
eBook Packages: Computer ScienceComputer Science (R0)