← Search

Edo Cohen-Karlik

5 accepted papers

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
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
2020

Regularizing Towards Permutation Invariance In Recurrent Models

NeurIPS 2020poster

In many machine learning problems the output should not depend on the order of the inputs. Such ``permutation invariant'' functions have been studied extensively recently. Here we argue that temporal architectures such as RNNs are highly relevant for such problems, despite the inherent dependence of…