一种神经网络通道参数的搜索方法及相关设备

    公开(公告)号:CN111401516B

    公开(公告)日:2024-04-26

    申请号:CN202010109184.0

    申请日:2020-02-21

    Abstract: 本申请实施例公开了一种神经网络通道参数的搜索方法及相关设备,首次提出一种基于算力(如FLOPs)的使用效率的神经网络通道参数搜索方法,其应用于人工智能领域,首先计算网络各层对算力的使用效率,然后增加算力的使用效率高的层的通道数,减少算力的使用效率低的层的通道数,该过程可迭代进行,最终获得对算力的使用效率非常高效的神经网络,缓解了目前神经网络通道参数搜索方法没有充分考虑到复杂度、使用效率低、搜索速度慢等问题。本申请还提出了一种计算神经网络各层对算力的使用效率的方法,该方法以一定比例随机丢弃部分通道并测试其对网络性能的影响,丢弃的通道对于不同测试样例是随机的,以此计算神经网络每层对算力的使用效率。

Patent Agency Ranking