Neural Network Pruning for ECG Arrhythmia Classification

<p>Convolutional Neural Networks (CNNs) are a widely accepted means of solving complex classification and detection problems in imaging and speech. However, problem complexity often leads to considerable increases in computation and parameter storage costs. Many successful attempts have been made in effectively reducing these overheads by pruning and compressing large CNNs with only a slight decline in model accuracy. In this study, two pruning methods are implemented and compared on the CIFAR-10 database and an ECG arrhythmia classification task. Each pruning method employs a pruning phase interleaved with a finetuning phase. It is shown that when performing the scale-factor pruning algorithm on ECG, finetuning time can be expedited by 1.4 times over the traditional approach with only 10% of expensive floating-point operations retained, while experiencing no significant impact on accuracy.</p>

Name thesis:degree_name

MS in Electrical Engineering

Discipline thesis:degree_discipline

Electrical Engineering

Year dc:date.available

2020

Author dc:creator

• Labarge, Isaac E

Contributors dc:contributor

• Xia-Hua Yu
• Electrical Engineering
• College of Engineering

• Convolutional Neural Network
• Pruning
• ECG
• Artificial Intelligence and Robotics
• Biomedical
• Data Science
• Software Engineering
• Theory and Algorithms

Identifier

10.15368/theses.2020.76

OAI identifier oai:identifier

oai:digitalcommons.calpoly.edu:theses-3684