An Analytical Theory of Curriculum Learning in Teacher-Student Networks
Luca Saglietti, Stefano Sarao Mannelli, Andrew M. Saxe
Abstract
In animals and humans, curriculum learning—presenting data in a curated order—is critical to rapid learning and effective pedagogy. A long history of experiments has demonstrated the impact of curricula in a variety of animals but, despite its ubiquitous presence, a theoretical understanding of the phenomenon is still lacking. Surprisingly, in contrast to animal learning, curricula strategies are not widely used in machine learning and recent simulation studies reach the conclusion that curricula are moderately effective or even ineffective in most cases. This stark difference in the importance of curriculum raises a fundamental theoretical question: when and why does curriculum learning help? In this work, we analyse a prototypical neural network model of curriculum learning in the high-dimensional limit, employing statistical physics methods. We study a task in which a sparse set of informative features are embedded amidst a large set of noisy features. We analytically derive average learning trajectories for simple neural networks on this task, which establish a clear speed benefit for curriculum learning in the online setting. However, when training experiences can be stored and replayed (for instance, during sleep), the advantage of curriculum in standard neural networks disappears, in line with observations from the deep learning literature. Inspired by synaptic consolidation techniques developed to combat catastrophic forgetting, we propose curriculum-aware algorithms that consolidate synapses at curriculum change points and investigate whether this can boost the benefits of curricula. We derive generalisation performance as a function of consolidation strength (implemented as an L 2 regularisation/elastic coupling connecting learning phases), and show that curriculum-aware algorithms can yield a large improvement in test performance. Our reduced analytical descriptions help reconcile apparently conflicting empirical results, trace regimes where curriculum learning yields the largest gains, and provide experimentally-accessible predictions for the impact of task parameters on curriculum benefits. More broadly, our results suggest that fully exploiting a curriculum may require explicit adjustments in the loss.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e9710a62-2342-40ac-9c9a-86c18a08d64fCited by top-tier papers11
- Compositional generalization through abstract representations in human and artificial neural networksTakuya Ito, Tim Klinger, Douglas Schultz, John Murray et al.NeurIPS 2022 · 65 citations
- Provable Advantage of Curriculum Learning on Parity Targets with Mixed InputsEmmanuel Abbe, Elisabetta Cornacchia, Aryo LotfiNeurIPS 2023 · 29 citations
- Curriculum Learning With Infant Egocentric VideosSaber Sheybani, Himanshu Hansaria, Justin Wood, Linda B. Smith et al.NeurIPS 2023 · 26 citations
- Maslow's Hammer in Catastrophic Forgetting: Node Re-Use vs. Node ActivationSebastian Lee, Stefano Sarao Mannelli, Claudia Clopath, Sebastian Goldt et al.ICML 2022 · 18 citations
- Why Do Animals Need Shaping? A Theory of Task Composition and Curriculum LearningJin Hwa Lee, Stefano Sarao Mannelli, Andrew M. SaxeICML 2024 · 15 citations
Builds on5
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- When Do Curricula Work?Xiaoxia Wu, Ethan Dyer, Behnam NeyshaburICLR 2021 · 141 citations
- Complex Dynamics in Simple Neural Networks: Understanding Gradient Flow in Phase RetrievalStefano Sarao Mannelli, Giulio Biroli, Chiara Cammarota, Florent Krzakala et al.NeurIPS 2020 · 32 citations
- Curriculum learning as a tool to uncover learning principles in the brainDaniel R. Kepple, Rainer Engelken, Kanaka RajanICLR 2022 · 21 citations
- Tilting the playing field: Dynamical loss functions for machine learningMiguel Ruiz-Garcia, Ge Zhang, Samuel S. Schoenholz, Andrea J. LiuICML 2021 · 12 citations
Related papers
- Biologically inspired sleep algorithm for increased generalization and adversarial robustness in deep neural networksTimothy Tadros, Giri P. Krishnan, Ramyaa Ramyaa, Maxim BazhenovICLR 2020 · 28 citations
- Overcoming Catastrophic Forgetting by Neuron-Level Plasticity ControlInyoung Paik, Sangjun Oh, Taeyeong Kwak, Injung KimAAAI 2020 · 60 citations
- Continual learning in recurrent neural networksBenjamin Ehret, Christian Henning, Maria R. Cervera, Alexander Meulemans et al.ICLR 2021 · 4,433 citations
- Sliced Cramer Synaptic Consolidation for Preserving Deeply Learned RepresentationsSoheil Kolouri, Nicholas A. Ketz, Andrea Soltoggio, Praveen K. PillyICLR 2020 · 42 citations
- Layerwise Optimization by Gradient Decomposition for Continual LearningShixiang Tang, Dapeng Chen, Jinguo Zhu, Shijie Yu et al.CVPR 2021
