湖北民族大学学报（自然科学版）2025，Vol.43Issue(1)：47-52,6.DOI:10.13501/j.cnki.42-1908/n.2025.03.016
基于场景复杂度的轻量化自适应剪枝模型
A Lightweight Adaptive Pruning Model Based on Scene Complexity
摘要
Abstract
To meet the requirements of efficient deployment of deep learning models in resource-constrained environments,the adaptive scene complexity-based mobile vision transformer(AC-MViT)pruning model was proposed.Firstly,a scene complexity evaluation module was introduced into the mobile vision transformer(MobileViT)network to dynamically adjust the number of channels and transformer layers according to the image complexity,so as to achieve the effect of reducing computational costs in simple scenes and retaining feature details in complex scenes.Meanwhile,the coupling strategy of dynamic convolution and transformer layers was applied,and random pruning and computational cost penalty were added during training to enhance the robustness of the model.The results showed that AC-MViT model had achieved remarkable results on the common objects in context 2017(COCO 2017)dataset and the pattern analysis,statical modeling and computational learning visual object classes 2012(PASCAL VOC 2012)dataset.Compared with MobileViT model,the mean average precision(mAP)decreased by 1.1%and 0.1%on COCO 2017 and PASCAL VOC 2012 dataset respectively,the floating-point operation speed was reduced by 43.1%and 46.6%respectively,the number of parameters decreased by 49.1%and 50.0%respectively,and the inference time was shortened by 48.4%and 52.5%respectively.The AC-MViT model performed excellently under various scene complexities,demonstrating its efficiency and balance in resource-constrained environments applications.关键词
深度学习/目标检测/自适应剪枝/轻量化/移动视觉变换器/复杂度评估Key words
deep learning/object detection/adaptive pruning/lightweight/MobileViT/complexity evaluation分类
计算机与自动化引用本文复制引用
包俊涛,吴岳,刘丙友..基于场景复杂度的轻量化自适应剪枝模型[J].湖北民族大学学报（自然科学版）,2025,43(1):47-52,6.基金项目
芜湖市核心技术攻关项目(2022hg11,2023yf012). （2022hg11,2023yf012）
