在图形处理单元上训练卷积神经网络

摘要

在图形处理单元上实现一卷积神经网络。然后，卷积神经网络通过一系列正向传递和反向传递来进行训练，其中卷积核和偏差矩阵在每一次反向传递中根据误差函数的梯度得到修正。该实现利用图形处理单元中的像素遮蔽单元的并行处理性能，并利用一组逐步推导的公式对像素遮蔽单元的计算进行编程。程序通过纹理来完成输入和输出，并且使用一多级求和运算过程对各像素遮蔽单元寄存器进行求和运算。