This paper examines knowledge distillation (KD) for model compression in edge environments. As the number of Internet of Things (IoT) devices increases, optimizing machine learning models for resource-constrained settings is necessary. Knowledge transfer from larger models (teachers) to smaller models (students) is studied to improve efficiency while maintaining performance. Experiments compare teacher-student pairs, including VGG19 with MobileNetV3_small, ResNet-152 with ShuffleNet V2 x0.5, and ResNet-152 with ResNet-18. Accuracy and computational efficiency are evaluated across CIFAR-10, CIFAR-100, and SLT-10 datasets. Results show that KD improves student model performance, particularly under data corruption. Self-distillation with ResNet-18 is analyzed using the same architecture. This research examines KD for optimizing deep learning models in edge deployment, addressing accuracy and resource constraints in machine learning applications.

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

Exploring Knowledge Distillation for Model Compression in Edge Environments

  • Telma Garção,
  • David Belo,
  • Joana Sousa,
  • João Ferreira

摘要

This paper examines knowledge distillation (KD) for model compression in edge environments. As the number of Internet of Things (IoT) devices increases, optimizing machine learning models for resource-constrained settings is necessary. Knowledge transfer from larger models (teachers) to smaller models (students) is studied to improve efficiency while maintaining performance. Experiments compare teacher-student pairs, including VGG19 with MobileNetV3_small, ResNet-152 with ShuffleNet V2 x0.5, and ResNet-152 with ResNet-18. Accuracy and computational efficiency are evaluated across CIFAR-10, CIFAR-100, and SLT-10 datasets. Results show that KD improves student model performance, particularly under data corruption. Self-distillation with ResNet-18 is analyzed using the same architecture. This research examines KD for optimizing deep learning models in edge deployment, addressing accuracy and resource constraints in machine learning applications.