mlsanityData to 7 Oct 2026

Positive Distribution Shift as a Framework for Understanding Tractable Learning

arXiv 2602.08907 ↗First posted 9 Feb 2026cs.LG stat.MLcited by 3

Not on the map. Why?

Abstract (arXiv)

We study a setting where the goal is to learn a target function f(x) with respect to a target distribution D(x), but training is done on i.i.d. samples from a different training distribution D'(x), labeled by the true target f(x). Such a distribution shift (here in the form of covariate shift) is usually viewed negatively, as hurting or making learning harder, and the traditional distribution shift literature is mostly concerned with limiting or avoiding this negative effect. In contrast, we argue that with a well-chosen D'(x), the shift can be positive and make learning easier -- a perspective called Positive Distribution Shift (PDS). Such a perspective is central to contemporary machine learning, where much of the innovation is in finding good training distributions D'(x), rather than changing the training algorithm. We further argue that the benefit is often computational rather than statistical, and that PDS allows computationally hard problems to become tractable even using standard gradient-based training. We formalize different variants of PDS, show how certain hard classes are easily learnable under PDS, and make connections with membership query learning.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 53 of its 94 references that are in the map. First posted on arXiv 9 Feb 2026.

Builds on

53 of 94 references.

References · 53

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
  3. 2022Exploring Length Generalization in Large Language Modelscited by 204
  4. 2023What Algorithms can Transformers Learn? A Study in Length Generalizationcited by 198
  5. 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 184
  6. 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
  7. 2023SGD learning on neural networks: leap complexity and saddle-to-saddle dynamicscited by 142
  8. 2022The merged-staircase property: a necessary and nearly sufficient condition for SGD learning of sparse functions on two-layer neural networkscited by 141
  9. 2020Online stochastic gradient descent on non-convex losses from high-dimensional inferencecited by 119
  10. 2022Learning Single-Index Models with Shallow Neural Networkscited by 92
  11. 2020Learning Parities with Neural Networkscited by 77
  12. 2023Generalization on the Unseen, Logic Reasoning and Degree Curriculumcited by 73
38 more
  1. 2023How Deep Neural Networks Learn Compositional Data: The Random Hierarchy Modelcited by 70
  2. 2023How Two-Layer Neural Networks Learn, One (Giant) Step at a Timecited by 67
  3. 2024The Benefits of Reusing Batches for Gradient Descent in Two-Layer Networks: Breaking the Curse of Information and Leap Exponentscited by 50
  4. 2024Neural network learns low-dimensional polynomials with SGD near the information-theoretic limitcited by 45
  5. 2021Quantifying the Benefit of Using Differentiable Learning over Tangent Kernelscited by 41
  6. 2024Towards a theory of how the structure of language is acquired by deep neural networkscited by 38
  7. 2016Reliably Learning the ReLU in Polynomial Timecited by 36
  8. 2023SGD Finds then Tunes Features in Two-Layer Neural Networks with near-Optimal Sample Complexity: A Case Study in the XOR problemcited by 36
  9. 2024Repetita Iuvant: Data Repetition Allows SGD to Learn High-Dimensional Multi-Index Functionscited by 35
  10. 2025Learning Compositional Functions with Transformers from Easy-to-Hard Datacited by 34
  11. 2023Gradient-Based Feature Learning under Structured Datacited by 32
  12. 2021An Analytical Theory of Curriculum Learning in Teacher-Student Networkscited by 27
  13. 2021On the Power of Differentiable Learning versus PAC and SQ Learningcited by 27
  14. 2022Hardness of Noise-Free Learning for Two-Hidden-Layer Neural Networkscited by 27
  15. 2021From Local Pseudorandom Generators to Hardness of Learningcited by 24
  16. 2021Probing transfer learning with a model of synthetic correlated datasetscited by 24
  17. 2016Deep Learning and Hierarchal Generative Modelscited by 23
  18. 2020Poly-time universality and limitations of deep learningcited by 23
  19. 2023Provable Advantage of Curriculum Learning on Parity Targets with Mixed Inputscited by 22
  20. 2014Complexity theoretic limitations on learning DNF'scited by 21
  21. 2023A Mathematical Model for Curriculum Learning for Paritiescited by 19
  22. 2022Learning to Reason with Neural Networks: Generalization, Unseen Data and Boolean Measurescited by 17
  23. 2022On the non-universality of deep learning: quantifying the cost of symmetrycited by 17
  24. 2020Hardness of Learning Neural Networks with Natural Weightscited by 15
  25. 2019ID3 Learns Juntas for Smoothed Product Distributionscited by 13
  26. 2024On the Complexity of Learning Sparse Functions with Statistical and Gradient Queriescited by 13
  27. 2023Computational Complexity of Learning Neural Networks: Smoothness and Degeneracycited by 10
  28. 2024Tilting the Odds at the Lottery: the Interplay of Overparameterisation and Curricula in Neural Networkscited by 9
  29. 2024Learning Gaussian Multi-Index Models with Gradient Flow: Time Complexity and Directional Convergencecited by 8
  30. 2025Low-dimensional Functions are Efficiently Learnable under Randomly Biased Distributionscited by 8
  31. 2024Learning High-Degree Parities: The Crucial Role of the Initializationcited by 7
  32. 2025Learning single-index models via harmonic decompositioncited by 7
  33. 2017Exact Learning from an Honest Teacher That Answers Membership Queriescited by 6
  34. 2025The Generative Leap: Sharp Sample Complexity for Efficiently Learning Gaussian Multi-Index Modelscited by 6
  35. 2013Moment-Matching Polynomialscited by 5
  36. 2017Exact Learning of Juntas from Membership Queriescited by 5
  37. 2025Shift is Good: Mismatched Data Mixing Improves Test Performancecited by 5
  38. 2019Adaptive Exact Learning of Decision Trees from Membership Queriescited by 4

50 most cited of 53 All 53 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.