AI消息速览

LIFT:用教师监督预训练潜在信息反馈 Transformer

事件日期 2026-09-29 · 学术前沿 · 已接受

事件日期2026-09-29
信息日期2026-09-29
入库日期2026-10-01
通道学术前沿
状态已接受
来源arXiv 论文

知识卡片:LIFT:用教师监督预训练潜在信息反馈 Transformer

一句话结论

LIFT(Latent Information Feedback Transformer)通过把循环状态学习转化为教师强制预测问题,在预训练中让 Transformer 语言模型学会将深层信息反馈到浅层/后续生成步;在 135M–1B 参数范围、token 匹配预算下,它持续优于标准 Transformer 和基线,并在状态跟踪任务上表现出显著样本效率优势(具体指标待核实)。

事件概述或研究问题

标准 Transformer 语言模型是前馈的:深层表示不会反馈到浅层,跨生成步向下流动信息的唯一路径是已解码 token。该狭窄通道迫使模型重新计算中间结果,并丢弃替代续写。本文研究能否在预训练阶段移除这一瓶颈,使语言模型具备跨生成步传播状态的能力。为此提出 LIFT 架构与训练方法。

方法/产品要点

  • 将循环状态学习转化为 teacher-forced prediction 问题。
  • 每个输入 token 与一个信息密集状态配对;该状态来自一个现成预训练语言模型的 next-token 分布。
  • 模型增加少量额外参数,训练目标同时预测下一个 token 和下一个状态。
  • 由于输入状态可预计算,预训练在位置上保持完全并行。
  • 推理时,模型自己预测的状态被反馈到后续步骤;计算开销较小,并随模型规模增大而降低。
  • 教师模型身份、状态表示维度、额外参数规模、预训练数据与超参数等:待核实。

主要结果或产业意义

  • 在 135M 到 1B 参数的预训练模型上,LIFT 在语言建模、下游推理任务和程序性任务中,于 token 匹配预算(token-matched budget)下持续优于标准 Transformer 和基线;与计算匹配预算(compute-matched)的 Transformer 相比持平或更好。
  • 在状态跟踪任务上的受控研究中,一个小型 LIFT 优于同尺寸 Transformer 用 8 倍数据训练的结果;即使训练所用状态来自一个在该任务上失败的 Transformer,这一优势仍存在。
  • 产业意义:若可复现,LIFT 提供了一种在预训练中学习跨生成步状态传播的可扩展方案,可能减少重复计算、保留替代续写信息,并提升样本效率;实际部署成本、吞吐量与适用范围待核实。

为什么重要

  • 它直接挑战标准 Transformer 的前馈信息流限制,探索“深度到浅层反馈”能否在预训练中被学习。
  • 通过教师监督预计算状态,把通常需要序列递归的状态学习变成可并行训练的目标,兼顾表达与效率。
  • 状态跟踪任务的受控结果暗示:反馈结构可能比单纯扩大数据或模型更有效地解决某些程序性/推理任务。
  • 与既有卡片的关系:现有 MMOE、DynoDINO、LEMUR 卡片分别关注扩散 Transformer 专家扩展、医学分割动态潜在信息、多目标 RL 对齐;本条 LIFT 聚焦语言模型预训练中的潜在信息反馈,与它们无直接事实重叠,增量在于 Transformer 内部跨层/跨步状态反馈与教师监督。

局限与不确定性

  • 摘要未给出具体评估数据集、基线名称、指标数值、统计显著性与消融细节:待核实。
  • “轻微计算开销”及“随模型规模增大而减少”的具体曲线、推理延迟与显存影响:待核实。
  • 教师状态来自现成预训练 LM,教师质量、规模、任务匹配度对 LIFT 的影响:待核实。
  • 状态跟踪任务的具体设置、失败 Transformer 的类型与规模:待核实。
  • 135M–1B 之外更大模型、不同架构、多模态或非语言任务的泛化性:待核实。
  • 是否开源代码、模型权重及可复现性细节:待核实。
  • 预训练数据、token 预算与计算预算的具体定义:待核实。

可用于图书/PPT/简报的角度

  • “Transformer 的信息流瓶颈”:前馈架构为何限制跨生成步状态传播。
  • “教师强制不只用于 token,也可用于潜在状态”:把循环学习变成并行可训练的预测问题。
  • “预训练阶段的架构创新”:不只在推理时改循环,而在预训练中注入反馈。
  • “样本效率与状态跟踪”:小模型 + 反馈 vs. 更大模型 + 更多数据。
  • “潜在信息反馈”作为 Transformer、RNN 与状态空间模型之间的中间路线。
  • 可视化建议:前馈信息流 vs. LIFT 反馈回路;预训练并行 vs. 推理反馈。
  • 引用具体数字和图表前需回到原文核实。

原始材料

  • 英文标题:Pretraining Latent Information Feedback Transformers with Teacher Supervision
  • 英文关键词:Latent Information Feedback Transformer (LIFT); teacher supervision; pretraining; language models; deep-to-shallow feedback; recurrent-state learning; next-token distribution; state prediction
  • 作者:Dor Tirosh, Ido Amos, Mor Geva
  • arXiv ID:2609.38149v1
  • 发布时间:2026-09-29T17:57:40Z
  • 更新时间:2026-09-29T17:57:40Z
  • 主分类:cs.CL
  • 分类:cs.CL
  • 摘要 URL:https://arxiv.org/abs/2609.38149v1
  • PDF URL:https://arxiv.org/pdf/2609.38149v1

发现内容有误?

免登录提交,管理员审核并完成修正后才会公开修订说明。

值得关注

它直接挑战标准 Transformer 的前馈信息流限制,探索“深度到浅层反馈”能否在预训练中被学习。 通过教师监督预计算状态,把通常需要序列递归的状态学习变成可并行训练的目标,兼顾表达与效率。 状态跟踪任务的受控结果暗示:反馈结构可能比单纯扩大数据或模型更有效地解决某些程序性/推理任务。 与既有卡片的关系:现有 MMOE、DynoDINO、LEMUR 卡…

相关主题

对应的教材知识点

教材知识点匹配尚未完成。