Publications

Works from the group.

Latest work

The newest papers and preprints from the group.

Unown placeholder for The Interplay of Data Structure and Imbalance in the Learning Dynamics of Diffusion Models

May 2026 / arXiv preprint

The Interplay of Data Structure and Imbalance in the Learning Dynamics of Diffusion Models

Flavio Nicoletti, Chenxiao Ma, Enrico Ventura, Luca Saglietti, Stefano Sarao Mannelli

Real-world datasets differ across classes in both structure and frequency, but most theory for diffusion models assumes homogeneous data. This work develops a high-dimensional analytical framework for class-dependent learning in score-based diffusion models. Using a random-features model trained on...

diffusion modelsdata imbalancefairness
Unown placeholder for Position: the Stochastic Parrot in the Coal Mine. Model Collapse is a Threat to Low-Resource Communities

May 2026 / ICML 2026 (Spotlight)

Position: the Stochastic Parrot in the Coal Mine. Model Collapse is a Threat to Low-Resource Communities

Devon Jarvis, Richard Klein, Benjamin Rosman, Steven James, Stefano Sarao Mannelli

Model collapse can degrade generative models when they are trained on outputs from earlier models. This position paper argues that the problem compounds existing concerns around large language models, including cultural bias, data degradation, environmental cost, and inefficient resource...

model collapselarge language modelsAI safety
Unown placeholder for Sharp description of local minima in the loss landscape of high-dimensional two-layer ReLU neural networks

Apr 2026 / ICML 2026

Sharp description of local minima in the loss landscape of high-dimensional two-layer ReLU neural networks

Jie Huang, Bruno Loureiro, Stefano Sarao Mannelli

We study the population loss landscape of two-layer ReLU networks in a realisable teacher-student setting with Gaussian covariates. The work shows that local minima admit an exact low-dimensional representation through summary statistics, giving a sharp and interpretable description of...

optimisationlandscapeneural networks

Archive

Earlier work, listed from newest to oldest.

Unown placeholder for Thinking of Neural Networks Like a Physicist: The Statistical Physics of Machine Learning

Apr 2026 / Proceedings of the Analytical Connectionism Schools 2023--2024, PMLR 320:15-41, 2026

Thinking of Neural Networks Like a Physicist: The Statistical Physics of Machine Learning

Kai Jappe Sandbrink, Stefano Sarao Mannelli, Florent Krzakala

This pedagogical paper introduces statistical-physics approaches to machine learning, based on material presented at Analytical Connectionism 2023. It reviews how tools such as the replica method and approximate message passing illuminate unsupervised learning problems, then turns to supervised learning...

statistical physicsmachine learningreview
Figure from Curriculum learning in humans and neural networks

Mar 2025 / Proceedings of the Annual Meeting of the Cognitive Science Society 47 (CogSci 2025)

Curriculum learning in humans and neural networks

Younes Strittmatter*, Stefano Sarao Mannelli*, Miguel Ruiz-Garcia, Sebastian Musslick, Markus Wolfgang Hermann Spitzer

The sequencing of training trials can significantly influence learning outcomes in humans and neural networks. However, studies comparing the effects of training curricula between the two have typically focused on the acquisition of multiple tasks. Here, we investigate curriculum...

curriculum learning
Figure from A Theory of Initialisation's Impact on Specialisation

Mar 2025 / ICLR 2025; J. Stat. Mech. 2025, 114001

A Theory of Initialisation's Impact on Specialisation

Devon Jarvis, Sebastian Lee, Clémentine Carla Juliette Dominé, Andrew M Saxe, Stefano Sarao Mannelli

Prior work has demonstrated a consistent tendency in neural networks engaged in continual learning tasks, wherein intermediate task similarity results in the highest levels of catastrophic interference. This phenomenon is attributed to the network's tendency to reuse learned features...

continual learning
Figure from Optimal Protocols for Continual Learning via Statistical Physics and Control Theory

Sept 2024 / ICLR 2025; J. Stat. Mech. 2025, 084004

Optimal Protocols for Continual Learning via Statistical Physics and Control Theory

Francesco Mori, Stefano Sarao Mannelli, Francesca Mignacco

Artificial neural networks often struggle with catastrophic forgetting when learning multiple tasks sequentially, as training on new tasks degrades the performance on previously learned ones. Recent theoretical work has addressed this issue by analysing learning curves in synthetic frameworks...

optimal controlcontinual learning
Figure from A meta-learning framework for rationalizing cognitive fatigue in neural systems

Jun 2024 / CogSci 2024 (Oral)

A meta-learning framework for rationalizing cognitive fatigue in neural systems

Yujun Li, Rodrigo Carrasco-Davis, Younes Strittmatter, Stefano Sarao Mannelli, Sebastian Musslick

The ability to exert cognitive control is central to human brain function, facilitating goal-directed task performance. However, humans exhibit limitations in the duration over which they can exert cognitive control -a phenomenon referred to as cognitive fatigue. This study...

cognitive controlfatiguecontinual learning
Figure from Bias in Motion: Theoretical Insights into the Dynamics of Bias in SGD Training

May 2024 / NeurIPS 2024

Bias in Motion: Theoretical Insights into the Dynamics of Bias in SGD Training

Anchit Jain, Rozhin Nobahari, Aristide Baratin, Stefano Sarao Mannelli

Machine learning systems often acquire biases by leveraging undesired features in the data, impacting accuracy variably across different sub-populations. Current understanding of bias formation mostly focuses on the initial and final stages of learning, leaving a gap in knowledge...

fairnessspurious correlationsdata imbalance
Figure from Tilting the Odds at the Lottery: the Interplay of Overparameterisation and Curricula in Neural Networks

May 2024 / ICML 2024

Tilting the Odds at the Lottery: the Interplay of Overparameterisation and Curricula in Neural Networks

Stefano Sarao Mannelli, Yaraslau Ivashinka, Andrew Saxe, Luca Saglietti

A wide range of empirical and theoretical works have shown that overparameterisation can amplify the performance of neural networks. According to the lottery ticket hypothesis, overparameterised networks have an increased chance of containing a sub-network that is well-initialised to...

curriculum learninglottery ticket hypothesis
Figure from RL Perceptron: Generalization Dynamics of Policy Learning in High Dimensions

Jun 2023 / Phys. Rev. X 15, 021051 (2025)

RL Perceptron: Generalization Dynamics of Policy Learning in High Dimensions

Nishil Patel, Sebastian Lee, Stefano Sarao Mannelli, Sebastian Goldt, Andrew Saxe

Reinforcement learning (RL) algorithms have proven transformative in a range of domains. To tackle real-world domains, these systems often use neural networks to learn policies directly from pixels or other high-dimensional sensory input. By contrast, much theory of RL...

reinforcement learningcurriculum learning
Unown placeholder for Probing transfer learning with a model of synthetic correlated datasets

Jan 2022 / Machine Learning: Science and Technology

Probing transfer learning with a model of synthetic correlated datasets

Federica Gerace, Luca Saglietti, Stefano Sarao Mannelli, Andrew Saxe, Lenka Zdeborová

Transfer learning can significantly improve the sample efficiency of neural networks, by exploiting the relatedness between a data-scarce target task and a data-abundant source task. Despite years of successful applications, transfer learning practice often relies on ad-hoc solutions, while...

transfer learning
Unown placeholder for Epidemic mitigation by statistical inference from contact tracing data

Aug 2021 / Proceedings of the National Academy of Sciences

Epidemic mitigation by statistical inference from contact tracing data

Antoine Baker, Indaco Biazzo, Alfredo Braunstein, Giovanni Catania, Luca Dall’Asta, Alessandro Ingrosso, Florent Krzakala, Fabio Mazza, Marc Mezard, Anna Paola Muntoni, Maria Refinetti, Stefano Sarao Mannelli, Lenka Zdeborova

Contact tracing is an essential tool to mitigate the impact of a pandemic, such as the COVID-19 pandemic. In order to achieve efficient and scalable contact tracing in real time, digital devices can play an important role. While a...

epidemic mitigation
Unown placeholder for Complex Dynamics in Simple Neural Networks: Understanding Gradient Flow in Phase Retrieval

Dec 2020 / NeurIPS 2020

Complex Dynamics in Simple Neural Networks: Understanding Gradient Flow in Phase Retrieval

Stefano Sarao Mannelli, Giulio Biroli, Chiara Cammarota, Florent Krzakala, Pierfrancesco Urbani, Lenka Zdeborova

Despite the widespread use of gradient-based algorithms for optimising high-dimensional non-convex functions, understanding their ability of finding good minima instead of being trapped in spurious ones remains to a large extent an open problem. Here we focus on gradient...

optimisationlandscape
Figure from Thresholds of descending algorithms in inference problems

Mar 2020 / Journal of Statistical Mechanics: Theory and Experiment

Thresholds of descending algorithms in inference problems

Stefano Sarao Mannelli, Lenka Zdeborova

We review recent works (Sarao Mannelli et al 2018 arXiv 1812.09066, 2019 Int. Conf. on Machine Learning 4333–42, 2019 Adv. Neural Information Processing Systems 8676–86) on analyzing the dynamics of gradient-based algorithms in a prototypical statistical inference problem. Using...

optimisationlandscapereview
Unown placeholder for Marvels and pitfalls of the Langevin algorithm in noisy high-dimensional inference

Mar 2020 / Physical Review X

Marvels and pitfalls of the Langevin algorithm in noisy high-dimensional inference

Stefano Sarao Mannelli, Giulio Biroli, Chiara Cammarota, Florent Krzakala, Pierfrancesco Urbani, Lenka Zdeborova

Gradient-descent-based algorithms and their stochastic versions have widespread applications in machine learning and statistical inference. In this work, we carry out an analytic study of the performance of the algorithm most commonly considered in physics, the Langevin algorithm, in...

optimisationlandscape
Unown placeholder for Who is Afraid of Big Bad Minima? Analysis of gradient-flow in spiked matrix-tensor models

Dec 2019 / NeurIPS 2019 (Spotlight)

Who is Afraid of Big Bad Minima? Analysis of gradient-flow in spiked matrix-tensor models

Stefano Sarao Mannelli, Giulio Biroli, Chiara Cammarota, Florent Krzakala, Lenka Zdeborova

Gradient-based algorithms are effective for many machine learning tasks, but despite ample recent effort and some progress, it often remains unclear why they work in practice in optimising high-dimensional non-convex functions and why they find good minima instead of...

optimisationlandscape

No publications match this filter.