Eshaan Nichani: Learning Compositional Functions with Transformers from Easy-to-Hard Data

Eshaan Nichani: Learning Compositional Functions with Transformers from Easy-to-Hard Data

🎙 Eshaan Nichani 👥 3K 📅 October 3, 2025 ⏱ 46 min 👁 168 📄 original study 🧭 2026-08-16
Available in: English (current) Français

Keywords

compositional reasoningtransformerscurriculum learningexpressivitySQ lower bound

Summary

The talk, presented by Eshaan Nichani, introduces a formal task called k-fold composition to model multi-step reasoning in transformers. The task involves composing contextual permutations (provided in the prompt) with hidden parametric permutations (learned during pre-training). The speaker proves that a transformer with log k + 1 layers can express this task efficiently, using a binary tree construction where each layer doubles the number of composed permutations. However, learning the task via gradient descent is shown to be computationally hard: a statistical query (SQ) lower bound demonstrates that any SQ learner requires either exponentially many queries or exponentially small tolerance. This implies that standard training methods like SGD would need exponential samples or steps. The talk concludes that curriculum learning, where training data is presented in increasing difficulty, is both necessary and sufficient to learn the k-fold composition efficiently. The presentation includes detailed constructions and proof sketches, emphasizing the role of residual streams and attention mechanisms.

156 words

Critical Evaluation

Value of the Information & Strength of the Argument

The talk provides a clear and rigorous theoretical framework for understanding compositional reasoning in transformers. The value lies in formalizing a complex cognitive task into a tractable mathematical problem, enabling precise analysis. The argumentation is solid: the expressivity construction is detailed and intuitive, and the SQ lower bound is well-motivated with a clear proof sketch. The speaker effectively connects the theoretical results to practical implications, such as the necessity of curriculum learning. The presentation is logically structured, building from definitions to results, and addresses potential questions from the audience, strengthening the overall argument.

Scientific Rigor, Source Quality, Title Accuracy

The talk is based on a specific research paper, which is cited in the description and linked to arXiv. The speaker does not reference external sources beyond the paper, but the content is presented with mathematical rigor. The title accurately reflects the content, focusing on learning compositional functions with transformers and the role of easy-to-hard data. The presentation is self-contained, with definitions and proofs sketched clearly. The audience questions are addressed, indicating a thorough understanding. Overall, the scientific rigor is high, and the sources are appropriate for a research seminar.

198 words

Title / Content Match

The title accurately reflects the content, which focuses on learning compositional functions with transformers and the role of easy-to-hard data (curriculum learning).

Quality & Reliability

8/10

The talk presents a rigorous theoretical analysis with formal proofs for expressivity and learning, backed by a peer-reviewed paper on arXiv. The claims are precise and well-motivated, though the presentation is concise and assumes familiarity with the topic.

Key Moments

Cited Sources

Concurring Sources

Contribution & Novelties

The talk introduces a novel formal task (k-fold composition) to model compositional reasoning in transformers, bridging contextual and parametric knowledge. It provides a constructive proof that transformers can express this task with logarithmic depth, and establishes a statistical query lower bound showing that learning without curriculum is computationally hard. The key insight is that curriculum learning (easy-to-hard data) is both necessary and sufficient for efficient gradient-based training, offering a theoretical justification for a common practice in training LLMs.

Pour aller plus loin :

  • Statistical query learning — The SQ model is central to the lower bound argument.
  • Transformer architecture — The original transformer paper, foundational to the model used.
  • Curriculum learning — The concept of training on easy examples first, shown to be necessary here.

125 words

Radar Profile

The radar profile shows high scores in information quality, technical level, and reliability, with slightly lower scores in information quantity and overall reliability, reflecting the focused and rigorous nature of the talk. The presentation is dense and technical, appealing to a specialized audience.

Reliability 8/10

💬 No comments were provided for analysis.