Learn from your own latents and not from tokens: A sample-complexity theory

ML NNGTB SSLRO
从扩散模型到大语言模型,生成式模型虽取得了卓越的性能表现,但其训练所需的数据量却比生物学习者所需高出数个数量级。与此相对,一种新兴的替代范式正逐渐兴起:网络被训练用于预测自身对相关视图或被遮蔽区域所形成的**潜在表征**,例如 data2vec 和 JEPA 所采用的方法——这一思想与皮层中基于预测编码(predictive coding)的神经计算理论密切相关。尽管这类方法在实证上展现出强劲效果,但对其背后理论机制的理解仍十分有限。其中两个核心问题尤为关键:(1)潜在表征预测究竟能在多大程度上提升数据利用效率?(2)将此类方法堆叠成多尺度层级结构是否真有增益?我们借助一种可解析的概率上下文无关文法(probabilistic context-free grammar, PCFG)作为理论分析的数据源,该文法能精确刻画自然语言与图像所共有的组合性结构。此类文法通过在深度为 $L$ 的隐变量语法树上递归应用产生式规则,生成由可见符号(token)构成的字符串。针对此类数据,监督学习或基于词元(token-level)的自监督学习均需样本量随 $L$ 呈**指数级增长**,方能准确恢复底层隐式语法树;而我们严格证明:潜在表征预测仅需样本量为**关于 $L$ 的常数级**(至多含对数因子)。我们通过三方面验证了该理论界:(i)一种分层聚类算法;(ii)一个端到端神经网络,其内部由若干“预测—聚类”模块构成,各模块在每一层级均通过梯度下降自主预测自身的潜在表征;(iii)首次对 data2vec 进行样本复杂度分析,结果表明其本质上已隐式实现了分层潜在预测。由此推断,显式地堆叠此类结构(如 H-JEPA 所倡导的方式)在很大程度上是冗余的。
Generative models, from diffusion models to large language models, achieve remarkable performance but at a cost in training data orders of magnitude larger than what biological learners require. An alternative paradigm has emerged in which networks are trained to predict their \emph{own} latent representations of related views or masked regions, as in data2vec and JEPA -- an idea related to predictive-coding accounts of the cortex. Despite strong empirical results, the theoretical understanding of these methods remains limited. Central questions include: by how much does latent prediction actually improve data efficiency? Is there a benefit to stacking such methods into multi-scale hierarchies? We answer both using as data a tractable probabilistic context-free grammar that captures the compositional structure of natural language and images. Such a grammar generates strings of visible tokens by recursively applying production rules along a tree of hidden symbols of depth $L$. For such data, supervised or token-level SSL require a number of samples \emph{exponential} in $L$ to recover the latent tree; we prove that latent prediction achieves this with a number of samples \emph{constant} in $L$, up to logarithmic factors. We confirm this bound with (i) a hierarchical clustering algorithm, (ii) an end-to-end neural network whose predictor-clusterer modules predict their own latents at each level via gradient descent, and (iii) the first sample-complexity analysis of data2vec, which we show implicitly performs hierarchical latent prediction. This suggests that explicit stacking such as H-JEPA is largely redundant.
许愿