← Search

Nadav Cohen

23 accepted papers

2025

DeciMamba: Exploring the Length Extrapolation Potential of Mamba

ICLR 2025poster

Long-range sequence processing poses a significant challenge for Transformers due to their quadratic complexity in input length. A promising alternative is Mamba, which demonstrates high performance and achieves Transformer-level capabilities while requiring substantially fewer computational resourc…

2025

Do Neural Networks Need Gradient Descent to Generalize? A Theoretical Study

NeurIPS 2025poster

Conventional wisdom attributes the mysterious generalization abilities of overparameterized neural networks to gradient descent (and its variants). The recent volume hypothesis challenges this view: it posits that these generalization abilities persist even when gradient descent is replaced by Guess…

Cited by 0SourceScholar
2025

The Implicit Bias of Structured State Space Models Can Be Poisoned With Clean Labels

NeurIPS 2025spotlight

Neural networks are powered by an implicit bias: a tendency of gradient descent to fit training data in a way that generalizes to unseen data. A recent class of neural network models gaining increasing popularity is structured state space models (SSMs). Prior work argued that the implicit bias of SS…

Cited by 0SourcecodeScholar
2024

Data-driven Coreference-based Ontology Building

EMNLP 2024finding

While coreference resolution is traditionally used as a component in individual document understanding, in this work we take a more global view and explore what can we learn about a domain from the set of all document-level coreference relations that are present in a large corpus. We derive corefere…

2024

Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial States

ICML 2024poster

In modern machine learning, models can often fit training data in numerous ways, some of which perform well on unseen (test) data, while others do not. Remarkably, in such cases gradient descent frequently exhibits an implicit bias that leads to excellent performance on unseen data. This implicit bi…

2024

Provable Benefits of Complex Parameterizations for Structured State Space Models

NeurIPS 2024poster

Structured state space models (SSMs), the core engine behind prominent neural networks such as S4 and Mamba, are linear dynamical systems adhering to a specified structure, most notably diagonal. In contrast to typical neural network modules, whose parameterizations are real, SSMs often use complex…

2023

Learning Low Dimensional State Spaces with Overparameterized Recurrent Neural Nets

ICLR 2023poster

Overparameterization in deep learning refers to settings where a trained Neural Network (NN) has representational capacity to fit the training data in many ways, some of which generalize well, while others do not. In the case of Recurrent Neural Networks (RNNs) there exists an additional layer of ov…

Cited by 5SourcePDFScholar
2023

On the Ability of Graph Neural Networks to Model Interactions Between Vertices

NeurIPS 2023poster

Graph neural networks (GNNs) are widely used for modeling complex interactions between entities represented as vertices of a graph. Despite recent efforts to theoretically analyze the expressive power of GNNs, a formal characterization of their ability to model interactions is lacking. The current p…

2023

What Makes Data Suitable for a Locally Connected Neural Network? A Necessary and Sufficient Condition Based on Quantum Entanglement.

NeurIPS 2023spotlight

The question of what makes a data distribution suitable for deep learning is a fundamental open problem. Focusing on locally connected neural networks (a prevalent family of architectures that includes convolutional and recurrent neural networks as well as local self-attention models), we address th…

2022

Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networks

ICML 2022spotlight

In the pursuit of explaining implicit regularization in deep learning, prominent focus was given to matrix and tensor factorizations, which correspond to simplified neural networks. It was shown that these models exhibit an implicit tendency towards low matrix and tensor ranks, respectively. Drawing…

2022

On the Implicit Bias of Gradient Descent for Temporal Extrapolation

AISTATS 2022poster

When using recurrent neural networks (RNNs) it is common practice to apply trained models to sequences longer than those seen in training. This “extrapolating” usage deviates from the traditional statistical learning setup where guarantees are provided under the assumption that train and test distri…

Cited by 6SourcePDFScholar
2019

A Convergence Analysis of Gradient Descent for Deep Linear Neural Networks

ICLR 2019poster

We analyze speed of convergence to global optimum for gradient descent training a deep linear neural network by minimizing the L2 loss over whitened data. Convergence at a linear rate is guaranteed when the following hold: (i) dimensions of hidden layers are at least the minimum of the input and ou…

Cited by 322SourcePDFScholar
2019

Implicit Regularization in Deep Matrix Factorization

NeurIPS 2019spotlight

Efforts to understand the generalization mystery in deep learning have led to the belief that gradient-based optimization induces a form of implicit regularization, a bias towards models of low "complexity." We study the implicit regularization of gradient descent over deep linear neural networks f…

2018

Deep Learning and Quantum Entanglement: Fundamental Connections with Implications to Network Design

ICLR 2018poster

Formal understanding of the inductive bias behind deep convolutional networks, i.e. the relation between the network's architectural features and the functions it is able to model, is limited. In this work, we establish a fundamental connection between the fields of quantum physics and deep learning…

Cited by 147SourcePDFScholar
2018

On the Optimization of Deep Networks: Implicit Acceleration by Overparameterization

ICML 2018oral

Conventional wisdom in deep learning states that increasing depth improves expressiveness but complicates optimization. This paper suggests that, sometimes, increasing depth can speed up optimization. The effect of depth on optimization is decoupled from expressiveness by focusing on settings where…

2018

“Zero-Shot” Super-Resolution Using Deep Internal Learning

CVPR 2018poster

Deep Learning has led to a dramatic leap in Super-Resolution (SR) performance in the past few years. However, being supervised, these SR methods are restricted to specific training data, where the acquisition of the low-resolution (LR) images from their high-resolution (HR) counterparts is predeterm…

2016

Deep SimNets

CVPR 2016poster

We present a deep layered architecture that generalizes convolutional neural networks (ConvNets). The architecture, called SimNets, is driven by two operators: (i) a similarity function that generalizes inner-product, and (ii) a log-mean-exp function called MEX that generalizes maximum and average.…

Cited by 57PDFScholar