[控场AI]
模型PTTT

ParaTTT

一种用于测试时训练(TTT)层的并行化方法,通过迭代精炼内循环梯度轨迹的估计,实现任意深度非线性记忆的并行化,保持精确在线梯度下降的收敛性。

时间轴 (近 90 天)

10月30日

ParaTTT(PTTT)通过迭代精炼内循环梯度轨迹的估计,来并行化任意深度的非线性记忆。

待验证50%
10月30日

ParaTTT 的每次精炼包含三个分块并行的过程:前向传播评估特征轨迹;精确的 gated-delta-rule 过程计算读出轨迹及其特征梯度;反向传播将这些梯度传播为新的估计。

待验证50%
10月30日

精确的在线梯度下降循环是 ParaTTT 迭代的不动点,而解码不需要精炼迭代。

待验证50%
10月30日

作者证明两层变体在充分收缩条件下具有几何收敛性。

待验证50%
10月30日

实践中,两次迭代就足够。

待验证50%
10月30日

在 4K 上下文下,两次精炼步骤的 PTTT 在匹配误差下评估内循环轨迹比使用分块近似的 TTT-MLP 快 32 倍。

待验证50%
10月30日

完整的 340M 模型训练速度快 4.9 倍。

待验证50%
10月30日

在 340M 和 1.3B 参数下训练的模型在语言建模困惑度和零样本准确率上匹配或超过 Gated DeltaNet 和 Mamba-2,并与同尺寸 Transformer 保持竞争力,且比线性循环具有更好的召回能力。

待验证50%

全部知识事实 (8)

来源文章