NanoCNN: Minority-Aware Neural Architecture Search for Edge Arrhythmia Classification

Arrhythmia is a life-threatening cardiovascular disease if not detected early. While deep learning models have demonstrated strong performance in ECG-based arrhythmia classification, deploying these models on resource-constrained wearable devices remains challenging. In this paper, we present a quantization-compatible neural architecture search (NAS) framework that discovers ultra-compact minority-aware convolutional neural networks (CNN). We formulate NAS as a multi-objective optimization problem, jointly maximizing balanced accuracy and minority-classes recall while minimizing model size and computational complexity. Furthermore, we constrain our search space to INT8-compatible operations. We evaluated our framework on the MIT-BIH Arrhythmia Database. We discovered NanoCNN models for binary and multi-class classification tasks. The models trained without augmentation achieved 98.7% and 98.21% overall accuracies outperforming the state-of-the-art. The discovered models required 38.3 K and 51.5 K multiply-accumulate operations (MAC) per inference, enabling their deployment on ARM Cortex-M4 microcontrollers. With augmentation and other minority-aware interventions, our model attained 91.6% balanced accuracy. Our results validated the effectiveness of the adopted search and training techniques for arrhythmia screening and diagnosis.