High-Dimensional Bayesian Optimization of Personalized Cardiac Model Parameters via an Embedded Generative Model
The estimation of patient-specific tissue properties in the form of model parameters is important for personalized physiological models. However, these tissue properties are spatially varying across the underlying anatomical model, presenting a significance challenge of high-dimensional (HD) optimization at the presence of limited measurement data. A common solution to reduce the dimension of the parameter space is to explicitly partition the anatomical mesh, either into a fixed small number of segments or a multi-scale hierarchy. This anatomy-based reduction of parameter space presents a fundamental bottleneck to parameter estimation, resulting in solutions that are either too low in resolution to reflect tissue heterogeneity, or too high in dimension to be reliably estimated within feasible computation. In this paper, we present a novel concept that embeds a generative variational auto-encoder (VAE) into the objective function of Bayesian optimization, providing an implicit low-dimensional (LD) search space that represents the generative code of the HD spatially-varying tissue properties. In addition, the VAE-encoded knowledge about the generative code is further used to guide the exploration of the search space. The presented method is applied to estimating tissue excitability in a cardiac electrophysiological model. Synthetic and real-data experiments demonstrate its ability to improve the accuracy of parameter estimation with more than 10x gain in efficiency.
KeywordsParameter estimation Model personalization Cardiac electrophysiology Variational auto-encoder Bayesian optimization
This work is supported by the National Science Foundation under CAREER Award ACI-1350374 and the National Institute of Heart, Lung, and Blood of the National Institutes of Health under Award R21Hl125998.
- 2.Brochu, E., Cora, V.M., De Freitas, N.: A tutorial on Bayesian optimization of expensive cost functions, with application to active user modeling and hierarchical reinforcement learning. arXiv preprint arXiv:1012.2599 (2010)
- 4.Dhamala, J., Sapp, J.L., Horacek, M., Wang, L.: Spatially-adaptive multi-scale optimization for local parameter estimation: application in cardiac electrophysiological models. In: Ourselin, S., Joskowicz, L., Sabuncu, M.R., Unal, G., Wells, W. (eds.) MICCAI 2016. LNCS, vol. 9902, pp. 282–290. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46726-9_33CrossRefGoogle Scholar
- 6.Kingma, D.P., Welling, M.: Auto-encoding variational bayes. arXiv preprint arXiv:1312.6114 (2013)
- 7.Lê, M., Delingette, H., Kalpathy-Cramer, J.: MRI based Bayesian personalization of a tumor growth model. IEEE TMI 35(10), 2329–2339 (2016)Google Scholar
- 8.Plonsey, R.: Bioelectric Phenomena. Wiley Online Library, Hoboken (1969)Google Scholar