← Search

Zhenxun Zhuang

5 accepted papers

2022

A Communication-Efficient Distributed Gradient Clipping Algorithm for Training Deep Neural Networks

NeurIPS 2022accept

In distributed training of deep neural networks, people usually run Stochastic Gradient Descent (SGD) or its variants on each machine and communicate with other machines periodically. However, SGD might converge slowly in training some deep neural networks (e.g., RNN, LSTM) because of the exploding…

2022

Robustness to Unbounded Smoothness of Generalized SignSGD

NeurIPS 2022accept

Traditional analyses in non-convex optimization typically rely on the smoothness assumption, namely requiring the gradients to be Lipschitz. However, recent evidence shows that this smoothness condition does not capture the properties of some deep learning objective functions, including the ones inv…

Cited by 83SourcePDFScholar
2021

A Second look at Exponential and Cosine Step Sizes: Simplicity, Adaptivity, and Performance

ICML 2021spotlight

Stochastic Gradient Descent (SGD) is a popular tool in training large-scale machine learning models. Its performance, however, is highly variable, depending crucially on the choice of the step sizes. Accordingly, a variety of strategies for tuning the step sizes have been proposed, ranging from coor…

2019

Surrogate Losses for Online Learning of Stepsizes in Stochastic Non-Convex Optimization

ICML 2019oral

Stochastic Gradient Descent (SGD) has played a central role in machine learning. However, it requires a carefully hand-picked stepsize for fast convergence, which is notoriously tedious and time-consuming to tune. Over the last several years, a plethora of adaptive gradient-based algorithms have eme…

Cited by 7SourcePDFScholar