TaH2 自适应循环 Transformer 提升 AIME 测试时扩展效率
清华团队提出 TaH2,让循环 Transformer 只对需要的 token 多迭代,AIME 上同等算力多拿 3.4 分,代码已开源。
清华团队提出 TaH2,让循环 Transformer 只对需要的 token 多迭代,AIME 上同等算力多拿 3.4 分,代码已开源。
朋友,这是Yifan Zhang团队提出的一个新架构,叫Recurrent Looped Transformer (RLT),它通过让解码器循环处理每个token来提升效率,和传统模型比,在保持每个token计算成本不变的情况下,深度可以随序列增长,挺有意思的。
循环Transformer训练难、部署贵的问题被Attractor Models用不动点求解优雅解决,做语言模型预训练或推理增强的团队值得关注——它用更少参数和成本实现了对更大模型的超越。