Deep Q-Network (DQN) Model for Disease Prediction Using Electronic Health Records (EHRs)

Abstract

Many efforts have proved that deep learning models are effective for disease prediction using electronic health records (EHRs). However, these models are not yet precise enough to predict diseases. Additionally, ethical concerns and the use of clustering and classification algorithms on small datasets limit their effectiveness. The complexity of data processing further complicates the interpretation of patient representation learning models, even though data augmentation strategies may help. Incomplete patient data also hinder model accuracy. This study aims to develop and evaluate a deep learning model that addresses these challenges. Our proposed approach is to design a disease prediction model based on deep Q-learning (DQL), which replaces the traditional Q-learning reinforcement learning algorithm with a neural network deep learning model, and the mapping capabilities of the Q-network are utilized. We conclude that the proposed model achieves the best accuracy (98%) compared with other models.