arXiv:2604.13359cs.LGcs.AR2026-04

在边缘设备上实现低功耗高精度的生物信号模型微调。

BioTrain: Sub-MB, Sub-50mW On-Device Fine-Tuning for Edge-AI on Biosignals

论文配图:BioTrain: Sub-MB, Sub-50mW On-Device Fine-Tuning for Edge-AI on Biosignals
图 1 · 摘自论文原文
  • 通过轻量级内存分配与网络优化,支持毫瓦级功耗下的全网络微调。
  • 新用户校准阶段准确率提升35%,比仅更新最后层高7%。
  • 适用于可穿戴设备上的隐私保护实时自适应,如脑电、眼电信号处理。

生物信号存在显著的跨个体和跨会话差异,导致部署后性能严重下降,因此边缘端自适应对保护用户隐私和保障系统可靠性至关重要。然而,现有基于超低功耗微控制器的可穿戴平台因全反向传播(BP)带来的巨大内存和计算开销,仅能支持浅层或稀疏的适配方案。本文提出 BioTrain 框架,在毫瓦级功耗和亚兆字节内存约束下,实现对先进生物信号模型的全网络微调。在 EEG 与 EOG 数据集上,通过离线与真实设备测试验证了其在新用户初始校准和长期信号漂移适应中的有效性。结果表明,全网络微调相比未适配基线最高提升 35% 准确率,在新用户校准时优于仅更新最后一层方案约 7%。在 GAP9 MCU 平台上,BioTrain 实现每秒 17 样本(EEG)和 85 样本(EOG)的高效训练吞吐量,功耗低于 50 mW。此外,其高效的内存分配器与网络拓扑优化支持大批次训练,显著降低峰值内存使用;在 GAP9 上实现全片上反向传播时,内存占用从 5.4 MB 降至 0.67 MB,减少 8.1 倍。

原文摘要 · Abstract (English)

Biosignals exhibit substantial cross-subject and cross-session variability, inducing severe domain shifts that degrade post-deployment performance for small, edge-oriented AI models. On-device adaptation is therefore essential to both preserve user privacy and ensure system reliability. However, existing sub-100 mW MCU-based wearable platforms can only support shallow or sparse adaptation schemes due to the prohibitive memory footprint and computational cost of full backpropagation (BP). In this paper, we propose BioTrain, a framework enabling full-network fine-tuning of state-of-the-art biosignal models under milliwatt-scale power and sub-megabyte memory constraints. We validate BioTrain using both offline and on-device benchmarks on EEG and EOG datasets, covering Day-1 new-subject calibration and longitudinal adaptation to signal drift. Experimental results show that full-network fine-tuning achieves accuracy improvements of up to 35% over non-adapted baselines and outperforms last-layer updates by approximately 7% during new-subject calibration. On the GAP9 MCU platform, BioTrain enables efficient on-device training throughput of 17 samples/s for EEG and 85 samples/s for EOG models within a power envelope below 50 mW. In addition, BioTrain's efficient memory allocator and network topology optimization enable the use of a large batch size, reducing peak memory usage. For fully on-chip BP on GAP9, BioTrain reduces the memory footprint by 8.1x, from 5.4 MB to 0.67 MB, compared to conventional full-network fine-tuning using batch normalization with batch size 8.

边缘计算生物信号模型微调低功耗

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。