From Compression to Deployment: Real-Time and Energy-Efficient FastGRNN on Ultra-Constrained Microcontrollers

21d ago · Global · primary source: export.arxiv.org

A research team has reproduced a compact recurrent neural network cell, FastGRNN, on two bare-metal microcontrollers, demonstrating real-time inference within a 566-byte weight budget and 100% prediction agreement with a PyTorch reference model. The work targets the 8-bit Arduino ATmega328P and the 16-bit MSP430, a chip with no hardware multiplier, 16 KB of Flash, and 512 bytes of SRAM [1]. The compression pipeline combines low-rank weight factorization, iterative hard-thresholding sparsity, and per-tensor Q15 post-training quantization with explicit activation calibration [1]. The deployed model achieves a macro F1 score of 0.918 on the HAPT test set for seed 0, with a five-seed Q15 mean of 0.853 plus or minus 0.107 [1]. Across 3,399 test windows, the microcontroller implementation matches a PyTorch reference at 100% prediction agreement for seed 0, and 99.91 to 100% C-equivalent agreement across five seeds [1]. PyTorch is an open-source deep learning library originally developed by Meta Platforms and now supported by the Linux Foundation, providing a high-level API for model training and inference [4]. Both platforms sustain real-time 50 Hz streaming inference, with 9.21 milliseconds per sample on the Arduino and 13 milliseconds on the MSP430 [1]. A 256-entry sigmoid and tangent look-up table delivers a 30.5x speedup on the multiplier-less MSP430 [1]. The study extends the original FastGRNN paper with four contributions: cross-platform bit-equivalent deterministic inference, characterization of recurrent warm-up latency with a median of 74 samples or 1.48 seconds and a worst-case of 125 samples or 2.50 seconds over 100 test windows, a deployable look-up-table recipe for multiplier-less embedded targets, and hardware energy characterization [1]. Active inference power is measured at 17.7 milliwatts, idle power below 0.09 milliwatts, and the look-up table yields a 96.7 percent energy reduction [1]. The authors, including Emre Can Kızılateş, frame the work as a response to a multi-year global semiconductor supply constraint and the growing energy and carbon cost of always-online inference, arguing for refactoring AI algorithms to fit the small, ubiquitous microcontrollers already in mass production in wearables, sensors, and edge appliances [1]. The open-source release adds to a landscape of publicly available AI software, which includes libraries, frameworks, and tools for machine learning, deep learning, and other domains [6].

infrastructureresearch-papertool-release

Background sources we checked (5)
  • arxiv.org ↗ The dominant trajectory of modern machine learning has been to scale up: larger models, larger accelerators, larger memory budgets. Yet a multi-year global semiconductor supply constraint and the growing energy and carbon cost of always-online inference expose the fragility of th…
  • arxiv.org ↗ We present a large, tunable neural conversational response generation model, DialoGPT (dialogue generative pre-trained transformer). Trained on 147M conversation-like exchanges extracted from Reddit comment chains over a period spanning from 2005 through 2017, DialoGPT extends th…
  • en.wikipedia.org ↗ PyTorch is an open-source deep learning library, originally developed by Meta Platforms and currently developed with support from the Linux Foundation. The successor to Torch, PyTorch provides a high-level API that builds upon optimised, low-level implementations of deep learning…
  • en.wikipedia.org ↗ Hangzhou DeepSeek Artificial Intelligence Basic Technology Research Co., Ltd., doing business as DeepSeek, is a Chinese artificial intelligence (AI) company that develops large language models (LLMs). Based in Hangzhou, Zhejiang, DeepSeek is owned and funded by High-Flyer, a Chin…
  • en.wikipedia.org ↗ These lists include projects which release their software under open-source licenses and are related to artificial intelligence projects. These include software libraries, frameworks, platforms, and tools used for machine learning, deep learning, natural language processing, comp…

Sources

Spot something wrong? Report an issue