Classification of current density vector map using transformer hybrid residual network

Lihui Zhu, Yunfeng Yang, Wenyue Yu, Qingxin Zeng, Zhan Zheng, Boqiang Lin (), Xiaohu Zhao and Zhenghui Hu

PLOS ONE, 2025, vol. 20, issue 12, 1-19

Abstract: The classification of the current density vector map (CDVM) reconstructed from magnetocardiogram (MCG) is an important indicator for assessing cardiac function and state in clinical diagnosis. Given the limited widespread application of MCG, research on CDVM often encounters challenges such as scarcity of data and difficulties in judgment. There is growing interest in computer-aided methods to assist physicians in analyzing cardiac cases using CDVM. This paper proposes a deep learning-based approachto classify the CDVM. To address the issue of insufficient processed MCG data, data augmentation is carried out by adding noise, making predictions based on auto regressive integrated moving average (ARIMA) model, and utilizing interpolation methods. A transformer hybrid residual network is then employed to classify the CDVM across categories 0 to 4, with transfer learning incorporated into the network structure to initialize model parameters, and the self-attention mechanism of the Transformer enhancing the feature extraction capability. This method achieved a classification accuracy of 97.52%, outperforming previous deep learning approaches, exhibiting both high precision and efficiency. Furthermore, its high scalability ensures that it will continue to meet the evolving needs of physicians as CDVM datasets undergo continuous expansion.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0338189 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 38189&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:0338189

DOI: 10.1371/journal.pone.0338189

Access Statistics for this article

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