Sep, 2022

通用激活函数的快速神经核嵌入

TL;DR本文提出了一种基于截断埃尔米特函数的方法,用于近似计算任何多层神经网络的高斯过程核(NNGP)和神经切向核(NTK)矩阵,同时克服了其他方法中数据点必须在单位球上的限制,可适用于任何 $R^d$ 空间中的点集。实验证明,相对于精确的卷积神经切向核计算,该方法在 CIFAR-10 数据集上针对具有 5 层结构的 Myrtle 网络的近似计算中实现了 106 倍的加速。