基于通道注意力机制的神经网络压缩方法及相关设备

摘要

本申请涉及人工智能领域，本申请公开了一种基于通道注意力机制的神经网络压缩方法及相关设备，所述方法包括：构建神经网络模型，并在所述神经网络模型中建立通道注意力机制，所述神经网络模型包含多个通道；对所述神经网络模型进行训练；在对所述神经网络模型的训练过程中根据所述通道注意力机制对所述神经网络模型中的通道进行删减。本申请通过在神经网络模型的全连接层前加入通道注意力层，并计算通道权重，将权重低的通道进行删减，实现对神经网络的压缩。