arXiv · 2312.12483
SCoTTi: Save Computation at Training Time with an adaptive framework
Abstract
On-device training is an emerging approach in machine learning where models are trained on edge devices, aiming to enhance privacy protection and real-time performance. However, edge devices typically possess restricted computational power and resources, making it challenging to perform computationally intensive model training tasks. Consequently, reducing resource consumption during training has become a pressing concern in this field. To this end, we propose SCoTTi (Save Computation at Training Time), an adaptive framework that addresses the aforementioned challenge. It leverages an optimizable threshold parameter to effectively reduce the number of neuron updates during training which corresponds to a decrease in memory and computation footprint. Our proposed approach demonstrates superior performance compared to the state-of-the-art methods regarding computational resource savings on various commonly employed benchmarks and popular architectures, including ResNets, MobileNet, and Swin-T.
Explore related subjects
Keep this discovery
Ziyu Lin, Enzo Tartaglione, Van-Tam Nguyen. 2023-12-19. SCoTTi: Save Computation at Training Time with an adaptive framework. https://doi.org/10.1109/iccvw60793.2023.00156
Cite the original work for its findings. Save a collection to share your selection of sources.