效率为免费:理想数据是可传输的表示
机器学习
2024-11-04 v2 人工智能
摘要
数据是现代机器学习中的根本机遇和挑战,目前限制了表示学习的可扩展性,并阻碍了模型演进的进程。本文从优化和泛化两个角度探讨数据的效率属性。我们的理论与实证分析揭示了一个意想不到的发现:对于给定任务,利用面向任务且与体系结构无关的公开可用模型(本文称为"先验模型")可有效产生高效数据。基于这一洞见,我们提出了Representation Learning Accelerator(ReLA),促进高效数据的形成与利用,从而加速表示学习。利用在CIFAR-10上预训练的ResNet-18作为先验模型来指导在ImageNet-1K上训练ResNet-50,可将计算成本降低50%,同时保持与使用原始BYOL(需100%成本)相同准确率。我们的代码可在https://github.com/LINs-lab/ReLA获取。
引用
@article{arxiv.2405.14669,
title = {Efficiency for Free: Ideal Data Are Transportable Representations},
author = {Peng Sun and Yi Jiang and Tao Lin},
journal= {arXiv preprint arXiv:2405.14669},
year = {2024}
}
备注
Code: https://github.com/LINs-lab/ReLA