将跨步滤波器梯度转换为卷积形式

人工智能 卷积神经网络
2021-10-23 14:32:33

我正在实现跨步二维卷积。我的公式如下所示:

(1)yi,j=m=0Fh1n=0Fw1xsi+m,sj+nfm,n,
在哪里s是步幅(一些消息来源可能将此称为“互相关”,但“卷积”与 PyTorch 的定义一致)

我计算了相对于过滤器的梯度:

(2)Efm,n=i=0(xhFh)/sj=0(xwFw)/sxsi+m,sj+nEyi,j

和一些简单的虚拟索引重新标记导致:

(3)Efi,j=m=0(xhFh)/sn=0(xwFw)/sxsm+i,sn+jEym,n

方程(3)看起来与第一个相似,但不完全一样(s是错误的术语!)。我的目标是将第二个方程转换为“卷积形式”,以便我可以使用现有的高效卷积算法来计算它。

有人可以帮我解决这个问题,或者指出我犯的任何错误吗?

0个回答
没有发现任何回复~