STCP-Net: A few-shot learning approach based on spectral-temporal convolution and an improved prototypical network for ECG classification

Jia Li, Dainan Gan, Xiaolin Zhan, Wei Yao, Bin Zou, Bin Zhang, Dong Li, Dan Li and Lin Li

PLOS ONE, 2026, vol. 21, issue 10, 1-17

Abstract: To address the heavy reliance of deep learning models on manually labeled data in ECG classification, we need to tackle two major limitations. First, sample sizes are often limited in clinical settings, making data annotation costly and impractical. Second, existing few-shot learning methods still fall short in two key areas: 1) Existing strategies pay insufficient attention to rare arrhythmia types. This makes it difficult to effectively identify rare clinical cases. 2) Single feature representation fails to fully integrate the global frequency, local time-frequency, and semantic information of ECG signals. This results in incomplete feature representation. To overcome these challenges, we propose a novel few-shot learning method for ECG classification. In terms of feature extraction, a spectral-temporal convolution feature extractor is designed to precisely focus on the key information of ECG by efficiently integrating the global frequency, local time-frequency features and semantic information of the signal. Furthermore, with regard to classification, an improved prototypical network based on cross-class bias diminishing is developed to enhance the recognition performance for rare arrhythmia type samples. The experimental results show that the proposed method achieves significant classification performance across multiple public datasets, especially on Chapman, it attains an average accuracy exceeding 90% in the 2-way 1-shot and 2-way 5-shot tasks.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0357954 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 57954&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0357954

DOI: 10.1371/journal.pone.0357954

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().