A Two-Stage ECG Classifier for Decentralized Inferencing Across Edge-Cloud Continuum
Li Xiaolin,Barry Cardiff,Deepu John
DOI: https://doi.org/10.1109/jsen.2024.3406780
IF: 4.3
2024-01-01
IEEE Sensors Journal
Abstract:In this article, we propose a multistage electrocardiogram (ECG) classifier for distributed machine learning (ML) inferencing across the edge-cloud continuum for wearable systems. Traditionally, biomedical data acquired from wearable systems are processed in one step, using a single-stage classifier deployed either on a cloud or on an edge device. Though there are merits, both approaches have limitations that relate to computational complexity, network connectivity, and so on. In this work, we propose a multistage, cascaded, ensemble classifier that aims to address these limitations by decentralizing the processing steps, while achieving good overall performance. We employed low-complexity, explainable boosting machines (EBMs) and convolutional neural networks (CNNs) to develop a multistage distributed ECG classifier, which achieves a high-sensitivity binary classification on the edge device and a more comprehensive multiclass classification on the cloud. In standalone performance evaluation using the MIT-BIH Arrhythmia database, the Stage-1 EBM classifier and Stage-2 CNN classifier achieved a maximum accuracy, sensitivity of 96.71%, 96.76%, and 99.49%, 98.19%, respectively. Furthermore, the distributed multistage classifier achieved a maximum cumulative binary classification accuracy, sensitivity of 99.64%, 99.01%, and multiclass classification accuracy, sensitivity of 99.56%, 98.79% when DtC equals 40%. Furthermore, we evaluated the use of the EBM classifier threshold as a control parameter to dynamically vary the system performance and network traffic based on real-time conditions. We verified the feasibility of the model, calculated the energy consumption, and estimated the latency. When streaming only 40% of the data to the cloud, it will result in 60% latency saving. With the proposed technique, energy consumption is reduced by approximately three times.