Skip to main content
Home
Distributed Systems Group
Universitat Politècnica de Catalunya BARCELONATECH
User account menu
  • login

On-Device Training of Machine Learning Models on Microcontrollers With a Look at Federated Learning

Author
Marc Grau
Roger Pueyo
Felix Freitag
Keywords
embedded systems
federated learning
keyword spotting
machine learning
Abstract

Recent progress in machine learning frameworks makes it now possible to run an inference with sophisticated machine learning models on tiny microcontrollers. Model training, however, is typically done separately on powerful computers. There, the training process has abundant CPU and memory resources to process the stored datasets. In this work, we explore a different approach: training the model directly on the microcontroller. We implement this approach for a keyword spotting task. Then, we extend the training process using federated learning among microcontrollers. Our experiments with model training show an overall trend of decreasing loss with the increase of training epochs.

Year of Publication
2021
Publisher
Association for Computing Machinery
Conference Location
New York, NY, USA
ISBN Number
9781450384780
URL
https://doi.org/10.1145/3462203.3475896
DOI
10.1145/3462203.3475896
  • DOI
  • Google Scholar
  • BibTeX
  • EndNote X3 XML
  • EndNote 7 XML
  • Endnote tagged
  • Marc
  • RIS

Main navigation

  • Home
  • Announcements
  • Projects
  • Research
  • Publications
  • About DSG
  • Location
  • Software
  • Talks
  • Members
  • Former Members
Powered by Drupal