← Search

Shai Shalev-Shwartz

16 accepted papers

2025

Jamba: Hybrid Transformer-Mamba Language Models

ICLR 2025poster

We present Jamba, a novel hybrid Transformer-Mamba mixture-of-experts (MoE) architecture. Jamba interleaves blocks of Transformer and Mamba layers, enjoying the benefits of both model families. MoE is added in some of these layers to increase model capacity while keeping active parameter usage manag…

Cited by 3SourcePDFScholar
2022

Knowledge Distillation: Bad Models Can Be Good Role Models

NeurIPS 2022accept

Large neural networks trained in the overparameterized regime are able to fit noise to zero train error. Recent work of Nakkiran and Bansal has empirically observed that such networks behave as “conditional samplers” from the noisy distribution. That is, they replicate the noise in the train data to…

Cited by 21SourcePDFScholar
2020

The Implicit Bias of Depth: How Incremental Learning Drives Generalization

ICLR 2020poster

A leading hypothesis for the surprising generalization of neural networks is that the dynamics of gradient descent bias the model towards simple solutions, by searching through the solution space in an incremental order of complexity. We formally define the notion of incremental learning dynamics an…

Cited by 92SourcecodeScholar
2018

SGD Learns Over-parameterized Networks that Provably Generalize on Linearly Separable Data

ICLR 2018poster

Neural networks exhibit good generalization behavior in the over-parameterized regime, where the number of network parameters exceeds the number of observations. Nonetheless, current generalization bounds for neural networks fail to explain this phenomenon. In an attempt to bridge this gap, we study…

Cited by 305SourcePDFScholar
2016

Learning a Metric Embedding for Face Recognition using the Multibatch Method

NeurIPS 2016poster

This work is motivated by the engineering task of achieving a near state-of-the-art face recognition on a minimal computing budget running on an embedded system. Our main technical contribution centers around a novel training method, called Multibatch, for similarity learning, i.e., for the task of…

Cited by 67SourcePDFScholar