Machine Learning: Why Do Simple Algorithms Work So Well?
- Author(s): Jin, Chi
- Advisor(s): Jordan, Michael I
- et al.
While state-of-the-art machine learning models are deep, large-scale, sequential and highly nonconvex, the backbone of modern learning algorithms are simple algorithms such as stochastic gradient descent, gradient descent with momentum or Q-learning (in the case of reinforcement learning tasks). A basic question endures---why do simple algorithms work so well even in these challenging settings?
To answer above question, this thesis focuses on four concrete and fundamental questions:
- In nonconvex optimization, can (stochastic) gradient descent or its variants escape saddle points efficiently?
- Is gradient descent with momentum provably faster than gradient descent in the general nonconvex setting?
- In nonconvex-nonconcave minmax optimization, what is a proper definition of local optima and is gradient descent ascent game-theoretically meaningful?
- In reinforcement learning, is Q-learning sample efficient?
This thesis provides the first line of provably positive answers to all above questions. In particular, this thesis will show that although the standard versions of these classical algorithms do not enjoy good theoretical properties in the worst case, simple modifications are sufficient to grant them desirable behaviors, which explain the underlying mechanisms behind their favorable performance in practice.