Abstract:Automatic diagnosis, as an important research direction in artificial intelligence engineering, has advanced significantly in recent years. In real diagnostic scenarios, after the patient informs the doctor about their most obvious symptoms, the doctor is often required to guide the patient to uncover potential symptom information. The doctor then makes the final diagnosis after obtaining sufficient evidence. Previous work has modeled this as a sequential decision-making process and trained an optimal policy using RL-based methods, achieving a high diagnosis correctness rate. However, the RL-based method only finds symptom sequences that improve reward through random trials, failing to directly capture the inter-relationships between symptom sequences in the inquiry process. Also, the doctors’ diagnostic experience, especially symptoms that often co-occur and are located in close proximity to each other in the doctors’ inquiry symptom sequence, has not been well explored in previous implementations. To address this, we propose a model that further enhances the learning of doctors’ diagnostic experience based on a two-part transformer model that completes symptom inquiry and disease diagnosis separately. We introduce a Doctor Diagnosis Experience Reinforcement Module (DDEM) to the symptom sequence inquiry part, reducing the model’s sensitivity to the order of symptom sequences within the same symptom cluster. Then, we further enhance the learning of this symptom distribution by corresponding reinforcement learning reward rules. Also, we propose three new transition rules when transitioning the symptom inquiry model to the disease diagnosis model to enable the symptom inquiry model and the disease diagnosis model to collaborate in a manner that closely aligns with the diagnostic logic used by doctors. Experiments on three public real-world medical dialogue datasets demonstrate that the proposed model improves diagnostic accuracy by 2%, 1.4%, and 0.7% and shows a clear advantage in symptom recall rate, highlighting its effectiveness.

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

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

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

Hierarchical Reinforcement Learning for Automatic Disease Diagnosis

Task-oriented Dialogue System for Automatic Diagnosis.

Conversational Disease Diagnosis via External Planner-Controlled Large Language Models

Efficient Evidence-Based Dialogue System for Medical Diagnosis.

“My Nose is Running.” “are You Also Coughing?”: Building A Medical Diagnosis Agent with Interpretable Inquiry Logics

Exploring LLM-based Data Annotation Strategies for Medical Dialogue Preference Alignment

Exploring a Big Data-Driven Dialogue Generation Method for Disease Queries Using RASA

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

ChatDoctor: A Medical Chat Model Fine-tuned on LLaMA Model using Medical Domain Knowledge

Towards a Reliable and Robust Dialogue System for Medical Automatic Diagnosis

"My Nose is Running.""are You Also Coughing?": Building A Medical Diagnosis Agent with Interpretable Inquiry Logics

Towards Conversational Diagnostic AI

ChatDoctor: A Medical Chat Model Fine-Tuned on a Large Language Model Meta-AI (LLaMA) Using Medical Domain Knowledge

Improving Mild Cognitive Impairment Prediction via Reinforcement Learning and Dialogue Simulation

Identification of pediatric respiratory diseases using a fine-grained diagnosis system

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

A Doctor’s Diagnosis Experience Enhanced Transformer Model for Automatic Diagnosis