一种基于保留因子剪枝的生成对抗网络模型的压缩方法

摘要

一种基于保留因子剪枝的生成对抗网络模型的压缩方法。本发明公开了一种生成对抗网络模型的压缩方法，属于人工智能技术领域，方法包括：给定预训练教师生成对抗网络模型和待压缩学生对抗网络模型，预训练教师生成对抗网络模型包括教师生成器和教师判别器，待压缩学生对抗网络模型包括学生生成器和学生判别器；为学生生成器的每个卷积核设置一个可学习的保留因子，根据边界参数，将保留因子转换成对应的卷积核的参与前向运算比例；通过对保留因子添加约束条件和收缩边界参数，确定参与前向运算比例为零的卷积核；移除参与前向运算比例为零的卷积核，获得轻量的学生生成器；根据教师生成器和教师判别器中的知识，对学生生成器进行微调训练；移除学生判别器，保留学生生成器，部署学生对抗网络模型。