Reinforcement Learning based Disease Progression Model for Alzheimer's
Disease
- URL: http://arxiv.org/abs/2106.16187v1
- Date: Wed, 30 Jun 2021 16:32:12 GMT
- Title: Reinforcement Learning based Disease Progression Model for Alzheimer's
Disease
- Authors: Krishnakant V. Saboo, Anirudh Choudhary, Yurui Cao, Gregory A.
Worrell, David T. Jones and Ravishankar K. Iyer
- Abstract summary: We model Alzheimer's disease (AD) progression by combining differential equations (DEs) and reinforcement learning (RL) with domain knowledge.
We use our model consisting of DEs (as a simulator) and the trained RL agent to predict individualized 10-year AD progression.
Our framework combines DEs with RL for modelling AD progression and has broad applicability for understanding other neurological disorders.
- Score: 3.1224202646855894
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: We model Alzheimer's disease (AD) progression by combining differential
equations (DEs) and reinforcement learning (RL) with domain knowledge. DEs
provide relationships between some, but not all, factors relevant to AD. We
assume that the missing relationships must satisfy general criteria about the
working of the brain, for e.g., maximizing cognition while minimizing the cost
of supporting cognition. This allows us to extract the missing relationships by
using RL to optimize an objective (reward) function that captures the above
criteria. We use our model consisting of DEs (as a simulator) and the trained
RL agent to predict individualized 10-year AD progression using baseline (year
0) features on synthetic and real data. The model was comparable or better at
predicting 10-year cognition trajectories than state-of-the-art learning-based
models. Our interpretable model demonstrated, and provided insights into,
"recovery/compensatory" processes that mitigate the effect of AD, even though
those processes were not explicitly encoded in the model. Our framework
combines DEs with RL for modelling AD progression and has broad applicability
for understanding other neurological disorders.
Related papers
- A Quantitative Approach for Evaluating Disease Focus and Interpretability of Deep Learning Models for Alzheimer's Disease Classification [17.549219224802]
Deep learning (DL) models have shown significant potential in Alzheimer's Disease (AD) classification.
We developed a quantitative disease-focusing strategy to enhance the interpretability of DL models.
We evaluated these models in terms of their abilities to focus on disease-relevant regions.
arXiv Detail & Related papers (2024-09-07T19:16:40Z) - Unifying Interpretability and Explainability for Alzheimer's Disease Progression Prediction [6.582683443485416]
Reinforcement learning has recently shown promise in predicting Alzheimer's disease (AD) progression.
It is not clear which RL algorithms are well-suited for this task.
Our work aims to merge predictive accuracy with transparency, assisting clinicians and researchers in enhancing disease progression modeling.
arXiv Detail & Related papers (2024-06-11T23:54:42Z) - The effect of data augmentation and 3D-CNN depth on Alzheimer's Disease
detection [51.697248252191265]
This work summarizes and strictly observes best practices regarding data handling, experimental design, and model evaluation.
We focus on Alzheimer's Disease (AD) detection, which serves as a paradigmatic example of challenging problem in healthcare.
Within this framework, we train predictive 15 models, considering three different data augmentation strategies and five distinct 3D CNN architectures.
arXiv Detail & Related papers (2023-09-13T10:40:41Z) - Understanding Augmentation-based Self-Supervised Representation Learning
via RKHS Approximation and Regression [53.15502562048627]
Recent work has built the connection between self-supervised learning and the approximation of the top eigenspace of a graph Laplacian operator.
This work delves into a statistical analysis of augmentation-based pretraining.
arXiv Detail & Related papers (2023-06-01T15:18:55Z) - Less is More: Mitigate Spurious Correlations for Open-Domain Dialogue
Response Generation Models by Causal Discovery [52.95935278819512]
We conduct the first study on spurious correlations for open-domain response generation models based on a corpus CGDIALOG curated in our work.
Inspired by causal discovery algorithms, we propose a novel model-agnostic method for training and inference of response generation model.
arXiv Detail & Related papers (2023-03-02T06:33:48Z) - Robust Alzheimer's Progression Modeling using Cross-Domain
Self-Supervised Deep Learning [3.0948853907734044]
We develop a cross-domain self-supervised learning approach for disease prognostic modeling as a regression problem using medical images as input.
We demonstrate that self-supervised pretraining can improve the prediction of Alzheimer's Disease progression from brain MRI.
We also show that pretraining on extended (but not labeled) brain MRI data outperforms pretraining on natural images.
arXiv Detail & Related papers (2022-11-15T23:04:15Z) - Normative Modeling via Conditional Variational Autoencoder and
Adversarial Learning to Identify Brain Dysfunction in Alzheimer's Disease [10.302206705998563]
We propose a novel normative modeling method by combining conditional variational autoencoder with adversarial learning (ACVAE) to identify brain dysfunction in Alzheimer's Disease (AD)
Our experiments on OASIS-3 database show that the deviation maps generated by our model exhibit higher sensitivity to AD compared to other deep normative models.
arXiv Detail & Related papers (2022-11-13T07:36:30Z) - Deep Stable Representation Learning on Electronic Health Records [8.256340233221112]
Causal Healthcare Embedding (CHE) aims at eliminating the spurious statistical relationship by removing the dependencies between diagnoses and procedures.
Our proposed CHE method can be used as a flexible plug-and-play module that can enhance existing deep learning models on EHR.
arXiv Detail & Related papers (2022-09-03T04:10:45Z) - Mixed Effects Neural ODE: A Variational Approximation for Analyzing the
Dynamics of Panel Data [50.23363975709122]
We propose a probabilistic model called ME-NODE to incorporate (fixed + random) mixed effects for analyzing panel data.
We show that our model can be derived using smooth approximations of SDEs provided by the Wong-Zakai theorem.
We then derive Evidence Based Lower Bounds for ME-NODE, and develop (efficient) training algorithms.
arXiv Detail & Related papers (2022-02-18T22:41:51Z) - A multi-stage machine learning model on diagnosis of esophageal
manometry [50.591267188664666]
The framework includes deep-learning models at the swallow-level stage and feature-based machine learning models at the study-level stage.
This is the first artificial-intelligence-style model to automatically predict CC diagnosis of HRM study from raw multi-swallow data.
arXiv Detail & Related papers (2021-06-25T20:09:23Z) - Efficient Model-Based Reinforcement Learning through Optimistic Policy
Search and Planning [93.1435980666675]
We show how optimistic exploration can be easily combined with state-of-the-art reinforcement learning algorithms.
Our experiments demonstrate that optimistic exploration significantly speeds-up learning when there are penalties on actions.
arXiv Detail & Related papers (2020-06-15T18:37:38Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.