Training | | arXiv:2605.15053 |★ 深度解读

TFGN: Task-Free, Replay-Free Continual Pre-Training Without Catastrophic Forgetting at LLM Scale

TFGN:无需任务标签与回放的LLM级持续预训练架构

🧭 Training

Independent Researcher

Anurup Ganguli

提出一种内部架构叠加层TFGN,在无回放、无任务ID的条件下,实现LLaMA 3.1 8B等多尺度的近零灾难性遗忘与>99.59%的梯度正交性

动机 灾难性遗忘是LLM在生产环境中持续学习的核心障碍。现有方法依赖数据回放、任务标识符或Fisher惩罚项,三者均无法满足隐私合规、任务标签未知、大规模无惩罚的部署需求。长上下文窗口不能替代持久权重内化,前沿模型在续训时仍面临显著的能力退化。

方法关键点

  • 架构叠加层:TFGN插入Transformer的逐块计算中,生成输入条件化的参数高效更新,不改变其余网络,保持前向传播全密集(无稀疏门控)。
  • 读写解耦:保护来自写入路径——不同域的梯度因结构设计自动落入近乎正交的子空间,无需正交性损失项;读取路径则保持共享,支持跨域正向迁移。
  • 无任务依赖:无需回放缓冲、任务ID、外部调度器或正则化惩罚,保护由架构本身保障。
  • 双模式:支持从头训练(FS)与预训练骨干改造(Retrofit),持续阶段的训练参数仅为全量的~1-14%。

关键实验

  • 设置:6个异构文本域(普通英语、Python、数学、生物医学、中文、JavaScript),每阶段10亿tokens,总分三步尺度(~398M、~739M、~9B)和FS/RF两种模式。
  • 基线:标准微调(Std-FT)和LoRA r=256。
  • 核心结果:
    • BWT(反向迁移):LLaMA 3.1 8B改造模式下低至**-0.007**,~739M从零训练模式比Std-FT改善约14倍。
    • 梯度正交性:所有条件下跨域梯度的L2正交比例**≥99.59%**,平均余弦相似度<0.10。
    • 稳定性:HellaSwag精度跨阶段波动<0.01;定性展示中基线完全生成域外字符,TFGN保持域一致文本。
    • 正向迁移:JavaScript未训练时,Python训练后其PPL下降26.8%(8B改造)和62.0%(739M从零),证实读取路径的跨域协同。

核心观点 稳定性是写入问题,不是读取问题;TFGN通过构造性梯度解耦,使LLM在不牺牲可塑性的前提下关闭灾难性遗忘。