NN 输出相对于输入的梯度

数据挖掘 机器学习 神经网络 深度学习 张量流 火炬
2022-02-12 01:30:49

我已经针对多个输入可以映射到同一输出的问题训练了一个神经网络 (NN)。我想用这个神经网络从输出到输入,即给定一个输出向量y, 我想找一个输入x这样 NN 返回一些z接近给定的输出y当输入x.

我正在考虑使用梯度下降来做到这一点。是否有任何常见的深度学习 API 允许您根据输入获取 NN 的梯度?

我环顾四周并没有发现任何东西,但我想在继续前进之前我会在这里检查一下。

2个回答

iNNvestigate是一个非常强大且编写良好的库,用于检查神经网络。其中,它包括gradient方法。

然而,如果你想合成一个输入,使网络产生特定的输出——这种技术被称为代码反转。遵循Mahendran 等人的论文。更多细节。我找到了以下实现:

就在这里。这称为“对输入的反向传播”。我想邀请您阅读这个基于lucid的很棒的博客你会在他们的笔记本上看到一些代码。