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.

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

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

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

A doctor's diagnosis experience enhanced transformer model for automatic diagnosis

Diaformer: Automatic Diagnosis via Symptoms Sequence Generation

BSODA: A Bipartite Scalable Framework for Online Disease Diagnosis

MTDiag: An Effective Multi-Task Framework for Automatic Diagnosis

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

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

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

Efficient Evidence-Based Dialogue System for Medical Diagnosis.

Towards a Reliable and Robust Dialogue System for Medical Automatic Diagnosis

SADE: A Speaker-Aware Dual Encoding Model Based on Diagbert for Medical Triage and Pre-Diagnosis.

Task-oriented Dialogue System for Automatic Diagnosis.

Improved ensemble learning in fault diagnosis system

Hierarchical Reinforcement Learning for Automatic Disease Diagnosis

DDxT: Deep Generative Transformer Models for Differential Diagnosis

Disease Inference With Symptom Extraction And Bidirectional Recurrent Neural Network

EIRAD: an Evidence-based Dialogue System with Highly Interpretable Reasoning Path for Automatic Diagnosis

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

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