Yunwen Lei

Dr. Yunwen Lei


Research Focus & Metrics

Academic Career Length

11Years (2016-Present)

(Calculated once Scilit publication data is synced: current year − year of first publication + 1)

Representative Publications

View All Publications
11 December 2019 in IEEE Transactions on Neural Networks and Learning Systems
Verified
Journal Article
Closed access
YLYunwen Lei
Yunwen Lei
THTing Hu
Ting Hu
GLGuiying Li
Guiying Li
KTKe Tang
Ke Tang
Stochastic gradient descent (SGD) is a popular and efficient method with wide applications in training deep neural nets and other nonconvex models. While the behavior of SGD is well understood in the convex learning setting, the existing theoretical results for SGD applied to nonconvex objective functions are far from mature. For example, existing results require to impose a nontrivial assumption on the uniform boundedness of gradients for all iterates encountered in the learning process, which is hard to verify in practical implementations. In this article, we establish a rigorous theoretical foundation for SGD in nonconvex learning by showing that this boundedness assumption can be removed without affecting convergence rates, and relaxing the standard smoothness assumption to Hölder continuity of gradients. In particular, we establish sufficient conditions for almost sure convergence as well as optimal convergence rates for SGD applied to both general nonconvex and gradient-dominated objective functions. A linear convergence is further derived in the case with zero variances.

20 January 2019 in IEEE Transactions on Information Theory
Verified
Journal Article
Closed access
YLYunwen Lei
Yunwen Lei
UDUrun Dogan
Urun Dogan
DZDing-Xuan Zhou
Ding-Xuan Zhou
MKMarius Kloft
Marius Kloft
In this paper, we study data-dependent generalization error bounds that exhibit a mild dependency on the number of classes, making them suitable for multi-class learning with a large number of label classes. The bounds generally hold for empirical multi-class risk minimization algorithms using an arbitrary norm as the regularizer. Key to our analysis is new structural results for multi-class Gaussian complexities and empirical ℓ ∞ -norm covering numbers, which exploit the Lipschitz continuity of the loss function with respect to the ℓ 2 - and ℓ ∞ -norm, respectively. We establish data-dependent error bounds in terms of the complexities of a linear function class defined on a finite set induced by training examples, for which we show tight lower and upper bounds. We apply the results to several prominent multi-class learning machines and show a tighter dependency on the number of classes than the state of the art. For instance, for the multi-class support vector machine of Crammer and Singer (2002), we obtain a data-dependent bound with a logarithmic dependency, which is a significant improvement of the previous square-root dependency. The experimental results are reported to verify the effectiveness of our theoretical findings.

23 March 2021 in IEEE Transactions on Pattern Analysis and Machine Intelligence
Verified
Journal Article
Closed access
YLYunwen Lei
Yunwen Lei
KTKe Tang
Ke Tang
Stochastic gradient descent (SGD) has become the method of choice for training highly complex and nonconvex models since it can not only recover good solutions to minimize training errors but also generalize well. Computational and statistical properties are separately studied to understand the behavior of SGD in the literature. However, there is a lacking study to jointly consider the computational and statistical properties in a nonconvex learning setting. In this paper, we develop novel learning rates of SGD for nonconvex learning by presenting high-probability bounds for both computational and statistical errors. We show that the complexity of SGD iterates grows in a controllable manner with respect to the iteration number, which sheds insights on how an implicit regularization can be achieved by tuning the number of passes to balance the computational and statistical errors. As a byproduct, we also slightly refine the existing studies on the uniform convergence of gradients by showing its connection to Rademacher chaos complexities.

1 January 2022 in Applied and Computational Harmonic Analysis
Verified
Journal Article
Closed access
PWPuyu Wang
Puyu Wang
YLYunwen Lei
Yunwen Lei
YYYiming Ying
Yiming Ying
HZHai Zhang
Hai Zhang

1 January 2020 in Applied and Computational Harmonic Analysis
Verified
Journal Article
Closed access
YLYunwen Lei
Yunwen Lei
DZDing-Xuan Zhou
Ding-Xuan Zhou

1 March 2017 in Neural Computation
Verified
Journal Article
Closed access
YLYunwen Lei
Yunwen Lei
DZDing-Xuan Zhou
Ding-Xuan Zhou
We study the convergence of the online composite mirror descent algorithm, which involves a mirror map to reflect the geometry of the data and a convex objective function consisting of a loss and a regularizer possibly inducing sparsity. Our error analysis provides convergence rates in terms of properties of the strongly convex differentiable mirror map and the objective function. For a class of objective functions with Hölder continuous gradients, the convergence rates of the excess (regularized) risk under polynomially decaying step sizes have the order [ ] after [ ] iterates. Our results improve the existing error analysis for the online composite mirror descent algorithm by avoiding averaging and removing boundedness assumptions, and they sharpen the existing convergence rates of the last iterate for online gradient descent without any boundedness assumptions. Our methodology mainly depends on a novel error decomposition in terms of an excess Bregman distance, refined analysis of self-bounding properties of the objective function, and the resulting one-step progress bounds.