POET: Training Neural Networks on Tiny Devices with Integrated
Rematerialization and Paging

POET: Training Neural Networks on Tiny Devices with Integrated Rematerialization and Paging

International Conference on Machine Learning (ICML), 2022

15 July 2022

Shishir G. Patil

Joseph E. Gonzalez

ArXiv (abs)PDF HTML Github (164★)

Papers citing "POET: Training Neural Networks on Tiny Devices with Integrated Rematerialization and Paging"

16 / 16 papers shown

Title
FTTE: Federated Learning on Resource-Constrained Devices Irene Tenison Anna Murphy Charles Beauville Lalana Kagal FedML 52 0 0 03 Oct 2025
FedEL: Federated Elastic Learning for Heterogeneous Devices Letian Zhang Bo Chen Jieming Bian Lei Wang Jie Xu FedML 104 0 0 21 Sep 2025
MUDAS: Mote-scale Unsupervised Domain Adaptation in Multi-label Sound Classification Jihoon Yun Chengzhang Li Dhrubojyoti Roy A. Arora 89 0 0 12 Jun 2025
SlimPipe: Memory-Thrifty and Efficient Pipeline Parallelism for Long-Context LLM Training Zheng Li Wenshu Fan Wei Zhang Tailing Yuan Bin Chen Chengru Song Chen Zhang 171 1 0 20 Apr 2025
Privacy-Enhanced Training-as-a-Service for On-Device Intelligence: Concept, Architectural Scheme, and Open Problems Zhiyuan Wu Sheng Sun Yuwei Wang Min Liu Bo Gao Tianliu He Wen Wang 276 0 0 24 Feb 2025
Asteroid: Resource-Efficient Hybrid Pipeline Parallelism for Collaborative DNN Training on Heterogeneous Edge DevicesACM/IEEE International Conference on Mobile Computing and Networking (MobiCom), 2024 Shengyuan Ye Liekang Zeng Xiaowen Chu Guoliang Xing Xu Chen 193 30 0 15 Aug 2024
Online Resource Allocation for Edge Intelligence with Colocated Model Retraining and Inference Huaiguang Cai Zhi Zhou Qianyi Huang 128 6 0 25 May 2024
Training Machine Learning models at the Edge: A Survey Aymen Rayane Khouas Mohamed Reda Bouadjenek Hakim Hacid Sunil Aryal 334 20 0 05 Mar 2024
Peer-to-Peer Deep Learning for Beyond-5G IoTAsilomar Conference on Signals, Systems and Computers (ACSSC), 2023 Srinivasa Pranav José M. F. Moura 202 3 0 29 Oct 2023
Efficient Memory Management for Large Language Model Serving with PagedAttentionSymposium on Operating Systems Principles (SOSP), 2023 Woosuk Kwon Zhuohan Li Siyuan Zhuang Ying Sheng Lianmin Zheng Cody Hao Yu Joseph E. Gonzalez Haotong Zhang Ion Stoica VLM 1.1K 3,933 0 12 Sep 2023
Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorchInternational Conference on Machine Learning (ICML), 2023 Xunyi Zhao Théotime Le Hellard Lionel Eyraud Julia Gusak Olivier Beaumont 249 10 0 03 Jul 2023
Breaking On-device Training Memory Wall: A Systematic Survey Shitian Li Chunlin Tian Kahou Tam Ruirui Ma Li Li 149 2 0 17 Jun 2023
XEngine: Optimal Tensor Rematerialization for Neural Networks in Heterogeneous EnvironmentsACM Transactions on Architecture and Code Optimization (TACO) (TACO), 2022 Manuela Schuler Richard Membarth P. Slusallek 225 4 0 19 Dec 2022
On-device Training: A First Overview on Existing Systems Shuai Zhu Thiemo Voigt Jeonggil Ko Fatemeh Rahimian 379 30 0 01 Dec 2022
OLLA: Optimizing the Lifetime and Location of Arrays to Reduce the Memory Usage of Neural Networks Benoit Steiner Mostafa Elhoushi Jacob Kahn James Hegarty 211 10 0 24 Oct 2022
On-Device Training Under 256KB MemoryNeural Information Processing Systems (NeurIPS), 2022 Ji Lin Ligeng Zhu Wei-Ming Chen Wei-Chen Wang Chuang Gan Song Han MQ 345 254 0 30 Jun 2022