arXiv subjects

Bin Shi

Publications and source records attributed to Bin Shi.

61 records · Page 4Linked to original sources

arXiv2022

On the Hyperparameters in Stochastic Gradient Descent with Momentum

Following the same routine as [SSJ20], we continue to present the theoretical analysis for stochastic gradient descent with momentum (SGD with momentum) in this paper. Differently, for SGD with momentum, we demonstrate it is the two hyperparameters together, the learning rate and the momentum coefficient, that play the significant role for the linear rate of convergence in non-convex optimization. Our analysis is based on the use of a hyperparameters-dependent stochastic differential equation (hp-dependent SDE) that serves as a continuous surrogate for SGD with momentum. Similarly, we establish the linear convergence for the continuous-time formulation of SGD with momentum and obtain an explicit expression for the optimal linear rate by analyzing the spectrum of the Kramers-Fokker-Planck operator. By comparison, we demonstrate how the optimal linear rate of convergence and the final gap for SGD only about the learning rate varies with the momentum coefficient increasing from zero to one when the momentum is introduced. Then, we propose a mathematical interpretation why the SGD with momentum converges faster and more robust about the learning rate than the standard SGD in practice. Finally, we show the Nesterov momentum under the existence of noise has no essential difference with the standard momentum.

cs.LG↗

arXiv2022

Comparison of Effect Size Measures for Mediation Analysis of Survival Outcomes with Application to the Framingham Heart Study

There is an increasing trend of research in mediation analysis for survival outcomes. Such analyses help researchers to better understand how exposure affects disease outcomes through mediators. However, due to censored observations in survival outcomes, it is not straightforward to extend mediation analysis from linear models to survival outcomes. In this article, we extend a mediation effect size measure based on $R^2$ in linear regression to survival outcomes. Due to multiple definitions of $R^2$ for survival models, we compare and evaluate five $R^2$ measures for mediation analysis. Based on extensive simulations, we recommend two $R^2$ measures with good operating characteristics. We illustrate the utility of the $R^2$-based mediation measures by analyzing the mediation effects of multiple lifestyle risk factors on the relationship between environmental exposures and time to coronary heart disease and all-cause mortality in the Framingham Heart Study.

stat.ME↗

arXiv2021

Conjugate and cut points in ideal fluid motion

Two fluid configurations along a flow are conjugate if there is a one parameter family of geodesics (fluid flows) joining them to infinitesimal order. Geometrically, they can be seen as a consequence of the (infinite dimensional) group of volume preserving diffeomorphisms having sufficiently strong positive curvatures which `pull' nearby flows together. Physically, they indicate a form of (transient) stability in the configuration space of particle positions: a family of flows starting with the same configuration deviate initially and subsequently re-converge (resonate) with each other at some later moment in time. Here, we first establish existence of conjugate points in an infinite family of Kolmogorov flows - a class of stationary solutions of the Euler equations - on the rectangular flat torus of any aspect ratio. The analysis is facilitated by a general criterion for identifying conjugate points in the group of volume preserving diffeomorphisms. Next, we show non-existence of conjugate points along Arnold stable steady states on the annulus, disk and channel. Finally, we discuss cut points, their relation to non-injectivity of the exponential map (impossibility of determining a flow from a particle configuration at a given instant) and show that the closest cut point to the identity is either a conjugate point or the midpoint of a time periodic Lagrangian fluid flow.

math.AP↗

arXiv2020

On Learning Rates and Schrödinger Operators

The learning rate is perhaps the single most important parameter in the training of neural networks and, more broadly, in stochastic (nonconvex) optimization. Accordingly, there are numerous effective, but poorly understood, techniques for tuning the learning rate, including learning rate decay, which starts with a large initial learning rate that is gradually decreased. In this paper, we present a general theoretical analysis of the effect of the learning rate in stochastic gradient descent (SGD). Our analysis is based on the use of a learning-rate-dependent stochastic differential equation (lr-dependent SDE) that serves as a surrogate for SGD. For a broad class of objective functions, we establish a linear rate of convergence for this continuous-time formulation of SGD, highlighting the fundamental importance of the learning rate in SGD, and contrasting to gradient descent and stochastic gradient Langevin dynamics. Moreover, we obtain an explicit expression for the optimal linear rate by analyzing the spectrum of the Witten-Laplacian, a special case of the Schrödinger operator associated with the lr-dependent SDE. Strikingly, this expression clearly reveals the dependence of the linear convergence rate on the learning rate -- the linear rate decreases rapidly to zero as the learning rate tends to zero for a broad class of nonconvex functions, whereas it stays constant for strongly convex functions. Based on this sharp distinction between nonconvex and convex problems, we provide a mathematical interpretation of the benefits of using learning rate decay for nonconvex optimization.

cs.LG↗

arXiv2019

Acceleration via Symplectic Discretization of High-Resolution Differential Equations

We study first-order optimization methods obtained by discretizing ordinary differential equations (ODEs) corresponding to Nesterov's accelerated gradient methods (NAGs) and Polyak's heavy-ball method. We consider three discretization schemes: an explicit Euler scheme, an implicit Euler scheme, and a symplectic scheme. We show that the optimization algorithm generated by applying the symplectic scheme to a high-resolution ODE proposed by Shi et al. [2018] achieves an accelerated rate for minimizing smooth strongly convex functions. On the other hand, the resulting algorithm either fails to achieve acceleration or is impractical when the scheme is implicit, the ODE is low-resolution, or the scheme is explicit.

math.OC↗

arXiv2018

Understanding the Acceleration Phenomenon via High-Resolution Differential Equations

Gradient-based optimization algorithms can be studied from the perspective of limiting ordinary differential equations (ODEs). Motivated by the fact that existing ODEs do not distinguish between two fundamentally different algorithms---Nesterov's accelerated gradient method for strongly convex functions (NAG-SC) and Polyak's heavy-ball method---we study an alternative limiting process that yields high-resolution ODEs. We show that these ODEs permit a general Lyapunov function framework for the analysis of convergence in both continuous and discrete time. We also show that these ODEs are more accurate surrogates for the underlying algorithms; in particular, they not only distinguish between NAG-SC and Polyak's heavy-ball method, but they allow the identification of a term that we refer to as "gradient correction" that is present in NAG-SC but not in the heavy-ball method and is responsible for the qualitative difference in convergence of the two methods. We also use the high-resolution ODE framework to study Nesterov's accelerated gradient method for (non-strongly) convex functions, uncovering a hitherto unknown result---that NAG-C minimizes the squared gradient norm at an inverse cubic rate. Finally, by modifying the high-resolution ODE of NAG-C, we obtain a family of new optimization methods that are shown to maintain the accelerated convergence rates of NAG-C for smooth convex functions.

math.OC↗

arXiv2017

A Conservation Law Method in Optimization

We propose some algorithms to find local minima in nonconvex optimization and to obtain global minima in some degree from the Newton Second Law without friction. With the key observation of the velocity observable and controllable in the motion, the algorithms simulate the Newton Second Law without friction based on symplectic Euler scheme. From the intuitive analysis of analytical solution, we give a theoretical analysis for the high-speed convergence in the algorithm proposed. Finally, we propose the experiments for strongly convex function, non-strongly convex function and nonconvex function in high-dimension.

math.OC↗