The Power of Power Law: Asymmetry Enables Compositional Reasoning

arXiv:2604.22951v1 Announce Type: new Abstract: Natural language data follows a power-law distribution, with most knowledge and skills appearing at very low frequency. While a common intuition suggests that reweighting or curating data towards a uniform distribution may help models better learn these long-tail skills, we find a counterintuitive result: across a wide range of compositional reasoning tasks, such as state tracking and multi-step arithmetic, training under power-law distributions consistently outperforms training under uniform distributions. To understand this advantage, we introduce a minimalist skill-composition task and show that learning under a power-law distribution provably requires significantly less training data. Our theoretical analysis reveals that power law sampling induces a beneficial asymmetry that improves the pathological loss landscape, which enables models to first acquire high-frequency skill compositions with low data complexity, which in turn serves as a stepping stone to efficiently learn rare long-tailed skills. Our results offer an alternative perspective on what constitutes an effective data distribution for training models.

Related

• Generating Verifiable Chain of Thoughts from Exection-Traces
• SkillFactory: Self-Distillation For Learning Cognitive Behaviors
• A Mechanistic Analysis of Looped Reasoning Language Models
• Thinking with Reasoning Skills: Fewer Tokens, More Accuracy
• Learning from Contrasts: Synthesizing Reasoning Paths from Diverse Search Trajectories

Source: arXiv cs.AI | 2026-04-28