English
Related papers

Related papers: A Framework for Provably Stable and Consistent Tra…

200 papers

Real-world analog systems intrinsically suffer from noise that can impede model convergence and accuracy on a variety of deep learning models. We demonstrate that differentiable activations like GELU and SiLU enable robust propagation of…

Machine Learning · Computer Science 2025-02-25 Vivswan Shah , Nathan Youngblood

The ability of neural networks to provide `best in class' approximation across a wide range of applications is well-documented. Nevertheless, the powerful expressivity of neural networks comes to naught if one is unable to effectively train…

Machine Learning · Computer Science 2020-07-15 Mark Ainsworth , Yeonjong Shin

Stability arguments are often used to prevent learning algorithms from having ever increasing activity and weights that hinder generalization. However, stability conditions can clash with the sparsity required to augment the energy…

Neural and Evolutionary Computing · Computer Science 2024-01-08 Luca Herranz-Celotti , Jean Rouat

Deep learning has been widely used in many fields, but the model training process usually consumes massive computational resources and time. Therefore, designing an efficient neural network training method with a provable convergence…

Machine Learning · Computer Science 2023-07-14 Lianke Qin , Zhao Song , Yuanyuan Yang

In recent years, stochastic gradient descent (SGD) based techniques has become the standard tools for training neural networks. However, formal theoretical understanding of why SGD can train neural networks in practice is largely missing.…

Machine Learning · Computer Science 2017-11-03 Yuanzhi Li , Yang Yuan

The objective of this paper is to enhance the optimization process for neural networks by developing a dynamic learning rate algorithm that effectively integrates exponential decay and advanced anti-overfitting strategies. Our primary…

Machine Learning · Computer Science 2025-08-04 Jatin Chaudhary , Dipak Nidhi , Jukka Heikkonen , Haari Merisaari , Rajiv Kanth

We discover restrained numerical instabilities in current training practices of deep networks with stochastic gradient descent (SGD), and its variants. We show numerical error (on the order of the smallest floating point bit and thus the…

Machine Learning · Computer Science 2024-06-13 Yuxin Sun , Dong Lao , Ganesh Sundaramoorthi , Anthony Yezzi

Numerous theories of learning propose to prevent the gradient from exponential growth with depth or time, to stabilize and improve training. Typically, these analyses are conducted on feed-forward fully-connected neural networks or simple…

Machine Learning · Computer Science 2024-01-08 Luca Herranz-Celotti , Jean Rouat

Deep learning empirically achieves high performance in many applications, but its training dynamics has not been fully understood theoretically. In this paper, we explore theoretical analysis on training two-layer ReLU neural networks in a…

Machine Learning · Statistics 2021-06-30 Shunta Akiyama , Taiji Suzuki

We study the overparametrization bounds required for the global convergence of stochastic gradient descent algorithm for a class of one hidden layer feed-forward neural networks, considering most of the activation functions used in…

Machine Learning · Computer Science 2022-11-17 Bartłomiej Polaczyk , Jacek Cyranka

The training of neural networks by gradient descent methods is a cornerstone of the deep learning revolution. Yet, despite some recent progress, a complete theory explaining its success is still missing. This article presents, for…

Machine Learning · Statistics 2026-04-15 Etienne Boursier , Loucas Pillaud-Vivien , Nicolas Flammarion

State-of-the-art training algorithms for deep learning models are based on stochastic gradient descent (SGD). Recently, many variations have been explored: perturbing parameters for better accuracy (such as in Extragradient), limiting SGD…

Machine Learning · Computer Science 2022-03-23 Amirkeivan Mohtashami , Martin Jaggi , Sebastian U. Stich

We study the training of deep neural networks by gradient descent where floating-point arithmetic is used to compute the gradients. In this framework and under realistic assumptions, we demonstrate that it is highly unlikely to find ReLU…

Machine Learning · Computer Science 2023-11-16 Clemens Karner , Vladimir Kazeev , Philipp Christian Petersen

Gradient descent (GD) type optimization schemes are the standard methods to train artificial neural networks (ANNs) with rectified linear unit (ReLU) activation. Such schemes can be considered as discretizations of gradient flows (GFs)…

Machine Learning · Computer Science 2022-09-27 Arnulf Jentzen , Adrian Riekert

We study the stability and convergence of training deep ResNets with gradient descent. Specifically, we show that the parametric branch in the residual block should be scaled down by a factor $\tau =O(1/\sqrt{L})$ to guarantee stable…

Machine Learning · Computer Science 2023-02-01 Huishuai Zhang , Da Yu , Mingyang Yi , Wei Chen , Tie-Yan Liu

Understanding the role of (stochastic) gradient descent (SGD) in the training and generalisation of deep neural networks (DNNs) with ReLU activation has been the object study in the recent past. In this paper, we make use of deep gated…

Machine Learning · Computer Science 2020-03-03 Chandrashekar Lakshminarayanan , Amit Vikram Singh

Gradient Descent (GD) and its variants are the primary tool for enabling efficient training of recurrent dynamical systems such as Recurrent Neural Networks (RNNs), Neural ODEs and Gated Recurrent units (GRUs). The dynamics that are formed…

Machine Learning · Computer Science 2025-07-10 James Hazelden , Laura Driscoll , Eli Shlizerman , Eric Shea-Brown

Neural networks are powerful functions with widespread use, but the theoretical behaviour of these functions is not fully understood. Creating deep neural networks by stacking many layers has achieved exceptional performance in many…

Machine Learning · Computer Science 2024-08-16 Cameron Jakub , Mihai Nica

While deep learning is successful in a number of applications, it is not yet well understood theoretically. A satisfactory theoretical characterization of deep learning however, is beginning to emerge. It covers the following questions: 1)…

Machine Learning · Computer Science 2019-08-27 Tomaso Poggio , Andrzej Banburski , Qianli Liao

The training of artificial neural networks (ANNs) with rectified linear unit (ReLU) activation via gradient descent (GD) type optimization schemes is nowadays a common industrially relevant procedure. Till this day in the scientific…

Machine Learning · Computer Science 2023-04-13 Simon Eberle , Arnulf Jentzen , Adrian Riekert , Georg S. Weiss