Task-oriented Dialogue System for Automatic Disease Diagnosis via Hierarchical Reinforcement Learning

04/29/2020
by   Kangenbei Liao, et al.
0

In this paper, we focus on automatic disease diagnosis with reinforcement learning (RL) methods in task-oriented dialogues setting. Different from conventional RL tasks, the action space for disease diagnosis (i.e., symptoms) is inevitably large, especially when the number of diseases increases. However, existing approaches to this problem employ a flat RL policy, which typically works well in simple tasks but has significant challenges in complex scenarios like disease diagnosis. Towards this end, we propose to integrate a hierarchical policy of two levels into the dialogue policy learning. The high level policy consists of a model named master that is responsible for triggering a model in low level, the low level policy consists of several symptom checkers and a disease classifier. Experimental results on both self-constructed real-world and synthetic datasets demonstrate that our hierarchical framework achieves higher accuracy in disease diagnosis compared with existing systems. Besides, the datasets (http://www.sdspeople.fudan.edu.cn/zywei/data/Fudan-Medical-Dialogue2.0) and codes (https://github.com/nnbay/MeicalChatbot-HRL) are all available now.

READ FULL TEXT
research
05/08/2022

DxFormer: A Decoupled Automatic Diagnostic System Based on Decoder-Encoder Transformer with Dense Symptom Representations

Diagnosis-oriented dialogue system queries the patient's health conditio...
research
02/20/2023

Deep Reinforcement Learning for Cost-Effective Medical Diagnosis

Dynamic diagnosis is desirable when medical tests are costly or time-con...
research
12/01/2021

Efficient Symptom Inquiring and Diagnosis via Adaptive Alignment of Reinforcement Learning and Classification

The medical automatic diagnosis system aims to imitate human doctors in ...
research
02/18/2018

Improving Mild Cognitive Impairment Prediction via Reinforcement Learning and Dialogue Simulation

Mild cognitive impairment (MCI) is a prodromal phase in the progression ...
research
05/24/2020

GoChat: Goal-oriented Chatbots with Hierarchical Reinforcement Learning

A chatbot that converses like a human should be goal-oriented (i.e., be ...
research
12/02/2020

FIT: a Fast and Accurate Framework for Solving Medical Inquiring and Diagnosing Tasks

Automatic self-diagnosis provides low-cost and accessible healthcare via...
research
07/17/2023

CoAD: Automatic Diagnosis through Symptom and Disease Collaborative Generation

Automatic diagnosis (AD), a critical application of AI in healthcare, em...

Please sign up or login with your details

Forgot password? Click here to reset