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

by   Weijie He, et al.

Automatic self-diagnosis provides low-cost and accessible healthcare via an agent that queries the patient and makes predictions about possible diseases. From a machine learning perspective, symptom-based self-diagnosis can be viewed as a sequential feature selection and classification problem. Reinforcement learning methods have shown good performance in this task but often suffer from large search spaces and costly training. To address these problems, we propose a competitive framework, called FIT, which uses an information-theoretic reward to determine what data to collect next. FIT improves over previous information-based approaches by using a multimodal variational autoencoder (MVAE) model and a two-step sampling strategy for disease prediction. Furthermore, we propose novel methods to substantially reduce the computational cost of FIT to a level that is acceptable for practical online self-diagnosis. Our results in two simulated datasets show that FIT can effectively deal with large search space problems, outperforming existing baselines. Moreover, using two medical datasets, we show that FIT is a competitive alternative in real-world settings.



page 1

page 2

page 3

page 4


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

For those seeking healthcare advice online, AI based dialogue agents cap...

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

The medical automatic diagnosis system aims to imitate human doctors in ...

A Bayesian Approach for Medical Inquiry and Disease Inference in Automated Differential Diagnosis

We propose a Bayesian approach for both medical inquiry and disease infe...

Unsupervised Online Feature Selection for Cost-Sensitive Medical Diagnosis

In medical diagnosis, physicians predict the state of a patient by check...

Diaformer: Automatic Diagnosis via Symptoms Sequence Generation

Automatic diagnosis has attracted increasing attention but remains chall...

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...

Inexpensive Cost-Optimized Measurement Proposal for Sequential Model-Based Diagnosis

In this work we present strategies for (optimal) measurement selection i...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.