Machine learning augmented reduced-order models for FFR-prediction

Abstract

Computational predictions in cardiovascular medicine have largely relied on explicit models derived from physical and physiological principles. Recently, the application of artificial intelligence in cardiovascular medicine has grown substantially. However, successful application of purely data-driven methods requires a sufficiently large and rich dataset. An alternative to purely data-driven methods is to incorporate prior physics-based knowledge into the learning process to reduce the amount and quality of data necessary for a performant model. We analyzed the benefit of this alternative for prediction of pressure and flow in pathological coronary arteries. We trained fully-connected feed forward neural networks (NN) to predict pressure losses in coronary arteries. The training and test data were obtained by solving the 3D incompressible Navier–Stokes (3DiNS) equations. The coronary flow and various geometrical data were used as inputs to train a purely data-driven NN. We investigated two methods for incorporation of prior physics-based knowledge from a reduced-order model (ROM) into NNs that predicted pressure losses across stenotic and healthy coronary segments. First, we trained NNs to predict the discrepancy between the ROM and 3DiNS pressure loss. Second, we augmented the data by including the ROM pressure loss prediction as an input feature to a NN that predicted 3DiNS pressure. Both approaches for incorporation of prior knowledge from the ROM significantly improved prediction of pressure losses across healthy and stenotic segments relative to the purely data-driven approach, especially for lower amounts of data. The incorporation of NN predictions of coronary segment pressure losses in a coronary network model resulted in Fractional Flow Reserve (FFR) predictions with error standard deviation of 0.021 with respect to 3DiNS FFR. In comparison, the standard deviation of repeated FFR measurements is 0.018.