• 

  The team

• 

  Showcase

• 

  Hardware Component

• 

  Exciting moment

• 

  Output terminal

ECG_classification

Autoencoder for ECG Classification :heartbeat: :brain: This repository contains code for an autoencoder implemented in Python using TensorFlow for ECG (Electrocardiogram) classification.

What is an Autoencoder? :thinking: An autoencoder is a type of neural network that can be used for unsupervised learning tasks, such as dimensionality reduction or data compression. It consists of two main parts: an encoder and a decoder. The encoder takes an input signal and compresses it into a lower-dimensional representation, called the latent space. The decoder then takes this compressed representation and attempts to reconstruct the original input signal.

Autoencoder Structure :building_construction: The autoencoder implemented in this code consists of the following parts:

Encoder: several convolutional layers followed by a fully connected layer that outputs the compressed representation.

Decoder: several deconvolutional layers followed by a final fully connected layer that produces the reconstructed signal.

Advantages of Using an Autoencoder for ECG Classification :muscle: The advantages of using an autoencoder for ECG classification are as follows:

Unsupervised Learning: An autoencoder can learn to extract useful features from the ECG signal without requiring labeled data.

Data Compression and Denoising: Autoencoders can be used for data compression and denoising, which can improve the performance of downstream tasks such as ECG classification. By compressing the ECG signal into a lower-dimensional representation, the autoencoder can reduce the amount of noise in the signal and remove redundant information, making it easier for a classifier to distinguish between different types of ECG signals.

Results

Loss Accuracy Table

Confusion Matrix

The confusion matrix for the model on the test set is:

Performance Metrics

The following performance metrics were computed for the model:

• Accuracy: 0.9810
• Precision: 0.9771
• Recall: 0.9893
• F1 Score: 0.9831
• Area Under Curve (AUC): 0.9799

The ROC curve for the model is shown below:

ONNX Runtime Docker Image for ARM 64

This Docker image installs ONNX and ONNX Runtime for ARM 64. It is based on the Ubuntu 20.04 operating system and includes the Intel oneAPI Base Toolkit for optimized performance on Intel architectures.

Usage

1. Install Docker: https://www.docker.com/get-started
2. Build the Docker image: docker build -t onnxruntime-arm64 .
3. Run the Docker container: docker run -it onnxruntime-arm64

Hardware:

To detect and correctly amplify heart signal to feed into the AI-based network, a analog sensor is designed based on the classical 3-opamp instramentation.
The overall schematic of the ECG amplifier:

The team also come up with an approximated model for human body, encorporating left and right atrium and a drivien right leg circuit

Dockerfile

The Dockerfile for this image can be found in the official ONNX Runtime GitHub repository. It installs the required dependencies, builds ONNX Runtime from source, and installs it system-wide.

Resources

For more information on using ONNX Runtime, see the official documentation:

• ONNX Runtime GitHub repository
• ONNX Runtime Python API documentation
• ONNX Runtime C++ API documentation

Conclusion :memo:

The implementation of an autoencoder for ECG classification is a promising technique that can be used to extract useful features from ECG signals. By compressing the ECG signal into a lower-dimensional representation, the autoencoder can reduce the amount of noise in the signal and remove redundant information, making it easier for a classifier to distinguish between different types of ECG signals. This technique can be particularly useful when labeled data is scarce or expensive to obtain.

Challenges we ran into

ONNX datatype conversion from float64 to tensor(float) Serial Communications

Accomplishments that we're proud of

Successful building the circuit

What's next for Fear No Age

Update GUI and Android APP

Built With

• arduino
• bandpass
• opamp
• raspberry-pi