Artificial-intelligence-based risk prediction and mechanism discovery for atrial fibrillation using heart...

Early diagnosis of atrial fibrillation (AF) is important for preventing stroke and other complications. Predicting AF risk in advance can improve early diagnostic efficiency. Deep learning has been used for disease risk prediction; however, it lacks adherence to evidence-based medicine standards. Identifying the underlying mechanisms behind disease risk prediction is important and required.

We developed an explainable deep learning model called HBBI-AI to predict AF risk using only heart beat-to-beat intervals (HBBIs) during sinus rhythm. We proposed a possible AF mechanism based on the model's explainability and verified this conjecture using confirmed AF risk factors while also examining new AF risk factors. Finally, we investigated the changes in clinicians' ability to predict AF risk using only HBBIs before and after learning the model's explainability.

HBBI-AI consistently performed well across large in-house and external public datasets. HBBIs with large changes or extreme stability were critical predictors for increased AF risk, and the underlying cause was autonomic imbalance. We verified various AF risk factors and discovered that autonomic imbalance was associated with all these factors. Finally, cardiologists effectively understood and learned from these findings to improve their abilities in AF risk prediction.

HBBI-AI effectively predicted AF risk using only HBBI information through evaluating autonomic imbalance. Autonomic imbalance may play an important role in many risk factors of AF rather than in a limited number of risk factors.

This study was supported in part by the National Key R&D Program and the National Natural Science Foundation of China.