面向3D生成对抗网络的频域加速器设计

三维生成对抗网络(3D generative adversarial networks,3D GANs)广泛应用于模型预测、对象生成等领域.针对当前3D GANs加速器存在计算开销高、访存能耗高等问题,提出一种基于快速傅里叶变换(FFTs)的3D GANs频域加速器设计(FAG).首先针对3D GANs加速器计算开销高的问题,提出支持频域3D GANs计算的硬件架构,利用频域低计算复杂度特性和频域反卷积中的零值模式以减少计算开销;其次针对3D GANs加速器访存能耗高的问题,提出支持3D卷积和反卷积计算的频域映射调度流,利用频域共轭对称特性和反卷积中的零值模式减少访存开销.基于ModelNet数据集和3个3D GANs模型(3DGAN,3D-IWGAN和3D-PhysNet)的评估结果表明,与具有代表性的频域加速器相比,FAG可使性能提高76%,能效提高141%;与具有代表性的空间域加速器相比,FAG的性能提高6倍,能效提高46倍.