Comparative Analysis of 1-D and 2-D Deep Convolutional Neural Networks in Magnetocardiogram Classification for Coronary Artery Disease

Jia Yifan1, Cui Yangyang2, Yadan Zhang2, Xiang Min2
1School of Instrumentation and Optoelectronics Engineering, Beihang University, Beiing 100191, China, 2Hangzhou Institute of National Extremely-weak Magnetic Field Infrastructure


Abstract

Coronary artery disease (CAD) imposes a substantial burden on healthcare systems, necessitating accurate diagnosis and effective treatment. Magnetocardiography (MCG) emerges as a promising non-invasive diagnostic tool for CAD. Nevertheless, the conventional approach to CAD diagnosis via one-dimensional (1-D) MCG is subjective and susceptible to inaccuracies. In response, a novel image-based MCG classification method has been introduced, leveraging a two-dimensional convolutional neural network (2-D-CNN) to distinguish MCG signals indicative of healthy cardiovascular function from those associated with CAD pathology. To further enhance the accuracy and robustness of CAD classification via MCG, we compared the classification performance of AlexNet, ResNet, and VGGNet networks using 1-D MCG and 2-D MCG images as inputs, respectively. Subsequently, to mitigate overfitting in 2-D networks, we initialized AlexNet-like, ResNet-like, and VGGNet-like architectures with weights pretrained on ImageNet. Our evaluation on the collected CAD dataset demonstrates that the proposed method achieves an accuracy of 95.2% and the 2-D-CNN outperforms the 1-D MCG, particularly in scenarios with limited data availability. Among the architectures under consideration, ResNet exhibited the most promising performance.