一种基于剪枝技术的高效神经网络搜索与训练方法

    公开(公告)号:CN115688908A

    公开(公告)日:2023-02-03

    申请号:CN202211192735.X

    申请日:2022-09-28

    Applicant: 东南大学

    Abstract: 本发明公开了一种兼顾检测效率和精度的面向合成孔径雷达图像舰船目标检测任务的可微神经网络搜索与训练方法。该方法在SAR图像数据集上进行可微神经网络架构搜索,即在预设的包含有多种不同卷积和池化操作的搜索空间上,利用可微架构搜索对超级网络进行优化,以得到满足目标任务约束的神经网络模型,并对模型进行完备训练。通过改造可微架构搜索,不仅在可微架构搜索过程中集成了通道剪枝技术和模型剪枝技术,还创建新的混合损失函数,将交叉熵损失和模型计算量结合起来。本发明在舰船目标的目标检测任务上能自动化生成轻量级神经网络模型,在模型效率和检测精度上同时具有优异表现。

Patent Agency Ranking