← Search

Johanni Brea

7 accepted papers

2025

Flat Channels to Infinity in Neural Loss Landscapes

NeurIPS 2025poster

The loss landscapes of neural networks contain minima and saddle points that may be connected in flat regions or appear in isolation. We identify and characterize a special structure in the loss landscape: channels along which the loss decreases extremely slowly, while the output weights of at least…

Cited by 0SourceScholar
2024

Expand-and-Cluster: Parameter Recovery of Neural Networks

ICML 2024poster

Can we identify the weights of a neural network by probing its input-output mapping? At first glance, this problem seems to have many solutions because of permutation, overparameterisation and activation function symmetries. Yet, we show that the incoming weight vector of each neuron is identifiable…

2023

Should Under-parameterized Student Networks Copy or Average Teacher Weights?

NeurIPS 2023poster

Any continuous function $f^*$ can be approximated arbitrarily well by a neural network with sufficiently many neurons $k$. We consider the case when $f^*$ itself is a neural network with one hidden layer and $k$ neurons. Approximating $f^*$ with a neural network with $n< k$ neurons can thus be seen…

2022

Kernel Memory Networks: A Unifying Framework for Memory Modeling

NeurIPS 2022accept

We consider the problem of training a neural network to store a set of patterns with maximal noise robustness. A solution, in terms of optimal weights and state update rules, is derived by training each individual neuron to perform either kernel classification or interpolation with a minimum weight…

Cited by 10SourcePDFScholar
2021

Fitting summary statistics of neural data with a differentiable spiking network simulator

NeurIPS 2021poster

Fitting network models to neural activity is an important tool in neuroscience. A popular approach is to model a brain area with a probabilistic recurrent spiking network whose parameters maximize the likelihood of the recorded activity. Although this is widely used, we show that the resulting model…

2021

Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariances

ICML 2021spotlight

We study how permutation symmetries in overparameterized multi-layer neural networks generate ‘symmetry-induced’ critical points. Assuming a network with $ L $ layers of minimal widths $ r_1^*, \ldots, r_{L-1}^* $ reaches a zero-loss minimum at $ r_1^*! \cdots r_{L-1}^*! $ isolated points that are p…

2018

Efficient Model-Based Deep Reinforcement Learning with Variational State Tabulation

ICML 2018oral

Modern reinforcement learning algorithms reach super-human performance on many board and video games, but they are sample inefficient, i.e. they typically require significantly more playing experience than humans to reach an equal performance level. To improve sample efficiency, an agent may build a…