arXiv:2411.12029stat.MLcs.LG2024-11NeurIPS被引 1

证明了在特征学习中,经验风险最小化能逼近已知最优特征的模型表现。

On the Efficiency of ERM in Feature Learning

  • 用经验风险最小化联合学习特征映射与线性预测器。
  • 当特征集合不大且最优特征唯一时,其误差接近已知最优特征的理论上限。
  • 结果适用于稀疏回归中的最优子集选择,提供新性能保证。

针对由特征映射集合 $\/mathcal{T}$ 索引的一组特征图,研究在平方损失下对这些特征映射诱导的线性类的并集进行经验风险最小化(ERM)在回归问题中的表现。该设置旨在捕捉特征学习的最简情形:模型需从数据中联合学习合适的特征映射与线性预测器。我们首先分析经验风险最小化序列的过失风险渐近分位数。惊人发现是,当集合 $\/mathcal{T}$ 不过大且存在唯一最优特征映射时,这些分位数与已知最优特征映射的预言者程序(oracle procedure)的过失风险分位数仅差一个因子二。我们进一步通过非渐近分析,量化了集合 $\/mathcal{T}$ 的全局复杂度对 ERM 过失风险的衰减效应,并将其与特征映射次优性的子水平集大小关联。作为应用,我们在一般假设下为稀疏线性回归中的最优子集选择提供了新的性能保证。

原文摘要 · Abstract (English)

Given a collection of feature maps indexed by a set $\mathcal{T}$, we study the performance of empirical risk minimization (ERM) on regression problems with square loss over the union of the linear classes induced by these feature maps. This setup aims at capturing the simplest instance of feature learning, where the model is expected to jointly learn from the data an appropriate feature map and a linear predictor. We start by studying the asymptotic quantiles of the excess risk of sequences of empirical risk minimizers. Remarkably, we show that when the set $\mathcal{T}$ is not too large and when there is a unique optimal feature map, these quantiles coincide, up to a factor of two, with those of the excess risk of the oracle procedure, which knows a priori this optimal feature map and deterministically outputs an empirical risk minimizer from the associated optimal linear class. We complement this asymptotic result with a non-asymptotic analysis that quantifies the decaying effect of the global complexity of the set $\mathcal{T}$ on the excess risk of ERM, and relates it to the size of the sublevel sets of the suboptimality of the feature maps. As an application of our results, we obtain new guarantees on the performance of the best subset selection procedure in sparse linear regression under general assumptions.

特征学习经验风险统计分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。