arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.12481stat.MLcs.LG

通过细粒度偏差-方差分析理解最优特征迁移

Understanding Optimal Feature Transfer via a Fine-Grained Bias-Variance Analysis

Harvard University(哈佛大学);Google DeepMind(谷歌DeepMind)

Yufan Li, Subhabrata Sen, Ben Adlam

AI总结:

本文通过细粒度偏差-方差分析研究迁移学习中最优特征表示,发现最优特征化自然稀疏,并揭示了从硬选择到软选择的相变现象。

摘要:

在迁移学习范式中,模型在数据丰富的预训练阶段学习有用的表示(或特征),然后使用预训练表示来提升模型在数据稀缺的下游任务上的性能。本文以优化下游性能为目标探索迁移学习。我们引入一个简单的线性模型,该模型以任意预训练特征变换作为输入。我们推导了下游风险的精确渐近结果及其细粒度偏差-方差分解。随后,我们识别出在一组下游任务上平均后能够优化渐近下游偏差和方差的预训练表示。我们的理论与实证分析揭示了一个令人惊讶的现象:即使在没有显式稀疏诱导先验或惩罚的情况下,最优特征化也自然具有稀疏性。此外,我们还识别出一个相变现象,即最优预训练表示从对相关特征的硬选择转变为软选择。

英文摘要:

In the transfer learning paradigm models learn useful representations (or features) during a data-rich pretraining stage, and then use the pretrained representation to improve model performance on data-scarce downstream tasks. In this work, we explore transfer learning with the goal of optimizing downstream performance. We introduce a simple linear model that takes as input an arbitrary pretrained feature transform. We derive exact asymptotics of the downstream risk and its \textit{fine-grained} bias-variance decomposition. We then identify the pretrained representation that optimizes the asymptotic downstream bias and variance averaged over an ensemble of downstream tasks. Our theoretical and empirical analysis uncovers the surprising phenomenon that the optimal featurization is naturally sparse, even in the absence of explicit sparsity-inducing priors or penalties. Additionally, we identify a phase transition where the optimal pretrained representation shifts from hard selection to soft selection of relevant features.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑