Model compression has been employed extensively recently to reduce the size of deep learning models, enabling efficient deployment in resource-constrained environments like edge devices. However, there is a trade-off between model performance and compression. In this study, we conducted several experiments to investigate the performances and weaknesses of three compression techniques: knowledge distillation, model pruning, and quantization on three different image classification tasks: pothole detection, paddy rice maturity detection, and soybean weed detection. Real-life datasets were collected and utilized solely for the purpose of this work. The experimental results reveal that model performance typically suffers after compression, which is the common notable drawback of model compression strategies. Therefore, we proposed a hybrid lossless-compressed model by combining pruning and quantization to reduce parameters and weights, resulting in a lightweight model. The proposed model is three times smaller than the vanilla CNN model. Moreover, the proposed compressed model achieved a state-of-the-art accuracy of 97% after compression, which shows the effectiveness of our approach.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Hybrid Lightweight Deep Learning Model for Edge Devices: Combining Knowledge Distillation, Pruning, and Quantization

  • Aminu Musa,
  • Mohamed Hamada,
  • Mohammed Hassan,
  • Usman Umar,
  • Alex Aaron,
  • Anand Mahendran

摘要

Model compression has been employed extensively recently to reduce the size of deep learning models, enabling efficient deployment in resource-constrained environments like edge devices. However, there is a trade-off between model performance and compression. In this study, we conducted several experiments to investigate the performances and weaknesses of three compression techniques: knowledge distillation, model pruning, and quantization on three different image classification tasks: pothole detection, paddy rice maturity detection, and soybean weed detection. Real-life datasets were collected and utilized solely for the purpose of this work. The experimental results reveal that model performance typically suffers after compression, which is the common notable drawback of model compression strategies. Therefore, we proposed a hybrid lossless-compressed model by combining pruning and quantization to reduce parameters and weights, resulting in a lightweight model. The proposed model is three times smaller than the vanilla CNN model. Moreover, the proposed compressed model achieved a state-of-the-art accuracy of 97% after compression, which shows the effectiveness of our approach.