Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device
Arrhythmia can lead to severe complications and early detection of arrhythmia is crucial to prevent progression. Electrocardiograms are the most reliable measure for detecting arrhythmia. This study aims to implement a practical ultra-edge-computing system for automated arrhythmia classification, in...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2024-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10704628/ |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850210065499291648 |
|---|---|
| author | Namho Kim Seongjae Lee Seungmin Kim Sung-Min Park |
| author_facet | Namho Kim Seongjae Lee Seungmin Kim Sung-Min Park |
| author_sort | Namho Kim |
| collection | DOAJ |
| description | Arrhythmia can lead to severe complications and early detection of arrhythmia is crucial to prevent progression. Electrocardiograms are the most reliable measure for detecting arrhythmia. This study aims to implement a practical ultra-edge-computing system for automated arrhythmia classification, incorporating a lightweight deep neural network-based model and low-power wearable electrocardiogram sensing device. The lightweight model was designed based on the WavelNet architecture, which has been previously introduced for precise arrhythmia classification. Model compression methods including knowledge distillation, pruning, and quantization were employed to enhance arrhythmia classification performance while reducing computational complexity. The lightweight model was integrated into a wearable sensing device featuring a resource-constrained microcontroller unit. The efficiency of the lightweight model and edge-computing system was evaluated regarding arrhythmia classification performance and computational complexity. This study was conducted using a widely used publicly accessible database following a benchmark training and evaluation procedure. Compared to a standard convolutional neural network-based model which exhibited 81.5% overall accuracy, the proposed lightweight model achieved more precise arrhythmia classification with achieving 87.1% overall accuracy. The lightweight model also exhibited significantly reduced computational complexity, with 51.5% fewer parameters, 78.1% smaller model size, 40.3% fewer multiply-accumulate operations, and 79.6% reduced inference time. The completed edge-computing system featured sufficiently short inference time and low memory usage. The proposed lightweight model and ultra-edge-computing system demonstrated advanced performance in classifying arrhythmia, with a significantly reduced computational burden. This outcome ensures the practicality of the system to achieve on-device real-time arrhythmia classification in the real-world. |
| format | Article |
| id | doaj-art-c2fbe55dff83459e8d4d0e4a319d8ae0 |
| institution | OA Journals |
| issn | 2169-3536 |
| language | English |
| publishDate | 2024-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj-art-c2fbe55dff83459e8d4d0e4a319d8ae02025-08-20T02:09:51ZengIEEEIEEE Access2169-35362024-01-011215054615056310.1109/ACCESS.2024.347332310704628Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge DeviceNamho Kim0Seongjae Lee1https://orcid.org/0009-0002-4325-2518Seungmin Kim2Sung-Min Park3https://orcid.org/0000-0002-8359-8110Department of Convergence IT Engineering, Pohang University of Science and Technology (POSTECH), Pohang-si, Republic of KoreaMajor in Medical Science and Engineering, Pohang University of Science and Technology (POSTECH), Pohang-si, Republic of KoreaMajor in Medical Science and Engineering, Pohang University of Science and Technology (POSTECH), Pohang-si, Republic of KoreaDepartment of Convergence IT Engineering, Pohang University of Science and Technology (POSTECH), Pohang-si, Republic of KoreaArrhythmia can lead to severe complications and early detection of arrhythmia is crucial to prevent progression. Electrocardiograms are the most reliable measure for detecting arrhythmia. This study aims to implement a practical ultra-edge-computing system for automated arrhythmia classification, incorporating a lightweight deep neural network-based model and low-power wearable electrocardiogram sensing device. The lightweight model was designed based on the WavelNet architecture, which has been previously introduced for precise arrhythmia classification. Model compression methods including knowledge distillation, pruning, and quantization were employed to enhance arrhythmia classification performance while reducing computational complexity. The lightweight model was integrated into a wearable sensing device featuring a resource-constrained microcontroller unit. The efficiency of the lightweight model and edge-computing system was evaluated regarding arrhythmia classification performance and computational complexity. This study was conducted using a widely used publicly accessible database following a benchmark training and evaluation procedure. Compared to a standard convolutional neural network-based model which exhibited 81.5% overall accuracy, the proposed lightweight model achieved more precise arrhythmia classification with achieving 87.1% overall accuracy. The lightweight model also exhibited significantly reduced computational complexity, with 51.5% fewer parameters, 78.1% smaller model size, 40.3% fewer multiply-accumulate operations, and 79.6% reduced inference time. The completed edge-computing system featured sufficiently short inference time and low memory usage. The proposed lightweight model and ultra-edge-computing system demonstrated advanced performance in classifying arrhythmia, with a significantly reduced computational burden. This outcome ensures the practicality of the system to achieve on-device real-time arrhythmia classification in the real-world.https://ieeexplore.ieee.org/document/10704628/Edge computingknowledge distillationmodel compressionon-device inferencepruning |
| spellingShingle | Namho Kim Seongjae Lee Seungmin Kim Sung-Min Park Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device IEEE Access Edge computing knowledge distillation model compression on-device inference pruning |
| title | Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device |
| title_full | Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device |
| title_fullStr | Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device |
| title_full_unstemmed | Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device |
| title_short | Automated Arrhythmia Classification System: Proof-of-Concept With Lightweight Model on an Ultra-Edge Device |
| title_sort | automated arrhythmia classification system proof of concept with lightweight model on an ultra edge device |
| topic | Edge computing knowledge distillation model compression on-device inference pruning |
| url | https://ieeexplore.ieee.org/document/10704628/ |
| work_keys_str_mv | AT namhokim automatedarrhythmiaclassificationsystemproofofconceptwithlightweightmodelonanultraedgedevice AT seongjaelee automatedarrhythmiaclassificationsystemproofofconceptwithlightweightmodelonanultraedgedevice AT seungminkim automatedarrhythmiaclassificationsystemproofofconceptwithlightweightmodelonanultraedgedevice AT sungminpark automatedarrhythmiaclassificationsystemproofofconceptwithlightweightmodelonanultraedgedevice |