arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.04194cs.LGcs.AIcs.CV
收藏

神经电容:基于边缘动力学的神经网络选择新视角

Neural Capacitance: A New Perspective of Neural Network Selection via Edge Dynamics

RPI(伦斯勒理工学院);IBM Research(IBM研究院);UCLA(加州大学洛杉矶分校)

Chunheng Jiang, Tejaswini Pedapati, Pin-Yu Chen, Yizhou Sun, Jianxi Gao

摘要:

在深度学习中,有效的模型选择用于识别适合于下游任务的预训练神经网络是一项基本但具有挑战性的任务。目前的做法需要昂贵的计算成本的模型训练性能预测。在本文中,我们提出了一个新的框架神经网络选择通过分析在训练过程中突触连接(边缘)的支配动力学。我们的框架是建立在这样一个事实,即反向传播在神经网络训练是等效的突触连接的动态演变。因此,收敛的神经网络与由这些边组成的网络系统的平衡状态相关联。为此,我们构建了一个网络映射$ϕ$,将神经网络$G_A$转换为定义在$G_A$中的那些边上的有向线图$G_B$。接下来,我们推导出一个神经电容度量$β_{\rm eff}$作为一个预测性的措施,普遍捕获的泛化能力$G_A$的下游任务,使用只有少数早期的训练结果。我们使用17个流行的预训练ImageNet模型和五个基准数据集(包括CIFAR10,CIFAR100,SVHN,Fashion MNIST和Birds)进行了广泛的实验,以评估我们框架的微调性能。我们的神经电容度量被证明是仅基于早期训练结果的模型选择的强大指标,并且比最先进的方法更有效。

英文摘要:

Efficient model selection for identifying a suitable pre-trained neural network to a downstream task is a fundamental yet challenging task in deep learning. Current practice requires expensive computational costs in model training for performance prediction. In this paper, we propose a novel framework for neural network selection by analyzing the governing dynamics over synaptic connections (edges) during training. Our framework is built on the fact that back-propagation during neural network training is equivalent to the dynamical evolution of synaptic connections. Therefore, a converged neural network is associated with an equilibrium state of a networked system composed of those edges. To this end, we construct a network mapping $ϕ$, converting a neural network $G_A$ to a directed line graph $G_B$ that is defined on those edges in $G_A$. Next, we derive a neural capacitance metric $β_{\rm eff}$ as a predictive measure universally capturing the generalization capability of $G_A$ on the downstream task using only a handful of early training results. We carried out extensive experiments using 17 popular pre-trained ImageNet models and five benchmark datasets, including CIFAR10, CIFAR100, SVHN, Fashion MNIST and Birds, to evaluate the fine-tuning performance of our framework. Our neural capacitance metric is shown to be a powerful indicator for model selection based only on early training results and is more efficient than state-of-the-art methods.

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

↑