NoProp: Training Neural Networks without Back-propagation or Forward-propagation

ML LFDDR NN POA DM
传统的深度学习方法需要通过从输出端向每个可学习参数反向传播误差信号,来计算每一层的梯度项。由于神经网络的堆叠结构,其中每一层都基于其下一层的表示构建,这种方法导致了分层的特征表示。模型顶层通常包含更抽象的特征,而底层的特征则相对不那么抽象。与此不同的是,我们提出了一种新的学习方法,称为 NoProp,它既不依赖前向传播也不依赖反向传播。相反,NoProp 从扩散模型和流匹配方法中汲取灵感,让每一层独立学习如何对一个带有噪声的目标进行去噪。我们认为这项工作迈出了引入一类新梯度无关学习方法的第一步,这类方法不学习分层表示——至少不是传统意义上的分层表示。NoProp 需要预先将每一层的表示固定为目标的带噪版本,并学习一个局部去噪过程,该过程可以在推理时被利用。我们在 MNIST、CIFAR-10 和 CIFAR-100 图像分类基准上展示了我们方法的有效性。实验结果表明,NoProp 是一种可行的学习算法,相比其他现有的无反向传播方法,它具有更高的准确性、更易于使用且计算效率更高。通过偏离传统的基于梯度的学习范式,NoProp 改变了网络内部的信用分配方式,从而实现了更高效的分布式学习,并可能影响学习过程的其他特性。
The canonical deep learning approach for learning requires computing a gradient term at each layer by back-propagating the error signal from the output towards each learnable parameter. Given the stacked structure of neural networks, where each layer builds on the representation of the layer below, this approach leads to hierarchical representations. More abstract features live on the top layers of the model, while features on lower layers are expected to be less abstract. In contrast to this, we introduce a new learning method named NoProp, which does not rely on either forward or backwards propagation. Instead, NoProp takes inspiration from diffusion and flow matching methods, where each layer independently learns to denoise a noisy target. We believe this work takes a first step towards introducing a new family of gradient-free learning methods, that does not learn hierarchical representations -- at least not in the usual sense. NoProp needs to fix the representation at each layer beforehand to a noised version of the target, learning a local denoising process that can then be exploited at inference. We demonstrate the effectiveness of our method on MNIST, CIFAR-10, and CIFAR-100 image classification benchmarks. Our results show that NoProp is a viable learning algorithm which achieves superior accuracy, is easier to use and computationally more efficient compared to other existing back-propagation-free methods. By departing from the traditional gradient based learning paradigm, NoProp alters how credit assignment is done within the network, enabling more efficient distributed learning as well as potentially impacting other characteristics of the learning process.
许愿