Deep learning-based ECG classification using a TensorFlow Lite model

Abstract

The number of IoT devices in healthcare is expected to rise sharply due to significantly increased demand since the COVID-19 pandemic. Deep learning and IoT devices are being employed to monitor body vitals and automate anomaly detection in clinical and non-clinical settings. Most of the current technology requires the transmission of raw data to a remote server, which is not efficient for resource-constrained IoT devices and embedded systems. In this work, we have developed machine learning models to be deployed on Raspberry Pi. We present an evaluation of our TensorFlow Model with various classification classes. We also present the evaluation of the corresponding TensorFlow Lite FlatBuffers to demonstrate their minimal run-time requirements while maintaining acceptable accuracy. Additionally, to address the problem of sensor and data integration when using multiple devices, we propose a unified server on our Edge Node.