Skip to main navigation Skip to search Skip to main content

When will gradient methods converge to max-margin classifier under ReLU models?

  • Tengyu Xu
  • , Yi Zhou
  • , Kaiyi Ji
  • , Yingbin Liang
  • Ohio State University
  • University of Utah

Research output: Contribution to journal › Article › peer-review

4 Scopus citations

Abstract

We study the implicit bias of gradient descent methods in solving a binary classification problem over a linearly separable data set. The classifier is described by a non-linear ReLU model and the objective function adopts the exponential loss function. We first characterize the landscape of the loss function and show that there can exist spurious asymptotic local minima besides asymptotic global minima. We then show that gradient descent (GD) can converge to either a global or a local max-margin direction or may diverge from the desired max-margin direction in a general context. For stochastic gradient descent (SGD), we show that it converges in expectation to either the global or the local max-margin direction if SGD converges. We further explore the implicit bias of these algorithms in learning a multineuron network under certain stationary conditions and show that the learned classifier maximizes the margins of each sample pattern partition under the ReLU activation.

Original languageEnglish
Article numbere354
JournalStat
Volume10
Issue number1
DOIs
StatePublished - Dec 2021

Fingerprint

Dive into the research topics of 'When will gradient methods converge to max-margin classifier under ReLU models?'. Together they form a unique fingerprint.

Cite this