arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04434cs.LGcs.AIstat.ML

利用神经元交互与临近预测网络加速训练

Accelerating Training with Neuron Interaction and Nowcasting Networks

Samsung – SAIT AI Lab(三星-SAIT AI实验室);Concordia University(康考迪亚大学);Université de Montréal(蒙特利尔大学);Mila(米拉研究所)

Boris Knyazev, Abhinav Moudgil, Guillaume Lajoie, Eugene Belilovsky, Simon Lacoste-Julien

AI总结:

本文提出NiNo网络,利用神经元连接性和图神经网络改进权重临近预测,解决Transformer中连接性建模难题,在视觉和语言任务中将Adam训练加速高达50%。

摘要:

当使用可学习更新规则替代经典自适应优化器(如Adam)时,神经网络训练可以被加速。然而,可学习更新规则在训练和使用过程中可能代价高昂且不稳定。近期,Jang等人(2023)提出了一种基于权重临近预测网络(WNNs)的简化方法来加速训练。在他们的方法中,Adam被用于大多数优化步骤,并且仅每隔几步,WNN会临近预测(预测近期未来)参数。我们通过提出神经元交互与临近预测(NiNo)网络来改进WNNs。与WNNs不同,NiNo利用神经元连接性和图神经网络来更准确地临近预测参数。我们进一步表明,在某些网络(如Transformer)中,准确建模神经元连接性具有挑战性。我们解决了这一问题及其他限制,使得NiNo在视觉和语言任务中将Adam训练加速高达50%。

英文摘要:

Neural network training can be accelerated when a learnable update rule is used in lieu of classic adaptive optimizers (e.g. Adam). However, learnable update rules can be costly and unstable to train and use. Recently, Jang et al. (2023) proposed a simpler approach to accelerate training based on weight nowcaster networks (WNNs). In their approach, Adam is used for most of the optimization steps and periodically, only every few steps, a WNN nowcasts (predicts near future) parameters. We improve WNNs by proposing neuron interaction and nowcasting (NiNo) networks. In contrast to WNNs, NiNo leverages neuron connectivity and graph neural networks to more accurately nowcast parameters. We further show that in some networks, such as Transformers, modeling neuron connectivity accurately is challenging. We address this and other limitations, which allows NiNo to accelerate Adam training by up to 50% in vision and language tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑