arrow
Return

Network properties determine neural network performance

delete2024-07-08
delete1
delete
OA
AI
C
Chunheng Jiang
Z
Z.-Q. Huang
T
Tejaswini Pedapati
P
Pin‐Yu Chen
Y
Yizhou Sun
J
Jianxi Gao *
DOI:10.1038/s41467-024-48069-8delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Machine learning influences numerous aspects of modern society, empowers new technologies, from Alphago to ChatGPT, and increasingly materializes in consumer products such as smartphones and self-driving cars. Despite the vital role and broad applications of artificial neural networks, we lack systematic approaches, such as network science, to understand their underlying mechanism. The difficulty is rooted in many possible model configurations, each with different hyper-parameters and weighted architectures determined by noisy data. We bridge the gap by developing a mathematical framework that maps the neural network's performance to the network characters of the line graph governed by the edge dynamics of stochastic gradient descent differential equations. This framework enables us to derive a neural capacitance metric to universally capture a model's generalization capability on a downstream task and predict model performance using only early training results. The numerical results on 17 pre-trained ImageNet models across five benchmark datasets and one NAS benchmark indicate that our neural capacitance metric is a powerful indicator for model selection based only on early training results and is more efficient than state-of-the-art methods. Understanding of artificial neural networks function, and their ability to effectively solve specific tasks, still require more rigorous analytical background. Using network science and dynamical systems tools, the authors develop a framework for predicting the performance of artificial neural networks
Keywords:
RESILIENCE
SYSTEMS
CURVES
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Nature Communications cover
Nature Communications
IF:
15.7
Papers:
9.2W
Citations:
91.2W

Organization

R
rensselaer polytechnic institute
Scholars:
7.0K
Papers: 6.5K
Citations: 6
I
international business machines (ibm)
Scholars:
5.7K
Papers: 4.5K
Citations: 4
University of California System cover
University of California System
Scholars:
37.5W
Papers: 33.7W
Citations: 6.6K
researcher View more organizations