Attention Autoencoder for Generative Latent Representational Learning in Anomaly Detection

Sensors (Basel). 2021 Dec 24;22(1):123. doi: 10.3390/s22010123.

Abstract

Today, accurate and automated abnormality diagnosis and identification have become of paramount importance as they are involved in many critical and life-saving scenarios. To accomplish such frontiers, we propose three artificial intelligence models through the application of deep learning algorithms to analyze and detect anomalies in human heartbeat signals. The three proposed models include an attention autoencoder that maps input data to a lower-dimensional latent representation with maximum feature retention, and a reconstruction decoder with minimum remodeling loss. The autoencoder has an embedded attention module at the bottleneck to learn the salient activations of the encoded distribution. Additionally, a variational autoencoder (VAE) and a long short-term memory (LSTM) network is designed to learn the Gaussian distribution of the generative reconstruction and time-series sequential data analysis. The three proposed models displayed outstanding ability to detect anomalies on the evaluated five thousand electrocardiogram (ECG5000) signals with 99% accuracy and 99.3% precision score in detecting healthy heartbeats from patients with severe congestive heart failure.

Keywords: anomaly detection; attention module; autoencoder; long short-term memory (LSTM); variational autoencoder (VAE).

MeSH terms

  • Algorithms*
  • Artificial Intelligence*
  • Attention
  • Electrocardiography
  • Humans
  • Normal Distribution