Prediction of cancer dynamics under treatment using Bayesian neural networks: A simulated study

Even Moa Myklebust,Arnoldo Frigessi,Fredrik Schjesvold,Jasmine Foo,Kevin Leder,Alvaro Köhn-Luque
2024-05-23
Abstract:Predicting cancer dynamics under treatment is challenging due to high inter-patient heterogeneity, lack of predictive biomarkers, and sparse and noisy longitudinal data. Mathematical models can summarize cancer dynamics by a few interpretable parameters per patient. Machine learning methods can then be trained to predict the model parameters from baseline covariates, but do not account for uncertainty in the parameter estimates. Instead, hierarchical Bayesian modeling can model the relationship between baseline covariates to longitudinal measurements via mechanistic parameters while accounting for uncertainty in every part of the model.
Quantitative Methods,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to predict cancer dynamics during the treatment process. Specifically, the researchers focus on how to use baseline covariates (such as blood test results, fluorescence in situ hybridization (FISH), age, and gender, etc.) to predict the M - protein trajectory of patients with multiple myeloma (MM), and then reflect the changes in tumor burden. The changes in M - protein levels during the treatment process can provide important information about the treatment effect, but current methods face challenges in dealing with patient - to - patient heterogeneity, the unclear importance of available biomarkers, and the sparseness and high noise of longitudinal measurement data. To address these challenges, the researchers proposed a hierarchical Bayesian model combined with Bayesian neural networks (BNN). This model can not only capture the nonlinear relationships and interactions between covariates but also improve the prediction accuracy by introducing uncertainty quantification. By comparing with the traditional linear mapping model, the researchers demonstrated the superior performance of the BNN model in predicting the M - protein trajectory, especially in the presence of covariate interactions. In summary, the main goal of this paper is to develop a more effective prediction tool to help doctors select the most appropriate treatment plan according to the baseline characteristics of patients, thereby achieving personalized medicine.