Edge AI Model Deployed for Real-Time Detection of Atrial Fibrillation Risk during Sinus Rhythm
Hongmin Wu, Takumi Sawada, Takafumi Goto, Tatsuya Yoneyama, Tetsuo Sasano, Ken Asada- General Medicine
Objectives: The study aimed to develop a deep learning-based edge AI model deployed on electrocardiograph (ECG) devices for the real-time detection of atrial fibrillation (AF) risk during sinus rhythm (SR) using standard 10 s, 12-lead electrocardiograms (ECGs). Methods: A novel approach was used to convert standard 12-lead ECGs into binary images for model input, and a lightweight convolutional neural network (CNN)-based model was trained using data collected by the Japan Agency for Medical and Research Development (AMED) between 2019 and 2022. Patients over 40 years old with digital, SR ECGs were retrospectively enrolled and divided into AF and non-AF groups. The data labeling was supervised by cardiologists. The dataset was randomly allocated into training, validation, and internal testing datasets. External testing was conducted on data collected from other hospitals. Results: The best-trained model achieved an AUC of 0.82 and 0.80, sensitivity of 79.5% and 72.3%, specificity of 77.8% and 77.7%, precision of 78.2% and 76.4%, and overall accuracy of 78.6% and 75.0% in the internal and external testing datasets, respectively. The deployed model and app package utilized 2.5 MB and 40 MB of the available ROM and RAM capacity on the edge ECG device, correspondingly. The processing time for AF risk detection was approximately 2 s. Conclusions: The model maintains comparable performance and improves its suitability for deployment on resource-constrained ECG devices, thereby expanding its potential impact to a wide range of healthcare settings. Its successful deployment enables real-time AF risk detection during SR, allowing for timely intervention to prevent AF-related serious consequences like stroke and premature death.