Search papers, labs, and topics across Lattice.
This paper introduces a heterogeneous adaptation pipeline that utilizes the Hailo-8L edge AI inference accelerator to facilitate on-device model adaptation for resource-constrained hardware. By partitioning the computational graph, the approach allows for frozen-backbone feature extraction in INT8 while fine-tuning a lightweight FP32 classification head on the CPU, resulting in significant efficiency gains. The method achieves up to 15.4x faster training times compared to a Raspberry Pi 5 CPU, while also maintaining competitive throughput and reducing energy consumption per sample.
Achieving 15.4x faster on-device model adaptation could revolutionize personalization in edge AI applications.
On-device model adaptation is essential to enable lifelong personalization on resource-constrained hardware, but compute, power, and memory limitations of such devices make end-to-end backpropagation impractical for modern deep neural networks. This work proposes a heterogeneous adaptation pipeline that repurposes a commercial edge AI inference accelerator, Hailo-8L, for frozen-backbone feature extraction during on-device training. The computational graph is partitioned so that the pre-trained backbone is quantized to INT8 and run on the accelerator, while only a lightweight FP32 classification head is fine-tuned on the host CPU, enabling frequent, energy-efficient in-field updates with most weights remaining fixed. Across multiple architectures and datasets, this pipeline achieves up to 15.4x faster wall-clock training time compared to a Raspberry Pi 5 CPU baseline, offers competitive throughput in favorable settings, and consistently reduces energy per sample. Post-training quantization restoration is shown to be crucial for preserving the quality of accelerator-generated features and mitigating accuracy loss in quantization-sensitive architectures. Overall, the results demonstrate a practical approach to efficient on-device adaptation using inference-oriented edge accelerators. The implementation is available at https://github.com/MatPiech/accelerator-training.