4 papers · 1 filter
Don't Stop Me Yet: Sampling Loss Minima via Dissipative Riemannian Mechanics
Albert Kjøller Jacobsen, Leo Uhre Jakobsen, Johanna Marie Gegenfurtner +1
The minima of modern neural network loss functions are typically not isolated, rather they form connected components of reparameterization invariant solutions on the training data.…
Reducing Memorisation in Generative Models via Riemannian Bayesian Inference
Johanna Marie Gegenfurtner, Albert Kjøller Jacobsen, Naima Elosegui Borras +2
Modern generative models can produce realistic samples, however, balancing memorisation and generalisation remains an open problem. We approach this challenge from a Bayesian persp…
Staying on the Manifold: Geometry-Aware Noise Injection
Albert Kjøller Jacobsen, Johanna Marie Gegenfurtner, Georgios Arvanitidis
It has been shown that perturbing the input during training implicitly regularises the gradient of the learnt function, leading to smoother models and enhancing generalisation. How…
Monge SAM: Robust Reparameterization-Invariant Sharpness-Aware Minimization Based on Loss Geometry
Albert Kjøller Jacobsen, Georgios Arvanitidis
Recent studies on deep neural networks show that flat minima of the loss landscape correlate with improved generalization. Sharpness-aware minimization (SAM) efficiently finds flat…