Refuel: Exploring sparse features in deep reinforcement learning for fast disease diagnosis
Journal
Advances in Neural Information Processing Systems
Journal Volume
2018-December
Pages
7322-7331
Date Issued
2018
Author(s)
Abstract
This paper proposes REFUEL, a reinforcement learning method with two techniques: reward shaping and feature rebuilding, to improve the performance of online symptom checking for disease diagnosis. Reward shaping can guide the search of policy towards better directions. Feature rebuilding can guide the agent to learn correlations between features. Together, they can find symptom queries that can yield positive responses from a patient with high probability. Experimental results justify that the two techniques in REFUEL allow the symptom checker to identify the disease more rapidly and accurately. ? 2018 Curran Associates Inc.All rights reserved.
Subjects
Diagnosis; Machine learning; Reinforcement learning; Disease diagnosis; High probability; Reinforcement learning method; Sparse features; Deep learning
Type
conference paper