通道剪枝与知识蒸馏相结合的轻量化SAR目标检测

作     者：黄启灏 靳国旺 熊新 王丽美 李佳豪 HUANG Qihao;JIN Guowang;XIONG Xin;WANG Limei;LI Jiahao

作者机构：信息工程大学地理空间信息学院河南郑州450001 河南城建学院测绘与城市空间信息学院河南平顶山467000 

出 版 物：《测绘学报》 (Acta Geodaetica et Cartographica Sinica)

年 卷 期：2024年第53卷第4期

页      面：712-723页

核心收录：

学科分类：083002[工学-环境工程] 0830[工学-环境科学与工程（可授工学、理学、农学学位）] 081802[工学-地球探测与信息技术] 08[工学] 0818[工学-地质资源与地质工程] 081602[工学-摄影测量与遥感] 0816[工学-测绘科学与技术] 

基　　金：国家自然科学基金(41474010 61401509 42201492) 

主　　题：SAR 目标检测 轻量化 通道剪枝 知识蒸馏 

摘      要：轻量化SAR目标检测方法对快速检测SAR影像中的地物目标具有重要意义。针对轻量化检测方法精度不高的问题,设计了一种通道剪枝与知识蒸馏相结合的轻量化SAR目标检测方法。该方法通过对复杂网络中批归一化层的缩放因子γ进行稀疏化训练,判别对应特征通道的重要程度,进而裁剪次要通道,并在微调训练后将其作为教师网络,构造知识蒸馏框架指导轻量模型训练,提高轻量模型的检测精度。采用YOLOv5-6.1算法搭建了检测框架,并在重组的MSAR和SSDD多类目标数据集上进行了训练和检测试验,结果表明该方法能够在保持模型体积仅3.73 MB的轻量化条件下,提升SAR目标检测精度,验证了本文方法的有效性。