Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Junchi Yang

dblp:259/3033 · DBLP profile ↗
← Back
9ranked-venue papers
5as first author
7since 2021 · last 2024
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 9 · 5 first-author · 7 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Theoretical computer science
4 papers
Mathematical optimization · 100%
Artificial intelligence
4 papers
Optimization for machine learning · 94% Learning theory · 3% Generative modeling · 2%

Topics — the 17 heaviest of 18, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Mathematical optimization
minimax optimization
2.142023
TiAda: A Time-scale Adaptive Algorithm for Nonconvex Minimax Optimization · ICLR 2023
Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimization · NeurIPS 2022
A Catalyst Framework for Minimax Optimization · NeurIPS 2020
Mathematical optimization
nonconvex optimization
1.222023
TiAda: A Time-scale Adaptive Algorithm for Nonconvex Minimax Optimization · ICLR 2023
Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimization · NeurIPS 2022
Machine learning › Optimization for machine learning › stochastic optimization › adaptive gradient methods
adagrad
0.712023
Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methods · NeurIPS 2023
Machine learning › Optimization for machine learning
adaptive algorithm
0.712023
Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methods · NeurIPS 2023
Machine learning › Optimization for machine learning
convex optimization
0.712023
Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization · NeurIPS 2023
Machine learning › Optimization for machine learning
minimax optimization
0.712023
Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization · NeurIPS 2023
Machine learning › Optimization for machine learning › stochastic gradient descent
step size schedule
0.712023
Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methods · NeurIPS 2023
Machine learning › Optimization for machine learning
stochastic gradient descent
0.712023
Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methods · NeurIPS 2023
Machine learning › Optimization for machine learning
stochastic gradient descent ascent
0.712023
Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization · NeurIPS 2023
Machine learning › Optimization for machine learning
stochastic optimization
0.712023
Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization · NeurIPS 2023
Mathematical optimization
continuous optimization
0.612022
Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimization · NeurIPS 2022
Mathematical optimization › minimax optimization
gradient descent ascent
0.612022
Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimization · NeurIPS 2022
Mathematical optimization › continuous optimization › convex optimization › proximal methods
accelerated proximal point method
0.412020
A Catalyst Framework for Minimax Optimization · NeurIPS 2020
Mathematical optimization › continuous optimization
convex optimization
0.412020
A Catalyst Framework for Minimax Optimization · NeurIPS 2020
Mathematical optimization › minimax optimization
nonconvex-nonconcave minimax
0.412020
Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax Problems · NeurIPS 2020
Machine learning › Generative modeling
generative adversarial network
0.112020
Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax Problems · NeurIPS 2020
Machine learning › Optimization for machine learning
variance reduction
0.112020
A Catalyst Framework for Minimax Optimization · NeurIPS 2020

Methods — techniques the papers use, named apart from their topics

variance reduction · 1.7adagrad · 1.2polyak-łojasiewicz inequality · 0.9alternating gradient descent ascent · 0.9accelerated proximal point · 0.9time-scale adaptation · 0.7stepsize decay · 0.7regularization · 0.7normalized SGD · 0.7inexact oracle analysis · 0.7AMSGrad · 0.7nested adaptive framework · 0.6adaptive stepsizes · 0.6AMSGRAD · 0.6catalyst framework · 0.4
YearPublicationVenuePosition
2024 Parameter-Agnostic Optimization under Relaxed Smoothness
abstract
Tuning hyperparameters, such as the stepsize, presents a major challenge of training machine learning models. To address this challenge, numerous adaptive optimization algorithms have been developed that achieve near-optimal complexities, even when stepsizes are independent of problem-specific parameters, provided that the loss function is $L$-smooth. However, as the assumption is relaxed to the more realistic $(L_0, L_1)$-smoothness, all existing convergence results still necessitate tuning of the stepsize. In this study, we demonstrate that Normalized Stochastic Gradient Descent with Momentum (NSGD-M) can achieve a (nearly) rate-optimal complexity without prior knowledge of any problem parameter, though this comes at the cost of introducing an exponential term dependent on $L_1$ in the complexity. We further establish that this exponential term is inevitable to such schemes by introducing a theoretical framework of lower bounds tailored explicitly for parameter-agnostic algorithms. Interestingly, in deterministic settings, the exponential factor can be neutralized by employing Gradient Descent with a Backtracking Line Search. To the best of our knowledge, these findings represent the first parameter-agnostic convergence results under the generalized smoothness condition. Our empirical experiments further confirm our theoretical insights.
Florian Hübler, Junchi Yang, Niao He
AISTATS2
2023 TiAda: A Time-scale Adaptive Algorithm for Nonconvex Minimax Optimization
Junchi Yang, Niao He
ICLR2
2023 Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methods
abstract
The classical analysis of Stochastic Gradient Descent (SGD) with polynomially decaying stepsize $\eta_t = \eta/\sqrt{t}$ relies on well-tuned $\eta$ depending on problem parameters such as Lipschitz smoothness constant, which is often unknown in practice. In this work, we prove that SGD with arbitrary $\eta > 0$, referred to as untuned SGD, still attains an order-optimal convergence rate $\widetilde{\mathcal{O}}(T^{-1/4})$ in terms of gradient norm for minimizing smooth objectives. Unfortunately, it comes at the expense of a catastrophic exponential dependence on the smoothness constant, which we show is unavoidable for this scheme even in the noiseless setting. We then examine three families of adaptive methods — Normalized SGD (NSGD), AMSGrad, and AdaGrad — unveiling their power in preventing such exponential dependency in the absence of information about the smoothness parameter and boundedness of stochastic gradients. Our results provide theoretical justification for the advantage of adaptive methods over untuned SGD in alleviating the issue with large gradients.
Junchi Yang, Ilyas Fatkhullin, Niao He
NeurIPS1
2023 Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization
abstract
Algorithmic reproducibility measures the deviation in outputs of machine learning algorithms upon minor changes in the training process. Previous work suggests that first-order methods would need to trade-off convergence rate (gradient complexity) for better reproducibility. In this work, we challenge this perception and demonstrate that both optimal reproducibility and near-optimal convergence guarantees can be achieved for smooth convex minimization and smooth convex-concave minimax problems under various error-prone oracle settings. Particularly, given the inexact initialization oracle, our regularization-based algorithms achieve the best of both worlds -- optimal reproducibility and near-optimal gradient complexity -- for minimization and minimax optimization. With the inexact gradient oracle, the near-optimal guarantees also hold for minimax optimization. Additionally, with the stochastic gradient oracle, we show that stochastic gradient descent ascent is optimal in terms of both reproducibility and gradient complexity. We believe our results contribute to an enhanced understanding of the reproducibility-convergence trade-off in the context of convex optimization.
Junchi Yang, Amin Karbasi, Niao He
NeurIPS2
2022 Faster Single-loop Algorithms for Minimax Optimization without Strong Concavity
abstract
Gradient descent ascent (GDA), the simplest single-loop algorithm for nonconvex minimax optimization, is widely used in practical applications such as generative adversarial networks (GANs) and adversarial training. Albeit its desirable simplicity, recent work shows inferior convergence rates of GDA in theory, even when assuming strong concavity of the objective in terms of one variable. This paper establishes new convergence results for two alternative single-loop algorithms – alternating GDA and smoothed GDA – under the mild assumption that the objective satisfies the Polyak-Lojasiewicz (PL) condition about one variable. We prove that, to find an $\epsilon$-stationary point, (i) alternating GDA and its stochastic variant (without mini batch) respectively require $O(\kappa^{2} \epsilon^{-2})$ and $O(\kappa^{4} \epsilon^{-4})$ iterations, while (ii) smoothed GDA and its stochastic variant (without mini batch) respectively require $O(\kappa \epsilon^{-2})$ and $O(\kappa^{2} \epsilon^{-4})$ iterations. The latter greatly improves over the vanilla GDA and gives the hitherto best known complexity results among single-loop algorithms under similar settings. We further showcase the empirical efficiency of these algorithms in training GANs and robust nonlinear regression.
Junchi Yang, Antonio Orvieto, Aurélien Lucchi, Niao He
AISTATS1
2022 Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimization
abstract
Adaptive algorithms like AdaGrad and AMSGrad are successful in nonconvex optimization owing to their parameter-agnostic ability – requiring no a priori knowledge about problem-specific parameters nor tuning of learning rates. However, when it comes to nonconvex minimax optimization, direct extensions of such adaptive optimizers without proper time-scale separation may fail to work in practice. We provide such an example proving that the simple combination of Gradient Descent Ascent (GDA) with adaptive stepsizes can diverge if the primal-dual stepsize ratio is not carefully chosen; hence, a fortiori, such adaptive extensions are not parameter-agnostic. To address the issue, we formally introduce a Nested Adaptive framework, NeAda for short, that carries an inner loop for adaptively maximizing the dual variable with controllable stopping criteria and an outer loop for adaptively minimizing the primal variable. Such mechanism can be equipped with off-the-shelf adaptive optimizers and automatically balance the progress in the primal and dual variables. Theoretically, for nonconvex-strongly-concave minimax problems, we show that NeAda with AdaGrad stepsizes can achieve the near-optimal $\widetilde{O}(\epsilon^{-2})$ and $\widetilde{O}(\epsilon^{-4})$ gradient complexities respectively in the deterministic and stochastic settings, without prior information on the problem's smoothness and strong concavity parameters. To the best of our knowledge, this is the first algorithm that simultaneously achieves near-optimal convergence rates and parameter-agnostic adaptation in the nonconvex minimax setting. Numerically, we further illustrate the robustness of the NeAda family with experiments on simple test functions and a real-world application.
Junchi Yang, Niao He
NeurIPS1
2021 The complexity of nonconvex-strongly-concave minimax optimization
abstract
This paper studies the complexity for finding approximate stationary points of nonconvex-strongly-concave (NC-SC) smooth minimax problems, in both general and averaged smooth finite-sum settings. We establish nontrivial lower complexity bounds for the two settings, respectively. Our result reveals substantial gaps between these limits and best-known upper bounds in the literature. To close these gaps, we introduce a generic acceleration scheme that deploys existing gradient-based methods to solve a sequence of crafted strongly-convex-strongly-concave subproblems. In the general setting, the complexity of our proposed algorithm nearly matches the lower bound; in particular, it removes an additional poly-logarithmic dependence on accuracy present in previous works. In the averaged smooth finite-sum setting, our proposed algorithm improves over previous algorithms by providing a nearly-tight dependence on the condition number.
Junchi Yang, Cristóbal Guzmán, Negar Kiyavash, Niao He
UAI2
2020 Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax Problems
abstract
Nonconvex minimax problems appear frequently in emerging machine learning applications, such as generative adversarial networks and adversarial learning. Simple algorithms such as the gradient descent ascent (GDA) are the common practice for solving these nonconvex games and receive lots of empirical success. Yet, it is known that these vanilla GDA algorithms with constant stepsize can potentially diverge even in the convex setting. In this work, we show that for a subclass of nonconvex-nonconcave objectives satisfying a so-called two-sided Polyak-{\L}ojasiewicz inequality, the alternating gradient descent ascent (AGDA) algorithm converges globally at a linear rate and the stochastic AGDA achieves a sublinear rate. We further develop a variance reduced algorithm that attains a provably faster rate than AGDA when the problem has the finite-sum structure.
Junchi Yang, Negar Kiyavash, Niao He
NeurIPS1
2020 A Catalyst Framework for Minimax Optimization
abstract
We introduce a generic \emph{two-loop} scheme for smooth minimax optimization with strongly-convex-concave objectives. Our approach applies the accelerated proximal point framework (or Catalyst) to the associated \emph{dual problem} and takes full advantage of existing gradient-based algorithms to solve a sequence of well-balanced strongly-convex-strongly-concave minimax problems. Despite its simplicity, this leads to a family of near-optimal algorithms with improved complexity over all existing methods designed for strongly-convex-concave minimax problems. Additionally, we obtain the first variance-reduced algorithms for this class of minimax problems with finite-sum structure and establish even faster convergence rate. Furthermore, when extended to the nonconvex-concave minimax optimization, our algorithm again achieves the state-of-the-art complexity for finding a stationary point. We carry out several numerical experiments showcasing the superiority of the Catalyst framework in practice.
Junchi Yang, Negar Kiyavash, Niao He
NeurIPS1