Abstract:Disease diagnosis is an essential and critical step in any disease treatment process. Automatic diagnostic testing has gained popularity in recent years due to its scalability, rationality, and efficacy. The major challenges for the diagnosis agent are inevitably large action space (symptoms) and varieties of diseases, which demand either rich domain knowledge or an intelligent learning framework. We propose a novel knowledge-infused context-driven (KI-CD) hierarchical reinforcement learning (HRL) based diagnosis dialogue system, which leverages a bayesian learning-inspired symptom investigation module called potential candidate module (PCM) for aiding context-aware, knowledge grounded symptom investigation. The PCM module serves as a context and knowledge guiding companion for lower-level policies, leveraging current context and disease-symptom knowledge to identify candidate diseases and potential symptoms, and reinforcing the agent for conducting an intelligent and context guided symptom investigation with the information enriched state and an additional critic known as learner critic. The knowledge-guided symptom investigation extracts an adequate set of symptoms for disease identification, whereas the context-aware symptom investigation aspect substantially improves topic (symptom) transition and enhances user experiences. Furthermore, we also propose and incorporate a hierarchical disease classifier (HDC) with the model for alleviating symptom state sparsity issues, which has led to a significant improvement in disease classification accuracy. Experimental results (both quantitatively and qualitatively) on the benchmarked dataset establish the need and efficacy of the proposed framework. The proposed framework outperforms the current state-of-the-art method in all evaluation metrics other than dialogue length (diagnosis success rate, average match rate, symptom identification rate, and disease classification accuracy by 7.1 %, 0.23 %, 19.67 % and 8.04 %, respectively). Although the inclusion of the hierarchical disease classifier (HDC) with the vanilla HRL framework outperforms the baselines and the state-of-the-art model, combining it with the potential candidate module (PCM) degrades the performance, owing to the restriction of candidate disease and potential symptom recommendations to only current disease group.1

A knowledge infused context driven dialogue agent for disease diagnosis using hierarchical reinforcement learning

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

A Knowledge-Enhanced Hierarchical Reinforcement Learning-Based Dialogue System for Automatic Disease Diagnosis

Hierarchical Reinforcement Learning for Automatic Disease Diagnosis

Towards knowledge-infused automated disease diagnosis assistant

Scalable Online Disease Diagnosis via Multi-Model-Fused Actor-Critic Reinforcement Learning

Learning from Failure: Towards Developing a Disease Diagnosis Assistant That Also Learns from Unsuccessful Diagnoses

Symptoms are known by their companies: towards association guided disease diagnosis assistant

End-to-End Knowledge-Routed Relational Dialogue System for Automatic Diagnosis

BSODA: A Bipartite Scalable Framework for Online Disease Diagnosis

RDKG: A Reinforcement Learning Framework for Disease Diagnosis on Knowledge Graph

A Knowledge-Enhanced Disease Diagnosis Method Based on Prompt Learning and BERT Integration

MedKPL: A heterogeneous knowledge enhanced prompt learning framework for transferable diagnosis

DxFormer: a decoupled automatic diagnostic system based on decoder–encoder transformer with dense symptom representations

A chatbot based question and answer system for the auxiliary diagnosis of chronic diseases based on large language model

Improving preliminary clinical diagnosis accuracy through knowledge filtering techniques in consultation dialogues

Advancing Multi-Organ Disease Care: A Hierarchical Multi-Agent Reinforcement Learning Framework

RDguru: A Conversational Intelligent Agent for Rare Diseases

Efficient symptom inquiring and diagnosis via adaptive alignment of reinforcement learning and classification

Building Conversational Diagnosis Systems for Fine-Grained Diseases Using Few Annotated Data.