一种面向边缘环境的深度学习模型压缩方法

By combining cross-layer feature fusion and distillation loss function, the challenge of deploying deep learning models in edge environments is solved, improving model performance and efficiency while reducing training costs and protecting data privacy.

CN118135240BActive Publication Date: 2026-07-17GUILIN UNIV OF ELECTRONIC TECH

Patent Information

Authority / Receiving Office
CN · China
Patent Type
Patents(China)
Current Assignee / Owner
GUILIN UNIV OF ELECTRONIC TECH
Filing Date
2024-01-18
Publication Date
2026-07-17

AI Technical Summary

Technical Problem

Existing large-scale deep learning models are difficult to deploy in resource-constrained edge environments, and existing knowledge distillation methods neglect training costs and efficiency issues when improving the accuracy of student networks.

Method used

A cross-layer feature fusion method is adopted, which aligns deep and shallow features through nearest neighbor upsampling and 1×1 convolution kernels, and combines the cross-layer distillation loss function to directly use the original feature map of the teacher model to perform knowledge distillation and train the student model.

Benefits of technology

It improves the performance and efficiency of deep learning models in edge environments, reduces training costs, and protects data privacy, making it suitable for resource-constrained edge devices.

✦ Generated by Eureka AI based on patent content.

Smart Images

  • Figure CN118135240B_ABST
    Figure CN118135240B_ABST
Patent Text Reader

Abstract

本发明公开了一种面向边缘环境的深度学习模型压缩方法,该方法是先收集边缘环境的图像数据,归集为数据集;对数据集进行图像增强处理,选择一个强大的教师模型对数据集进行预训练;再选择一个较小模型作为学生模型,利用经过预训练的教师模型进行知识蒸馏来训练学生模型,在学生模型训练过程中,添加跨层蒸馏损失函数,采用最邻近上采样方法,将教师模型的深层特征以及学生模型的深层特征进行向上的特征融合,以此增强浅层特征的注意力,使教师模型在浅层特征也能转移更多的知识,提升学生模型性能的效果;将经过知识蒸馏的学生模型部署到边缘环境进行图像识别工作。该方法仅对原始特征图进行处理,无需额外的辅助网络或对知识进行浓缩处理。
Need to check novelty before this filing date? Find Prior Art