On-device Training: A First Overview on Existing Systems
Not on the map. Why?
Abstract (arXiv)
The recent breakthroughs in machine learning (ML) and deep learning (DL) have catalyzed the design and development of various intelligent systems over wide application domains. While most existing machine learning models require large memory and computing power, efforts have been made to deploy some models on resource-constrained devices as well. A majority of the early application systems focused on exploiting the inference capabilities of ML and DL models, where data captured from different mobile and embedded sensing components are processed through these models for application goals such as classification and segmentation. More recently, the concept of exploiting the mobile and embedded computing resources for ML/DL model training has gained attention, as such capabilities allow (i) the training of models via local data without the need to share data over wireless links, thus enabling privacy-preserving computation by design, (ii) model personalization and environment adaptation, and (ii) deployment of accurate models in remote and hardly accessible locations without stable internet connectivity. This work targets to summarize and analyze state-of-the-art systems research that allows such on-device model training capabilities and provide a survey of on-device training from a systems perspective.
Key facts. Cited by 18 papers in the map (data through 7 Oct 2026). Builds on 76 of its 158 references that are in the map. First posted on arXiv 1 Dec 2022.
Builds on
76 of 158 references.
References · 76
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,773
38 more
- 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,827
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2018YOLOv3: An Incremental Improvementcited by 2,652
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2016Matching Networks for One Shot Learningcited by 2,246
- 2017Gradient Episodic Memory for Continual Learningcited by 1,871
- 2013Fine-Grained Visual Classification of Aircraftcited by 1,701
- 2020Knowledge Distillation: A Surveycited by 1,578
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
- 2018Continual Lifelong Learning with Neural Networks: A Reviewcited by 1,313
- 2017Mixed Precision Trainingcited by 1,135
- 2017Learning Transferable Architectures for Scalable Image Recognitioncited by 1,097
- 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,054
- 2020Adaptive Federated Optimizationcited by 1,036
- 2017VGGFace2: A dataset for recognising faces across pose and agecited by 911
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2015Dueling Network Architectures for Deep Reinforcement Learningcited by 769
- 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 741
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 697
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
- 2017In-Datacenter Performance Analysis of a Tensor Processing Unitcited by 610
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 571
- 2019Importance Estimation for Neural Network Pruningcited by 531
- 2018Learning to Learn without Forgetting by Maximizing Transfer and Minimizing Interferencecited by 511
- 2013DeCAF: A Deep Convolutional Activation Feature for Generic Visual Recognitioncited by 422
- 2014CNN Features off-the-shelf: an Astounding Baseline for Recognitioncited by 316
- 2020MCUNet: Tiny Deep Learning on IoT Devicescited by 269
- 2015Net2Net: Accelerating Learning via Knowledge Transfercited by 245
- 2021Accelerating Sparse Deep Neural Networkscited by 213
- 2018Online Learning: A Comprehensive Surveycited by 203
- 2019Meta-Learning Representations for Continual Learningcited by 189
- 2022On-Device Training Under 256KB Memorycited by 158
- 2016Continuous Deep Q-Learning with Model-based Accelerationcited by 149
- 2017Towards Accurate Binary Convolutional Neural Networkcited by 148
- 2020Sparse GPU Kernels for Deep Learningcited by 138
50 most cited of 76 All 76 (JSON)
Built on by
Citing papers per year: 2024: 6; 2025: 7; 2026: 5.
18 papers.
Sort:
By area:
- 2024Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmarkcited by 113
- 2024Streaming Deep Reinforcement Learning Finally Workscited by 24
- 2024Training Machine Learning models at the Edge: A Surveycited by 11
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devicescited by 3
- 2025P3SL: Personalized Privacy-Preserving Split Learning on Heterogeneous Edge Devicescited by 2
- 2024Learning Using a Single Forward Passcited by 1
- 2024Federating Dynamic Models using Early-Exit Architectures for Automatic Speech Recognition on Heterogeneous Clientscited by 1
- 2025Onboard Optimization and Learning: A Surveycited by 1
- 2025AdaBet: Gradient-free Layer Selection for Efficient Training of Deep Neural Networkscited by 1
- 2026TrainDeeploy: Hardware-Accelerated Parameter-Efficient Fine-Tuning of Small Transformer Models at the Extreme Edgecited by 1
- 2024Residual-INR: Communication Efficient On-Device Learning Using Implicit Neural Representationcited by 0
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEGcited by 0
6 more
- 2025Precision-Scalable Microscaling Datapaths with Optimized Reduction Tree for Efficient NPU Integrationcited by 0
- 2025Multi-modal On-Device Learning for Monocular Depth Estimation on Ultra-low-power MCUscited by 0
- 2026BioTrain: Sub-MB, Sub-50mW On-Device Fine-Tuning for Edge-AI on Biosignalscited by 0
- 2026Uncertainty-Guided Edge Learning for Deep Image Regression in Remote Sensingcited by 0
- 2026Empowering On-Device Model Adaptation with an Edge AI Inference Acceleratorcited by 0
- 2026Thermo-FL: Thermal-Aware Robust Federated Fine-Tuning of Large Language Models for Edge AIcited by 0
- 2026Thermo-FL: Thermal-Aware Robust Federated Fine-Tuning of Large Language Models for Edge AI21 Aug 2026
- 2026Empowering On-Device Model Adaptation with an Edge AI Inference Accelerator20 Jul 2026
- 2026Uncertainty-Guided Edge Learning for Deep Image Regression in Remote Sensing7 May 2026
- 2026BioTrain: Sub-MB, Sub-50mW On-Device Fine-Tuning for Edge-AI on Biosignals14 Apr 2026
- 2026TrainDeeploy: Hardware-Accelerated Parameter-Efficient Fine-Tuning of Small Transformer Models at the Extreme Edge10 Mar 2026
- 2025Multi-modal On-Device Learning for Monocular Depth Estimation on Ultra-low-power MCUs26 Nov 2025
- 2025Precision-Scalable Microscaling Datapaths with Optimized Reduction Tree for Efficient NPU Integration9 Nov 2025
- 2025AdaBet: Gradient-free Layer Selection for Efficient Training of Deep Neural Networks3 Oct 2025
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devices21 Sep 2025
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEG18 Aug 2025
- 2025P3SL: Personalized Privacy-Preserving Split Learning on Heterogeneous Edge Devices23 Jul 2025
- 2025Onboard Optimization and Learning: A Survey7 May 2025
6 more
- 2024Streaming Deep Reinforcement Learning Finally Works18 Oct 2024
- 2024Residual-INR: Communication Efficient On-Device Learning Using Implicit Neural Representation10 Aug 2024
- 2024Federating Dynamic Models using Early-Exit Architectures for Automatic Speech Recognition on Heterogeneous Clients27 May 2024
- 2024Training Machine Learning models at the Edge: A Survey5 Mar 2024
- 2024Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmark18 Feb 2024
- 2024Learning Using a Single Forward Pass15 Feb 2024
- 2024Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmarkcited by 113
- 2024Streaming Deep Reinforcement Learning Finally Workscited by 24
- 2024Training Machine Learning models at the Edge: A Surveycited by 11
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devicescited by 3
- 2025P3SL: Personalized Privacy-Preserving Split Learning on Heterogeneous Edge Devicescited by 2
- 2024Learning Using a Single Forward Passcited by 1
- 2025Onboard Optimization and Learning: A Surveycited by 1
- 2025AdaBet: Gradient-free Layer Selection for Efficient Training of Deep Neural Networkscited by 1
- 2026TrainDeeploy: Hardware-Accelerated Parameter-Efficient Fine-Tuning of Small Transformer Models at the Extreme Edgecited by 1
- 2024Residual-INR: Communication Efficient On-Device Learning Using Implicit Neural Representationcited by 0
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEGcited by 0
- 2025Precision-Scalable Microscaling Datapaths with Optimized Reduction Tree for Efficient NPU Integrationcited by 0
12 most cited of 17 All 17 (JSON)
- 2026Thermo-FL: Thermal-Aware Robust Federated Fine-Tuning of Large Language Models for Edge AI21 Aug 2026
- 2026Empowering On-Device Model Adaptation with an Edge AI Inference Accelerator20 Jul 2026
- 2026Uncertainty-Guided Edge Learning for Deep Image Regression in Remote Sensing7 May 2026
- 2026BioTrain: Sub-MB, Sub-50mW On-Device Fine-Tuning for Edge-AI on Biosignals14 Apr 2026
- 2026TrainDeeploy: Hardware-Accelerated Parameter-Efficient Fine-Tuning of Small Transformer Models at the Extreme Edge10 Mar 2026
- 2025Multi-modal On-Device Learning for Monocular Depth Estimation on Ultra-low-power MCUs26 Nov 2025
- 2025Precision-Scalable Microscaling Datapaths with Optimized Reduction Tree for Efficient NPU Integration9 Nov 2025
- 2025AdaBet: Gradient-free Layer Selection for Efficient Training of Deep Neural Networks3 Oct 2025
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devices21 Sep 2025
- 2025Toward Storage-Aware Learning with Compressed Data An Empirical Exploratory Study on JPEG18 Aug 2025
- 2025P3SL: Personalized Privacy-Preserving Split Learning on Heterogeneous Edge Devices23 Jul 2025
- 2025Onboard Optimization and Learning: A Survey7 May 2025
12 newest of 17 All 17 (JSON)