arXiv:2410.00645cs.LG2024-10ICLR被引 9

用低秩随机特征与预训练模型实现理论可靠且性能强的持续学习

LoRanPAC: Low-rank Random Features and Pre-trained Models for Bridging Theory and Practice in Continual Learning

  • 将预训练特征升维后解最小范数最小二乘问题,通过截断SVD稳定训练
  • 可处理数百个任务,在多个数据集上超越当前最优方法
  • 兼具理论保证与实证性能,适合追求稳健性的持续学习研究者

持续学习的目标是训练一个能顺序处理多个任务的模型。近期方法利用泛化能力强的大规模预训练模型取得了优异性能,但缺乏理论保障,易出现意外失败;而原理严谨的方法往往表现不佳。本文提出LoRanPAC,通过将预训练特征映射到高维空间,构建过参数化的最小范数最小二乘问题。发现升维后的特征严重病态,导致训练误差大(数值不稳定)和泛化误差增加。为此,我们持续对升维特征进行奇异值分解截断。该方法对超参数选择不敏感,可处理数百个任务,在多个数据集上优于当前最优方法。更重要的是,其持续学习过程满足递推关系,通过适当截断部分SVD分量,可证明其保持小的训练与测试误差。这使得方法兼具稳定性、强性能与理论保障。代码已开源:https://github.com/liangzu/loranpac。

原文摘要 · Abstract (English)

The goal of continual learning (CL) is to train a model that can solve multiple tasks presented sequentially. Recent CL approaches have achieved strong performance by leveraging large pre-trained models that generalize well to downstream tasks. However, such methods lack theoretical guarantees, making them prone to unexpected failures. Conversely, principled CL approaches often fail to achieve competitive performance. In this work, we aim to bridge this gap between theory and practice by designing a simple CL method that is theoretically sound and highly performant. Specifically, we lift pre-trained features into a higher dimensional space and formulate an over-parametrized minimum-norm least-squares problem. We find that the lifted features are highly ill-conditioned, potentially leading to large training errors (numerical instability) and increased generalization errors. We address these challenges by continually truncating the singular value decomposition of the lifted features. Our approach, termed LoRanPAC, is stable with respect to the choice of hyperparameters, can handle hundreds of tasks, and outperforms state-of-the-art CL methods on multiple datasets. Importantly, our method satisfies a recurrence relation throughout its continual learning process, which allows us to prove it maintains small training and test errors by appropriately truncating a fraction of SVD factors. This results in a stable continual learning method with strong empirical performance and theoretical guarantees. Code available: https://github.com/liangzu/loranpac.

持续学习理论保证SVD截断预训练模型

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