返回目录

Efficient YOLO

Efficient Yolo: A Lightweight Model For Embedded Deep Learning Object Detection

ICME Workshops 2020第一作者计算机视觉目标检测模型压缩

面向嵌入式道路目标检测,以稀疏初始化、层级与通道剪枝压缩 YOLOv3;当前解析依据公开摘要。

论文署名Zixuan Wang, Jiacheng Zhang, Zhicheng Zhao, Fei Su

English summary

文章目录 · 7 节

阅读范围

这篇论文的 IEEE 公开页面提供摘要和引言,全文 PDF 当前需要访问权限。本页据此整理已确认的信息;训练细节、方法图、消融表和测试集评价规则,仍需取得全文后补充核对。

问题:让检测模型在设备上跑起来

道路目标检测需要同时满足识别效果和推理时延。服务器上表现良好的模型,搬到嵌入式设备后,可能遇到参数量、计算量和运行速度的限制。Efficient YOLO 以 YOLOv3 为基础,研究如何压缩模型,同时保留检测能力。

公开方法:训练与结构压缩相配合

摘要给出三项关键设计:用迭代初始化策略促进初始训练阶段的稀疏性;结合层级剪枝和通道剪枝减少模型负担;利用外部数据支持检测精度。

从一般方法角度理解,层级剪枝改变网络深度,通道剪枝改变中间特征宽度,两者都会改变实际计算结构。至于这篇论文如何选择具体层和通道、如何恢复精度,公开摘要没有提供足够细节,不能由方法名称推定。

已报告的实验结果

IEEE 官方摘要 报告:相对原始版本,模型大小减少 96.93%,计算量减少 84.36%;在 NVIDIA Jetson TX2 上,推理速度达到原版的 2.23 倍,测试集 mAP 为 0.492。

这些数字对应不同评价维度:文件或模型规模、理论计算成本、实际硬件速度、检测效果。不能把计算量的下降比例直接解释成同等幅度的延迟下降;0.492 也需要结合论文的测试集和 mAP 定义阅读,不能直接与其他基准榜单比较。

后续补全全文时,优先核对精度与速度是否使用一致的输入尺寸及推理配置。

在我的研究路径中

这是我学生时期面向部署约束的一项工作。我以 Zixuan Wang 署名,列第一作者。它与后续工业算法的联系,在于把模型效果放到资源和运行条件中评估。这里不把摘要中的实验扩大为其他设备或业务场景的实测结论。

书目信息

正式出版物是 2020 IEEE International Conference on Multimedia & Expo Workshops(ICMEW),页码 1—6。会议议程中的报告题目也可在 ICME 2020 官方 Programme Guide 查到。

English overview

Efficient YOLO targets deployment on embedded hardware by compressing YOLOv3. The public abstract describes sparse initialization, layer-level pruning and channel-wise pruning, with evaluation on Jetson TX2. I am the first author; a full-text analysis is pending access to the paper.