无限学习率调度比余弦退火更适配持续预训练,减少遗忘。
Beyond Cosine Decay: On the effectiveness of Infinite Learning Rate Schedule for Continual Pre-training
- 用无限学习率替代余弦退火,缓解重热阶段遗忘问题。
- 在图像和语言任务中均超越余弦退火,小规模到大规模都有效。
- 适合需要长期学习新数据且不丢旧知识的研究者。
不断增长的无标签数据为人工智能训练带来机遇与挑战。自监督学习(SSL)虽能从海量无标签数据中提取有意义表征,但现有方法在面对真实世界数据流的非平稳、非独立同分布特性时,仍难以避免遗忘先前知识。近期研究采用重复余弦退火调度进行大规模持续预训练,但该方法(1)在重热阶段固有地导致遗忘,(2)尚未与现有持续SSL方法进行系统对比。本文系统比较了广泛使用的余弦调度与近期提出的无限学习率调度,实证发现后者是更优替代方案。我们在多种图像与语言数据集上进行广泛实验,结果表明,无限学习率调度在持续预训练中表现稳定优于重复余弦衰减,且不受固定迭代预算限制。例如,在小型MAE预训练设置中,其性能超越多个文献中的强基线。进一步扩展至更大规模的MAE与自回归语言模型预训练,结果仍显示该调度在两种场景下均有效,显著优于重复余弦衰减,并在零样本语言模型基准上取得更好效果。
原文摘要 · Abstract (English)
The ever-growing availability of unlabeled data presents both opportunities and challenges for training artificial intelligence systems. While self-supervised learning (SSL) has emerged as a powerful paradigm for extracting meaningful representations from vast amounts of unlabeled data, existing methods still struggle to adapt to the non-stationary, non-IID nature of real-world data streams without forgetting previously learned knowledge. Recent works have adopted a repeated cosine annealing schedule for large-scale continual pre-training; however, these schedules (1) inherently cause forgetting during the re-warming phase and (2) have not been systematically compared to existing continual SSL methods. In this work, we systematically compare the widely used cosine schedule with the recently proposed infinite learning rate schedule and empirically find the latter to be a more effective alternative. Our extensive empirical evaluation across diverse image and language datasets demonstrates that the infinite learning rate schedule consistently enhances continual pre-training performance compared to a repeated cosine decay without being restricted to a fixed iteration budget. For instance, in a small-scale MAE pre-training setup, it outperforms several strong baselines from the literature. We then scale up our experiments to larger MAE pre-training and autoregressive language model pre-training. Our results show that the infinite learning rate schedule remains effective at scale, surpassing repeated cosine decay for both MAE pre-training and zero-shot LM benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。