Bridging the Resource Gap: Deploying Advanced Imitation Learning Models Onto Affordable Embedded Platforms
Advanced imitation learning with structures like the transformer is increasingly demonstrating its advantages in robotics. However, deploying these large-scale models on embedded platforms remains a major challenge. In this paper, we propose a pipeline that facilitates the migration of advanced imitation learning algorithms to edge devices. The process is achieved via an efficient model compression method and a practical asynchronous parallel method Temporal Ensemble with Dropped Actions (TEDA) that enhances the smoothness of operations. To show the efficiency of the proposed pipeline, large-scale imitation learning models are trained on a server and deployed on an edge device to complete various manipulation tasks.
Authors
- Rui Deng (ORCID: https://orcid.org/0000-0002-6666-9095)
- Zhuoyi Pang (ORCID: https://orcid.org/0000-0002-8581-8498)
- Yuhang Dong
- Genqing Zhou (ORCID: https://orcid.org/0000-0003-4506-7658)
- Hongrui Zhu (ORCID: https://orcid.org/0000-0003-3405-7032)
- Huiyao Ge (ORCID: https://orcid.org/0000-0002-4928-1777)
- Jian Zhang (ORCID: https://orcid.org/0000-0003-0184-5593)
- Zhi Ping Xu (ORCID: https://orcid.org/0000-0001-6070-5035)
- Ruixiang Wang (ORCID: https://orcid.org/0000-0003-4666-9917)
- Shi Lu (ORCID: https://orcid.org/0000-0002-5972-7226)
Institutions
- Jiangnan University (CN)
- Tongji University (CN)
- Harbin Institute of Technology (CN)
- California University of Pennsylvania (US)
- Zhejiang University (CN)
- Tsinghua University (CN)
Publication Details
- Published
- 2024-12-10
- DOI
- https://doi.org/10.1109/robio64047.2024.10907641
- Primary Topic
- Reinforcement Learning in Robotics
- Type
- article
- Field-Weighted Citation Impact
- 0.00