A self-taught artificial agent for multi-physics computational model personalization

Personalization is the process of fitting a model to patient data, a critical step towards application of multi-physics computational models in clinical practice. Designing robust personalization algorithms is often a tedious, time-consuming, model- and data-specific process. We propose to use artificial intelligence concepts to learn this task, inspired by how human experts manually perform it. The problem is reformulated in terms of reinforcement learning. In an off-line phase, Vito, our self-taught artificial agent, learns a representative decision process model through exploration of the computational model: it learns how the model behaves under change of parameters. The agent then automatically learns an optimal strategy for on-line personalization. The algorithm is model-independent; applying it to a new model requires only adjusting few hyper-parameters of the agent and defining the observations to match. The full knowledge of the model itself is not required. Vito was tested in a synthetic scenario, showing that it could learn how to optimize cost functions generically. Then Vito was applied to the inverse problem of cardiac electrophysiology and the personalization of a whole-body circulation model. The obtained results suggested that Vito could achieve equivalent, if not better goodness of fit than standard methods, while being more robust (up to 11% higher success rates) and with faster (up to seven times) convergence rate. Our artificial intelligence approach could thus make personalization algorithms generalizable and self-adaptable to any patient and any model.

[1]  Jan Haas,et al.  Vito - A Generic Agent for Multi-physics Model Personalization: Application to Heart Modeling , 2015, MICCAI.

[2]  Alistair A. Young,et al.  Modelling passive diastolic mechanics with quantitative MRI of cardiac structure and function , 2009, Medical Image Anal..

[3]  Hervé Delingette,et al.  Personalised Electromechanical Model of the Heart for the Prediction of the Acute Effects of Cardiac Resynchronisation Therapy , 2009, FIMH.

[4]  Liming Xiang,et al.  Kernel-Based Reinforcement Learning , 2006, ICIC.

[5]  Sergei Vassilvitskii,et al.  k-means++: the advantages of careful seeding , 2007, SODA '07.

[6]  Gerald Tesauro,et al.  TD-Gammon, a Self-Teaching Backgammon Program, Achieves Master-Level Play , 1994, Neural Computation.

[7]  Nicolas Smith,et al.  Computational methods to reduce uncertainty in the estimation of cardiac conduction properties from electroanatomical recordings , 2014, Medical Image Anal..

[8]  P Moireau,et al.  Estimation of tissue contractility from cardiac cine-MRI using a biomechanical heart model , 2012, Biomechanics and modeling in mechanobiology.

[9]  Hervé Delingette,et al.  Cardiac Electrophysiological Activation Pattern Estimation From Images Using a Patient-Specific Database of Synthetic Image Sequences , 2014, IEEE Transactions on Biomedical Engineering.

[10]  Dorin Comaniciu,et al.  Learning-Based Detection and Tracking in Medical Imaging: A Probabilistic Approach , 2013 .

[11]  Long-Ji Lin,et al.  Reinforcement learning for robots using neural networks , 1992 .

[12]  Andrew D. McCulloch,et al.  A Computational Framework for Patient-Specific Multi-Scale Cardiac Modeling , 2010 .

[13]  Nico H. L. Kuijpers,et al.  Modeling Cardiac Electromechanics and Mechanoelectrical Coupling in Dyssynchronous and Failing Hearts , 2012, Journal of Cardiovascular Translational Research.

[14]  Hervé Delingette,et al.  Personalization of Cardiac Motion and Contractility From Images Using Variational Data Assimilation , 2012, IEEE Transactions on Biomedical Engineering.

[15]  Karen S. Frese,et al.  Towards Personalized Cardiology: Multi-Scale Modeling of the Failing Heart , 2015, PloS one.

[16]  Nicolas Smith,et al.  A Comparative Study of Graph-Based, Eikonal, and Monodomain Simulations for the Estimation of Cardiac Activation Times , 2012, IEEE Transactions on Biomedical Engineering.

[17]  R. Bellman Dynamic programming. , 1957, Science.

[18]  Andrew W. Moore,et al.  Reinforcement Learning: A Survey , 1996, J. Artif. Intell. Res..

[19]  M. Powell The BOBYQA algorithm for bound constrained optimization without derivatives , 2009 .

[20]  H. H. Rosenbrock,et al.  An Automatic Method for Finding the Greatest or Least Value of a Function , 1960, Comput. J..

[21]  Matthew Crosby,et al.  Association for the Advancement of Artificial Intelligence , 2014 .

[22]  Danijel Skocaj,et al.  Multivariate online kernel density estimation with Gaussian kernels , 2011, Pattern Recognit..

[23]  Shane Legg,et al.  Massively Parallel Methods for Deep Reinforcement Learning , 2015, ArXiv.

[24]  Hugo A. Katus,et al.  Robust Image-Based Estimation of Cardiac Tissue Parameters and Their Uncertainty from Noisy Data , 2014, MICCAI.

[25]  Adarsh Krishnamurthy,et al.  Patient-specific models of cardiac biomechanics , 2013, J. Comput. Phys..

[26]  N. Westerhof,et al.  An artificial arterial system for pumping hearts. , 1971, Journal of applied physiology.

[27]  Alejandro F. Frangi,et al.  Three-dimensional modeling for functional analysis of cardiac images, a review , 2001, IEEE Transactions on Medical Imaging.

[28]  Shane Legg,et al.  Human-level control through deep reinforcement learning , 2015, Nature.

[29]  Robert Babuska,et al.  Experience Replay for Real-Time Reinforcement Learning Control , 2012, IEEE Transactions on Systems, Man, and Cybernetics, Part C (Applications and Reviews).

[30]  P. Hunter,et al.  Integration from proteins to organs: the Physiome Project , 2003, Nature Reviews Molecular Cell Biology.

[31]  Hervé Delingette,et al.  Current-Based 4D Shape Analysis for the Mechanical Personalization of Heart Models , 2012, MCV.

[32]  Peter Hunter,et al.  Integration from proteins to organs: the IUPS Physiome Project , 2005, Mechanisms of Ageing and Development.

[33]  Jeffrey C. Lagarias,et al.  Convergence Properties of the Nelder-Mead Simplex Method in Low Dimensions , 1998, SIAM J. Optim..

[34]  Nasser M. Nasrabadi,et al.  Pattern Recognition and Machine Learning , 2006, Technometrics.

[35]  Hugo A. Katus,et al.  Automatic image-to-model framework for patient-specific electromechanical modeling of the heart , 2014, 2014 IEEE 11th International Symposium on Biomedical Imaging (ISBI).

[36]  Dorin Comaniciu,et al.  Four-Chamber Heart Modeling and Automatic Segmentation for 3-D Cardiac CT Volumes Using Marginal Space Learning and Steerable Features , 2008, IEEE Transactions on Medical Imaging.

[37]  D. Noble Modeling the Heart--from Genes to Cells to the Whole Organ , 2002, Science.

[38]  P J Hunter,et al.  Myocardial material parameter estimation-a comparative study for simple shear. , 2006, Journal of biomechanical engineering.

[39]  H Zhang,et al.  Models of cardiac tissue electrophysiology: progress, challenges and open questions. , 2011, Progress in biophysics and molecular biology.

[40]  Oliver Kroemer,et al.  Learning to select and generalize striking movements in robot table tennis , 2012, AAAI Fall Symposium: Robots Learning Interactively from Human Teachers.

[41]  Nassir Navab,et al.  Data-driven estimation of cardiac electrical diffusivity from 12-lead ECG signals , 2014, Medical Image Anal..

[42]  Roy C. P. Kerckhoffs,et al.  Patient-specific modeling of dyssynchronous heart failure: a case study. , 2011, Progress in biophysics and molecular biology.

[43]  Roy C. P. Kerckhoffs,et al.  Cardiac resynchronization: insight from experimental and computational models. , 2008, Progress in biophysics and molecular biology.

[44]  Richard S. Sutton,et al.  Reinforcement Learning: An Introduction , 1998, IEEE Trans. Neural Networks.

[45]  Dorin Comaniciu,et al.  Model based non-invasive estimation of PV loop from echocardiography , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[46]  Kevin F. Augenstein,et al.  Method and apparatus for soft tissue material parameter estimation using tissue tagged Magnetic Resonance Imaging. , 2005, Journal of biomechanical engineering.

[47]  Evon M. O. Abu-Taieh,et al.  Comparative Study , 2020, Definitions.

[48]  Sébastien Ourselin,et al.  The estimation of patient-specific cardiac diastolic functions from clinical measurements , 2012, Medical Image Anal..

[49]  Alejandro F. Frangi,et al.  Personalization of a cardiac electromechanical model using reduced order unscented Kalman filtering from regional volumes , 2013, Medical Image Anal..

[50]  Jan Peters,et al.  Model learning for robot control: a survey , 2011, Cognitive Processing.

[51]  Branislav Kveton,et al.  Kernel-Based Reinforcement Learning on Representative States , 2012, AAAI.

[52]  Hervé Delingette,et al.  Efficient probabilistic model personalization integrating uncertainty on data and parameters: Application to eikonal-diffusion models in cardiac electrophysiology. , 2011, Progress in biophysics and molecular biology.

[53]  Jan Haas,et al.  Estimation of Regional Electrical Properties of the Heart from 12-Lead ECG and Images , 2014, STACOM.

[54]  Hervé Delingette,et al.  Velocity-based cardiac contractility personalization from images using derivative-free optimization. , 2015, Journal of the mechanical behavior of biomedical materials.

[55]  R. Kerckhoffs,et al.  Patient specific modeling of the cardiovascular system , 2010 .

[56]  André da Motta Salles Barreto,et al.  Practical Kernel-Based Reinforcement Learning , 2014, J. Mach. Learn. Res..