FMM-Head: Enhancing Autoencoder-Based ECG Anomaly Detection with Prior KnowledgeShow others and affiliations
2025 (English)In: Pattern Recognition and Artificial Intelligence - 4th International Conference, ICPRAI 2024, Proceedings, Springer Nature , 2025, p. 18-32Conference paper, Published paper (Refereed)
Abstract [en]
Detecting anomalies in electrocardiogram data is crucial to identify deviations from normal heartbeat patterns and provide timely intervention to at-risk patients. Various AutoEncoder models (AE) have been proposed to tackle the anomaly detection task with machine learning (ML). However, these models do not explicitly consider the specific patterns of ECG leads, thus compromising learning efficiency. In contrast, we replace the decoding part of the AE with a reconstruction head (namely, FMM-Head) based on prior knowledge of the ECG shape. Our model consistently achieves higher anomaly detection capabilities than state-of-the-art models, up to 0.31 increase in area under the ROC curve (AUROC), with as little as half the original model size and explainable extracted features. The processing time of our model is four orders of magnitude lower than solving an optimization problem to obtain the same parameters, thus making it suitable for real-time ECG parameters extraction and anomaly detection. The code is available at: https://github.com/giacomoverardo/FMM-Head.
Place, publisher, year, edition, pages
Springer Nature , 2025. p. 18-32
Keywords [en]
AutoEncoders, ECG anomaly detection, Machine Learning
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-361152DOI: 10.1007/978-981-97-8702-9_2Scopus ID: 2-s2.0-85219192392OAI: oai:DiVA.org:kth-361152DiVA, id: diva2:1944107
Conference
4th International Conference on Pattern Recognition and Artificial Intelligence, ICPRAI 2024, Jeju Island, Korea, Jul 3 2024 - Jul 6 2024
Note
Part of ISBN 9789819787012
QC 20250313
2025-03-122025-03-122025-03-13Bibliographically approved