Improving Mild Cognitive Impairment Prediction via Reinforcement Learning and Dialogue Simulation
By Fengyi Tang, Kaixiang Lin, Ikechukwu Uchendu, Hiroko H. Dodge, Jiayu Zhou. 2018
9 citations recorded by the source corpus.
Mild cognitive impairment (MCI) is a prodromal phase in the progression from normal aging to dementia, especially Alzheimers disease. Even though there is mild cognitive decline in MCI patients, they have normal overall cognition and thus is challenging to distinguish from normal aging. Using transcribed data obtained from recorded conversational interactions between participants and trained interviewers, and applying supervised learning models to these data, a recent clinical trial has shown a promising result in differentiating MCI from normal aging. However, the substantial amount of interactions with medical staff can still incur significant medical care expenses in practice. In this paper, we propose a novel reinforcement learning (RL) framework to train an efficient dialogue agent on existing transcripts from clinical trials. Specifically, the agent is trained to sketch disease-specific lexical probability distribution, and thus to converse in a way that maximizes the diagnosis accuracy and minimizes the number of conversation turns. We evaluate the performance of the proposed reinforcement learning framework on the MCI diagnosis from a real clinical trial. The results show that while using only a few turns of conversation, our framework can significantly outperform state-of-the-art supervised learning approaches.
How researchers see this paper
The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.
- Chao Yu
- Jiming Liu
- Shamim Nemati
- Anita M. Preininger
- Fei Wang
- Guoqi Ren
- Wei Zhang
- Aliaksei Sandryhaila
Citations in the corpus
- Reinforcement Learning in Healthcare: A Survey (2019, methodology): “[214] applied the DQN framework to train an efficient dialogue agent to sketch a disease-specific lexical probability distribution and thus to converse in a way that maximizes the diagnosis accuracy and minimizes the number of conversation turns.”
- AI in Health: State of the Art, Challenges, and Future Directions (2019, methodology): “[64] built a conversational agent based on transcripts from these clinical trials using reinforcement learning techniques [65].”