Logo: to the web site of Uppsala University

uu.sePublications from Uppsala University
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Transferability and Adversarial Training in Automatic Classification of the Electrocardiogram with Deep Learning
Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Division of Systems and Control. Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Automatic control. Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Artificial Intelligence.ORCID iD: 0000-0001-5183-234X
Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Division of Systems and Control. Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Automatic control. Uppsala University, Disciplinary Domain of Science and Technology, Mathematics and Computer Science, Department of Information Technology, Artificial Intelligence.ORCID iD: 0000-0003-3632-8529
2024 (English)In: Computers in cardiology, ISSN 0276-6574, Vol. 51Article in journal (Refereed) Published
Abstract [en]

Automatic electrocardiogram (ECG) analysis using deep neural networks has seen promising results in recent years. However, these models are susceptible to domain shifts when they are applied to new data distributions not previously trained on. We investigate how training on worst-case artificial samples through adversarial training can help promote models that can easily be molded through fine-tuning to new datasets. We compare the area under the precision-recall curve (AUPRC) for the classification of atrial fibrillation using two cohorts: we use PTB-XL for training and CODE-15% for fine-tuning and evaluating the models. Our results show that adversarially trained models on ECG data yield higher transferability when fine-tuned on new datasets compared to normally trained models (0.732 vs. 0.685 AUPRC). We also note that they even have the ability to supersede models solely trained on the new dataset using more total time and data (0.732 vs. 0.685 AUPRC). Our work thus paves the way for the training of general models that can be applied to different types of new settings with high performance. © 2024 IEEE Computer Society. All rights reserved.

Place, publisher, year, edition, pages
Computing in Cardiology , 2024. Vol. 51
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:uu:diva-573254DOI: 10.22489/CinC.2024.096Scopus ID: 2-s2.0-105028380449OAI: oai:DiVA.org:uu-573254DiVA, id: diva2:2021073
Available from: 2025-12-12 Created: 2025-12-12 Last updated: 2026-06-01Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Schön, Thomas B.Horta Ribeiro, Antônio

Search in DiVA

By author/editor
Schön, Thomas B.Horta Ribeiro, Antônio
By organisation
Division of Systems and ControlAutomatic controlArtificial Intelligence
In the same journal
Computers in cardiology
Computer Sciences

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 13 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf