Causal discovery approach with reinforcement learning for risk factors of type II diabetes mellitus

Abstract

Background: Statistical correlation analysis is currently the most typically used approach for investigating the risk factors of type 2 diabetes mellitus (T2DM). However, this approach does not readily reveal the causal relationships between risk factors and rarely describes the causal relationships visually.

Results: Considering the superiority of reinforcement learning in prediction, a causal discovery approach with reinforcement learning for T2DM risk factors is proposed herein. First, a reinforcement learning model is constructed for T2DM risk factors. Second, the process involved in the causal discovery method for T2DM risk factors is detailed. Finally, several experiments are designed based on diabetes datasets and used to verify the proposed approach.

Conclusions: The experimental results show that the proposed approach improves the accuracy of causality mining between T2DM risk factors and provides new evidence to researchers engaged in T2DM prevention and treatment research.

Keywords: Reinforcement learning; Risk factors; Type 2 diabetes mellitus (T2DM).