Search papers, labs, and topics across Lattice.
This paper introduces a novel meta-representation learning framework called complexity minimization, aimed at theoretically analyzing the scaling behavior of pre-training in machine learning. By evaluating and minimizing the worst-case downstream model complexity across various domains, the framework provides a rigorous explanation for the observed reduction in sample complexity as pre-training data increases. Empirical results confirm that integrating complexity regularization into existing meta-learning approaches enhances downstream sample efficiency, demonstrating the practical implications of the theoretical insights.
Complexity minimization reveals that as pre-training data scales, few-shot adaptation error rates significantly decrease, challenging existing theories on sample complexity.
Pre-training has become a fundamental paradigm in modern machine learning, with one of its key empirical benefits being reduced downstream sample complexity as the scale of pre-training data increases. However, existing theoretical frameworks for pre-training do not fully explain this phenomenon. In this paper, we introduce complexity minimization, a novel meta-representation learning framework designed to enable theoretical analysis of this scaling behavior, which learns representations by evaluating the downstream model complexity best suited to each domain and minimizing the worst-case such complexity across source domains. Our end-to-end theoretical analysis, spanning pre-training through downstream regression, shows that this framework provably captures this scaling behavior; in particular, we show that the error rate of few-shot adaptation improves as the amount of meta-training data grows. Empirically, we demonstrate that incorporating complexity regularization into existing meta-learning methods consistently improves downstream sample efficiency.