Virtual disease landscape using mechanics-informed machine learning: Application to esophageal disorders

Esophageal disorders are related to the mechanical properties and function of the esophageal wall. Therefore, to understand the underlying fundamental mechanisms behind various esophageal disorders, it is crucial to map mechanical behavior of the esophageal wall in terms of mechanics-based parameters corresponding to altered bolus transit and increased intrabolus pressure. We present a hybrid framework that combines fluid mechanics and machine learning to identify the underlying physics of various esophageal disorders (motility disorders, eosinophilic esophagitis, reflux disease, scleroderma esophagus) and maps them onto a parameter space which we call the virtual disease landscape (VDL). A one-dimensional inverse model processes the output from an esophageal diagnostic device called the functional lumen imaging probe (FLIP) to estimate the mechanical “health” of the esophagus by predicting a set of mechanics-based parameters such as esophageal wall stiffness, muscle contraction pattern and active relaxation of esophageal wall. The mechanics-based parameters were then used to train a neural network that consists of a variational autoencoder that generated a latent space and a side network that predicted mechanical work metrics for estimating esophagogastric junction motility. The latent vectors along with a set of discrete mechanics-based parameters define the VDL and formed clusters corresponding to specific esophageal disorders. The VDL not only distinguishes among disorders but also displayed disease progression over time. Finally, we demonstrated the clinical applicability of this framework for estimating the effectiveness of a treatment and tracking patients' condition after a treatment.