Multi-Agent Pruning and Quantization in Mixed-Precision Models
In order to improve the size reduction of the deep learning models for deploying to edge devices, this study employs the Multi-Agent Deep Deterministic Policy Gradient (MADDPG) method, combined with the Preserve Ratio for each layer, to formulate different compression strategies without manual parameter tuning. Additionally, it evaluates the model compression effectiveness using metrics such as Top-1 accuracy, model size, parameter count, FLOPs, and latency. The results show that compressing MobileNet-V2 at FP32 precision on the target hardware reduces inference latency by 45.8%. Moreover, the model size decreases by 79.23%, the parameter counts by 60%, and the model computational load by 90.35%.
Authors
- Zhongwei Chen (ORCID: https://orcid.org/0000-0003-3463-5509)
- Alan Liu (ORCID: https://orcid.org/0000-0002-8884-6662)
- Michael Hsieh
Institutions
- National Chung Cheng University (TW)
Publication Details
- Published
- 2024-10-06
- DOI
- https://doi.org/10.1109/smc54092.2024.10831664
- Primary Topic
- Robotics and Sensor-Based Localization
- Type
- article
- Field-Weighted Citation Impact
- 0.00