Discovering a reaction-diffusion model for Alzheimer's disease by combining PINNs with symbolic regression

Misfolded tau proteins play a critical role in the progression and pathology of Alzheimer's disease. Recent studies suggest that the spatio-temporal pattern of misfolded tau follows a reaction-diffusion type equation. However, the precise mathematical model and parameters that characterize the progression of misfolded protein across the brain remain incompletely understood. Here, we use deep learning and artificial intelligence to discover a mathematical model for the progression of Alzheimer's disease using longitudinal tau positron emission tomography from the Alzheimer's Disease Neuroimaging Initiative database. Specifically, we integrate physics informed neural networks (PINNs) and symbolic regression to discover a reaction-diffusion type partial differential equation for tau protein misfolding and spreading. First, we demonstrate the potential of our model and parameter discovery on synthetic data. Then, we apply our method to discover the best model and parameters to explain tau imaging data from 46 individuals who are likely to develop Alzheimer's disease and 30 healthy controls. Our symbolic regression discovers different misfolding models $f(c)$ for two groups, with a faster misfolding for the Alzheimer's group, $f(c) = 0.23c^3 - 1.34c^2 + 1.11c$, than for the healthy control group, $f(c) = -c^3 +0.62c^2 + 0.39c$. Our results suggest that PINNs, supplemented by symbolic regression, can discover a reaction-diffusion type model to explain misfolded tau protein concentrations in Alzheimer's disease. We expect our study to be the starting point for a more holistic analysis to provide image-based technologies for early diagnosis, and ideally early treatment of neurodegeneration in Alzheimer's disease and possibly other misfolding-protein based neurodegenerative disorders.

[1]  G. Karniadakis,et al.  A Generative Modeling Framework for Inferring Families of Biomechanical Constitutive Laws in Data-Sparse Regimes , 2023, Journal of the Mechanics and Physics of Solids.

[2]  G. Karniadakis,et al.  Leveraging Multi-time Hamilton-Jacobi PDEs for Certain Scientific Machine Learning Problems , 2023, ArXiv.

[3]  Zhengbin Xu,et al.  Solving free-surface problems for non-shallow water using boundary and initial conditions-free physics-informed neural network (bif-PINN) , 2023, J. Comput. Phys..

[4]  G. Karniadakis,et al.  L-HYDRA: Multi-Head Physics-Informed Neural Networks , 2023, ArXiv.

[5]  G. Holzapfel,et al.  Automated model discovery for skin: Discovering the best model, data, and experiment , 2022, bioRxiv.

[6]  M. Kohandel,et al.  A PINN Approach to Symbolic Differential Operator Discovery with Sparse Data , 2022, ArXiv.

[7]  K. Linka,et al.  Automated model discovery for human brain using Constitutive Artificial Neural Networks , 2022, bioRxiv.

[8]  C. Canuto,et al.  Enforcing Dirichlet boundary conditions in physics-informed neural networks and variational physics-informed neural networks , 2022, ArXiv.

[9]  K. Linka,et al.  A new family of Constitutive Artificial Neural Networks towards automated model discovery , 2022, ArXiv.

[10]  Apostolos F. Psaros,et al.  NeuralUQ: A comprehensive library for uncertainty quantification in neural differential equations and operators , 2022, ArXiv.

[11]  G. Karniadakis,et al.  Bayesian Physics-Informed Neural Networks for real-world nonlinear dynamical systems , 2022, Computer Methods in Applied Mechanics and Engineering.

[12]  George Em Karniadakis,et al.  Uncertainty Quantification in Scientific Machine Learning: Methods, Metrics, and Comparisons , 2022, J. Comput. Phys..

[13]  M. Peirlinck,et al.  Bayesian Physics-Based Modeling of Tau Propagation in Alzheimer's Disease , 2021, Frontiers in Physiology.

[14]  Minglang Yin,et al.  Physics-informed neural networks (PINNs) for fluid mechanics: a review , 2021, Acta Mechanica Sinica.

[15]  G. Karniadakis,et al.  Identifiability and predictability of integer- and fractional-order epidemiological models using physics-informed neural networks , 2021, Nature Computational Science.

[16]  E. Kuhl,et al.  Network Diffusion Modeling Explains Longitudinal Tau PET Data , 2020, Frontiers in Neuroscience.

[17]  Rui Xu,et al.  Discovering Symbolic Models from Deep Learning with Inductive Biases , 2020, NeurIPS.

[18]  G. Karniadakis,et al.  Conservative physics-informed neural networks on discrete domains for conservation laws: Applications to forward and inverse problems , 2020 .

[19]  Liu Yang,et al.  B-PINNs: Bayesian Physics-Informed Neural Networks for Forward and Inverse PDE Problems with Noisy Data , 2020, J. Comput. Phys..

[20]  G. Karniadakis,et al.  Physics-informed neural networks for high-speed flows , 2020, Computer Methods in Applied Mechanics and Engineering.

[21]  E. Kuhl.  Connectomics of neurodegeneration , 2019, Nature Neuroscience.

[22]  Zhiping Mao,et al.  DeepXDE: A Deep Learning Library for Solving Differential Equations , 2019, AAAI Spring Symposium: MLPS.

[23]  Max Tegmark,et al.  AI Feynman: A physics-inspired method for symbolic regression , 2019, Science Advances.

[24]  Paris Perdikaris,et al.  Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations , 2019, J. Comput. Phys..

[25]  E. Kuhl,et al.  Prion-like spreading of Alzheimer’s disease within the brain’s connectome , 2019, bioRxiv.

[26]  George Em Karniadakis,et al.  fPINNs: Fractional Physics-Informed Neural Networks , 2018, SIAM J. Sci. Comput..

[27]  Charles Blundell,et al.  Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles , 2016, NIPS.

[28]  V. Grolmusz,et al.  Parameterizable consensus connectomes from the Human Connectome Project: the Budapest Reference Connectome Server v3.0 , 2016, Cognitive Neurodynamics.

[29]  Vince Grolmusz,et al.  Parameterizable consensus connectomes from the Human Connectome Project: the Budapest Reference Connectome Server v3.0 , 2016, Cognitive Neurodynamics.

[30]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[31]  Julien Cohen-Adad,et al.  The Human Connectome Project and beyond: Initial applications of 300mT/m gradients , 2013, NeuroImage.

[32]  G. Karniadakis,et al.  Multi-Element Generalized Polynomial Chaos for Arbitrary Probability Measures , 2006, SIAM J. Sci. Comput..

[33]  Jorge Nocedal,et al.  Algorithm 778: L-BFGS-B: Fortran subroutines for large-scale bound-constrained optimization , 1997, TOMS.

[34]  John Whitehead,et al.  Finite bandwidth, finite amplitude convection , 1969, Journal of Fluid Mechanics.

[35]  Lee A. Segel,et al.  Distant side-walls cause slow amplitude modulation of cellular convection , 1969, Journal of Fluid Mechanics.

[36]  E. Parzen.  On Estimation of a Probability Density Function and Mode , 1962 .

[37]  R. Fisher.  THE WAVE OF ADVANCE OF ADVANTAGEOUS GENES , 1937 .

[38]  M. Bourdenx,et al.  [Re] Spread of alpha-synuclein pathology through the brain connectome is modulated by selective vulnerability and predicted by network analysis , 2021 .

[39]  A. Wills,et al.  Physics-informed machine learning , 2021, Nature Reviews Physics.

[40]  A. N. Kolmogorov,et al.  A study of the equation of diffusion with increase in the quantity of matter, and its application to a biological problem , 1937 .