← 返回 2026-10-09

SanSi:面向 System 1.5 思维的循环式类型化决策模型 SanSi: A Looped Typed Decision Model for System 1.5 Thinking

Shuyu Gan, Young-Jun Lee, Dongyeop Kang 📅 2026-10-06 👍 16 2026-11-11 18:30
RLHF裁判 循环Transformer 概率校准 测试时算力 潜在推理 类型化决策模型

同一组层循环8次并逐环读出概率,以算力换参数,1.4B决策模型比同形状单次模型高13.5点

前置知识

类型化决策模型(Typed Decision Model)

一类不生成文本、直接输出决策的模型:调用方给定状态(一段文本、规则或记录)、问题和预先声明的 K 个选项,模型在单次前向传播中为每个选项返回一个概率,例如判断一条消息是否为垃圾邮件。商业 Jev API 与其开源复现 Kev 都遵循这一契约,返回的概率被下游系统用于接受、升级或路由决策。

SanSi 就是一个类型化决策模型,全文评估的正是这种选项概率的质量(准确率、校准、证据察觉),循环机制必须在不破坏这一契约的前提下叠加算力。

循环 Transformer(Looped Transformer)

不堆叠不同层,而是让同一组 Transformer 层反复作用于自己的隐状态:$h_t = NF_\theta(h_{t-1})$,每过一遍就精炼一次表示。思想源自 Universal Transformers,可学习迭代算法并对更长输入泛化。Ouro 就是为循环而预训练的语言模型(预训练时用 4 个循环)。

SanSi 的全部增益来自把这个栈循环 8 次,且消融证明:若骨干没有循环预训练,事后给它加环毫无收益,这是理解本文结论边界的关键。

System 1.5 与潜在推理(Latent Reasoning)

认知双系统中 System 1 是快速直觉(对应单次前向),System 2 是慢速深思(对应生成思维链文本)。潜在推理指在隐状态空间中迭代计算而不生成 token,介于两者之间,故称 System 1.5:比单次多花算力,但不付生成文本的 token、延迟和解析代价。

这是论文的概念框架——循环被定位为 System 1.5,理解它才能明白作者为何反复强调不生成 token、保持类型化契约。

LoRA(低秩适配)

冻结预训练权重,只训练注入到注意力与前馈投影中的低秩矩阵(如 rank 64 的 $U_tV_t$ 修正),可训练参数量和显存开销很小,效果接近全量微调。

SanSi 冻结骨干、只训 60.6M 的 LoRA 适配器与 0.27M 读出头(共 61M),这使循环带来的增益不会被新参数混淆,也让训练成本极低、复现门槛小。

严格评分规则(Cross-Entropy 与 Brier 分数)

评分规则衡量概率预测与目标分布的差距,严格指当且仅当预测恰等于目标分布时损失最小。交叉熵 $-\sum_k y^*_k\log p_k$ 只看正确选项的概率、重罚自信的错误;Brier 分数 $\sum_k (p_k-y^*_k)^2$ 考虑所有选项且有界。两者都是严格评分规则。

SanSi 每环的损失是 CE+Brier 之和;消融显示去掉 Brier 项校准变差、无答案项更敢乱答,而 RL 奖励在期望上等价于半个 Brier 的策略梯度,这是读懂消融章节的钥匙。

ECE 与 AUROC(校准与排序指标)

ECE(期望校准误差)把样本按置信度分桶,比较每桶平均置信度与准确率之差,衡量说 90% 是否真有 90% 对;AUROC 是按分数把两类样本排开的概率,0.5 为随机、1 为完美。本文用 AUROC 检验置信度能否区分答对/答错、有证据/无证据的样本。

RQ3 的结论全靠这两个指标:校准只在第 3 环最好,置信度对错题同样上升(区分对错的 AUROC 仅从 0.760 到 0.795),读这些数字需要这两个概念。

GRPO 与无金标准奖励

GRPO 是策略梯度强化学习算法:同一问题采样一组答案,以组内平均奖励为基线计算优势后更新策略,不需要价值网络。当任务没有可验证的金标准答案时,可用另一个模型(裁判)对候选输出打分作为奖励信号。

第 8 节把冻结的 SanSi 当作唯一奖励来训练生成器,验证循环概率作为裁判的价值;其中奖励=正确性减去自报概率的设计,正与 Brier 梯度的理论联系相呼应。

研究动机

越来越多的语言模型调用要的不是文本而是一个决策:这条消息是不是垃圾、这个案例是否满足政策、四个答案中哪一个能从文档推出。类型化决策模型(商业 Jev API 及开源复现 Kev)正是为此设计:调用方声明选项,模型单次前向传播返回每个选项的概率,无需生成和解析文本。但单次前向给每个决策的计算量是固定的,很多决策需要多步依赖推理——例如 Omar 诚实、Bert 说 Omar 说真话、Ben 说 Bert 说谎这样的链。Jev API 在此类链式任务上 1 步准确率 100%,3 步降至 62.5%,6 步起接近随机水平。想提高准确率只有两条路:换更大的模型(增加参数与显存),或生成思维链(增加 token 和延迟,且失去类型化契约与校准概率)。

本文的目标是本文要在单次前向(System 1)与生成式推理(System 2)之间寻找中间态:循环(looping),即把同一组 Transformer 层递归地作用于自己的隐状态若干次后再做一次类型化读出,作者称之为 System 1.5 思维。具体目标是构建 SanSi 模型族,并系统回答四个问题:(1) 循环到底带来多少准确率、付出什么代价(RQ1);(2) 答案如何随循环逐次演变、何时定型(RQ2);(3) 循环对调用方赖以决策的概率做了什么——置信度、校准、对缺失证据的察觉(RQ3);(4) 循环能否买到推理深度、是否泛化到训练未见深度(RQ4)。

与已有工作不同的是,已有工作的切入各有限制:Jev-LCT 只循环非循环模型最上面两层且早早退出,无法说明完整循环栈的潜力;循环语言模型研究集中于预训练方法、稳定性、停机准则和架构变体;把循环嫁接到非循环预训练模型的工作(如 Retrofitted Recurrence)需要远大于微调的训练预算。SanSi 的独特之处在于:起点是整个网络都为循环而预训练的 Ouro 主干;每个循环都读出并用严格评分规则(交叉熵+Brier)训练,使一个模型同时服务 1 到 8 环的任意预算;并用同一数据、同一配方、同一随机种子做受控对照(同形状的 SmolLM2、更新家族的 Qwen3.5、对手官方配方的 Kev-4B),把循环的贡献与参数量、骨干强弱、训练配方干净地剥离。

核心方法

直觉上,SanSi 让小模型遇到难题时多想几遍:把 24 层 Transformer 栈反复套用自己的隐状态,每转一圈就重新读一次选项概率,答案从错误逐步修正到正确(图 2b 的 PAWS 例子中选项 A 的概率从 0.19 到 0.69 再到 0.98)。技术路线上:以预训练为循环而生的 Ouro-1.4B/2.6B 为骨干并冻结,只训练 LoRA 适配器(rank 64,60.6M 参数,所有环共享)和每环一个小读出头(0.27M);递归定义为 $h_t = NF_\theta(h_{t-1})$,$t=1,\dots,T$,输入 token 不再注入;在第 $t$ 环的末提示 token 上读出选项概率;训练时每个环都计算损失 $\mathcal{L}=\frac{1}{T}\sum_{t=1}^{T}\big[-\sum_k y^*_k\log p_{t,k}+\sum_k(p_{t,k}-y^*_k)^2\big]$,每环损失的梯度反传穿过之前所有环。$T=8$,是 Ouro 预训练循环数的两倍。

核心创新有二。其一是每环都读、每环都训:以往循环或早退模型只在最终深度输出,而 SanSi 在每个循环后都读出选项概率并用严格评分规则训练,使同一模型天然成为 anytime 模型——一次前向同时产生预算 1 到 8 的全部决策,调用方按算力预算取用第几环即可。其二是用计算换参数且不破坏类型化契约:同样的层跑 $T$ 遍,每环都是完整一次前向,推理成本 7.7 倍但参数不变,调用方拿到的仍是校准的选项概率而非生成文本。这与 Jev-LCT 只循环顶部两层、与自适应计算/早退研究提出停机规则都不同——SanSi 不提停机规则,预算由调用方显式选择。消融进一步揭示关键前提:把循环加到非循环预训练的 SmolLM2 上训练毫无收益(第 8 环 58.3% 对单次 58.4%),收益根植于 Ouro 的循环预训练,配方只是把这种潜力转化为决策能力。

方法步骤详情

第一步,任务渲染:每条决策渲染为提示词,依次是状态、Question: 加问题、Options: 加形如 (A)…(B)… 的选项、Answer:,在最后一个 token 处读出;选项由单个字母 token 命名,最多 26 个。第二步,循环计算:token 嵌入 $h_0$ 经 $h_t=NF_\theta(h_{t-1})$ 递归 8 次,$N$ 是模型最终归一化层,成本为 $T$ 次前向。第三步,读出:取 $h_t$ 在末提示 token 的行 $z_t$,$\ell_t=(W+U_tV_t)z_t/\tau_t$,$p_t=\mathrm{softmax}(\ell_t)$,只保留 K 个声明选项;$W$ 是冻结语言模型头中选项字母对应的行,rank-16 修正 $U_tV_t$ 与温度 $\tau_t$ 每环一套、初始化为恒等。第四步,训练目标:目标分布为 one-hot(有正确项)、均匀分布(关键证据被移除的无答案项)或众包标注分布;损失为 CE+Brier 按环平均。第五步,优化:AdamW,LoRA 学习率 $10^{-4}$、读出 $10^{-3}$,1,000 步、每步 16 条,bfloat16;主模型训练约 313 GPU 分钟(单张 RTX A6000 约 5.2 小时)。

技术新颖性

技术新颖性有四。第一,把循环从预训练技巧升级为类型化决策模型的测试时算力旋钮,并证明其有效性依赖于为循环而预训练的骨干——这与 Universal Transformers、Retrofitted Recurrence 等在训练期引入循环的工作本质不同。第二,anytime 读出设计:每环一个从冻结 LM 头恒等初始化出发的 rank-16 读出头,使读哪一环成为部署时而非训练时的选择;消融显示只在最后一环放损失会使早期环崩溃(第 1 环仅 35.6%)。第三,训练目标的理论刻画:正确性减去自报概率的 REINFORCE 奖励在期望上等价于半个 Brier 得分的策略梯度,作者据此实现 Jev 式 RL 训练(71.7% 对监督 72.0%),并发现 RL 版本校准更好(ECE 0.076 对 0.093)但漏检缺失证据更差(证据 AUROC 0.919 对 0.935)。第四,评估方法学:以同形状同配方对照、以更新家族做参照、以对手官方配方做公平性检验,并按题型与训练距离分格报告,使循环增益的归因非常干净。

SanSi. (a) Training: one stack of 24 layers is applied T times; after every loop the option probabilities are read and trained towards the target. (b) Test time: a test item from PAWS (Zhang et al., 2019) followed through the loops; the first loop prefers the wrong option, the later loops the correct one.
Figure 2: SanSi. (a) Training: one stack of 24 layers is applied T times; after every loop the option probabilities are read and trained towards the target. (b) Test time: a test item from PAWS (Zhang et al., 2019) followed through the loops; the first loop prefers the wrong option, the later loops the correct one.

实验结果

在 10,027 条测试决策(59 个来源)上,SanSi 达 72.0%:比同形状同数据同配方的 SmolLM2-1.7B(58.4%)高 13.5 点(95% 区间 [12.5, 14.5],三个种子 12.9–14.8),比同规模较新的 Qwen3.5-2B 高 5.3 点,仅比三倍参数的 Qwen3.5-4B(73.8%)低 1.8 点;Ouro-1.4B 单环只达 58.6%,证明增益来自循环而非骨干。增益随任务而变:分类仅 +3.1,多步推理 +15.1、长文档 +16.7、知识 +17.6(该类型只在测试集出现,第 1 环 47.1% 到第 8 环 70.2%)。代价是推理 7.7 倍、训练 9.5 倍 GPU 时间;3 环即拿 88% 增益(70.4%)。逐环看:第 2 环 +8.5、第 3 环 +3.5、第 4 环 +1.2、5–8 环合计 +0.4;第 8 环累计修正 21.0%、破坏 7.3%,约三比一。概率上,证据 AUROC 从 0.835 升至 0.935,对 382 个无答案项给硬答案的比例从 27.1% 降至 17.5%,但置信度在对错题上都上升(0.769→0.869 与 0.567→0.660),ECE 第 3 环最优(0.082)、第 8 环回升至 0.093。深度任务(训练 k≤8、测试 k≤16):谎者链未见深度 SanSi 72.6%,Qwen3.5-4B 仅 50.0%(随机水平);对象交换 63.7% 对 34.2%。作无金标准 RL 裁判时,第 8 环奖励把生成器 F1 从 39.5 提到 47.3(+7.7),第 1 环奖励反而损害(29.1)。

The decision suite: number of items by type and by distance from the training data. Type C contains the unanswerable and crowd-labelled items. The test set also contains the 231 public JevBench items (10,027 test items in total).
Table 1: The decision suite: number of items by type and by distance from the training data. Type C contains the unanswerable and crowd-labelled items. The test set also contains the 231 public JevBench items (10,027 test items in total).
Main results on the 10,027 test items: mean ± standard deviation over three training seeds (bold: best). All models are trained on the same data, with our recipe except Kev-4B (our data), which uses Kev's.
Table 2: Main results on the 10,027 test items: mean ± standard deviation over three training seeds (bold: best). All models are trained on the same data, with our recipe except Kev-4B (our data), which uses Kev's.
Training choices: accuracy on the 10,027 test items and ECE, mean ± standard deviation over three seeds. Every variant differs from SanSi in one choice.
Table 3: Training choices: accuracy on the 10,027 test items and ECE, mean ± standard deviation over three seeds. Every variant differs from SanSi in one choice.
What looping costs and where it helps. (a) Accuracy against GPU time of one pass over the test set (unit: one loop of Ouro-1.4B). (b) Accuracy by item type, ordered by the gain of SanSi at loop 8 over SmolLM2-1.7B (thick line; number on the right: gain in points).
Figure 3: What looping costs and where it helps. (a) Accuracy against GPU time of one pass over the test set (unit: one loop of Ouro-1.4B). (b) Accuracy by item type, ordered by the gain of SanSi at loop 8 over SmolLM2-1.7B (thick line; number on the right: gain in points).
SanSi read after each loop on the 10,027 test items (mean ± s.d., three seeds); horizontal lines: single-pass models. (a) Accuracy, with the gains of loops 2 and 3. (b) Mean confidence on right and wrong answers. (c) ECE on answerable items. (d) Evidence AUROC.
Figure 4: SanSi read after each loop on the 10,027 test items (mean ± s.d., three seeds); horizontal lines: single-pass models. (a) Accuracy, with the gains of loops 2 and 3. (b) Mean confidence on right and wrong answers. (c) ECE on answerable items. (d) Evidence AUROC.
How SanSi's answers change across loops (9,645 answerable items; mean ± s.d., three seeds). (a) Answers fixed or broken by loop T relative to loop 1. (b) Answers that the loops after T will still fix or break. (c) Mean settling loop by how many single-pass models answer correctly. (d) Share of loop-8 answers that are right, by settling loop.
Figure 5: How SanSi's answers change across loops (9,645 answerable items; mean ± s.d., three seeds). (a) Answers fixed or broken by loop T relative to loop 1. (b) Answers that the loops after T will still fix or break. (c) Mean settling loop by how many single-pass models answer correctly. (d) Share of loop-8 answers that are right, by settling loop.
Accuracy by depth k, with SanSi read after 1, 2, 4 and 8 loops. Shaded: depths unseen in training (k > 8). Dashed: single-pass Qwen3.5-4B trained on the same data.
Figure 6: Accuracy by depth k, with SanSi read after 1, 2, 4 and 8 loops. Shaded: depths unseen in training (k > 8). Dashed: single-pass Qwen3.5-4B trained on the same data.
Four test items: SanSi's probability for each option after every loop, and the three single-pass models (seed 0). Correct option starred and outlined; column maxima in bold. Item 3 has no correct option: its deciding sentence (struck out) was removed.
Figure 7: Four test items: SanSi's probability for each option after every loop, and the three single-pass models (seed 0). Correct option starred and outlined; column maxima in bold. Item 3 has no correct option: its deciding sentence (struck out) was removed.
GRPO on 2WikiMultiHopQA with SanSi's probability as the only reward: test F1 (3,000 questions) by the loop at which SanSi is read (filled: mean of three seeds; hollow: seeds; dashed: before training).
Figure 9: GRPO on 2WikiMultiHopQA with SanSi's probability as the only reward: test F1 (3,000 questions) by the loop at which SanSi is read (filled: mean of three seeds; hollow: seeds; dashed: before training).
查看结构化数据
任务指标本文基线提升
10,027 条类型化决策套件(全量) 准确率 SanSi 72.0% SmolLM2-1.7B(同形状同配方)58.4% +13.5 点
同套件,对比同规模较新骨干 准确率 SanSi 72.0%(第 8 环) Qwen3.5-2B 66.7% +5.3 点(第 2 环即持平)
同套件,对比三倍参数单次模型 准确率(参数仅 1/3,算力 3.3×) SanSi 72.0% Qwen3.5-4B 73.8% −1.8 点(落后但接近)
同套件,对手官方配方对照 准确率 SanSi 72.0% Kev-4B(our data)74.3% −2.4 点(参数仅 1/3;SanSi-2.6B 反超 +1.5)
缺失证据察觉(786 条可答/无答案配对项) 证据 AUROC / 硬答案率 0.935 / 17.5%(第 8 环) 0.835 / 27.1%(第 1 环) +0.099 / −9.7 点
谎者链,未见深度 k=9–16 准确率(随机 50%) SanSi 第 8 环 72.6% Qwen3.5-4B 单次 50.0% +22.6 点
对象交换,未见深度 k=9–16 准确率(随机 20%) SanSi 第 8 环 63.7% Qwen3.5-4B 单次 34.2% +29.5 点
SanSi 作唯一奖励的 GRPO(2WikiMultiHopQA) 生成器 token F1 47.3(第 8 环奖励) 无验证器训练 39.5 +7.7 点

局限与改进

作者承认的限制:结论建立在单一循环预训练家族 Ouro(1.4B/2.6B)上,多数分析用 1.4B;增益依赖循环预训练,能否推广到其他或更大骨干是开放问题;读固定环数、无早退或停机规则;深度结论基于两个程序生成任务;验证器案例只用一个数据集一个生成器;套件仅英语且由既有数据集拼成,无法排除骨干预训练时见过部分公开数据;每组只 3 个种子,小子集差异需结合区间读。我自己的观察:其一,等算力下大模型仍赢(3 环时 Qwen3.5-4B 领先 3.4 点),8 环推理 7.7 倍、2.6B 版 14.8 倍成本,部署经济性存疑;其二,答案定型后置信度继续攀升,ECE 从第 3 环的 0.082 回升到 0.093,近迁移组尤其糟(0.053→0.149),过度自信风险真实;其三,11.7% 的题曾在某些环正确、最终环却错,但论文未给选环规则,放弃了 83.3% 的理论上限;其四,跑超过训练环数会掉点(第 16 环 70.1%),深度扩展有硬上限;其五,纯知识缺失仍会全程高置信答错(ARC 例 0.96→0.99),循环无法替代检索。

独立分析的弱点

独立分析的弱点:(1) 固定环数无停机——56.7% 的答案第 1 环后再不变,却仍要跑满 8 环付 7.7 倍算力,而晚定型的答案错误率更高(第 8 环定型者仅 35.6% 正确)且置信度虚高,改进方向是引入 PonderNet 式的每环停机概率,或用定型环号作为人工审核的触发信号;(2) 置信度对错题同样上升,区分对错能力几乎没变(AUROC 0.760→0.795),说明循环修正答案但不教自我怀疑,改进方向是在定型后惩罚置信度增长,或默认平均 8 环概率(论文显示这一免费操作把 ECE 从 0.093 砍到 0.044);(3) 循环不弥补知识缺口:知识题仍落后 Qwen3.5-4B 4.9 点,缺知识的题全程高置信答错,改进方向是与检索增强结合,让循环负责整合、检索负责知道;(4) 无法事后加装循环:给 SmolLM2 加环要么训不动(33.5%)要么不用(58.3%),限制了配方对存量非循环模型的适用性,改进方向是参照 McLeish 等用大预算继续预训练植入循环;(5) RL 版本漏检证据变差(硬答案 21.6% 对 17.5%),若用于安全关键的路由决策应保留 Brier 监督项或混合两种目标。

未来方向

作者提出的方向:为每个条目选择循环数的规则——11.7% 的题曾在某环正确而最终错误,理想选环上限可达 83.3%;把发现推广到更多、更大的循环预训练骨干;把无金标准奖励下的验证器(judge)规模化。基于成果可延伸的:把定型环号用作难度感知路由——易题 1 环即走、难题多环并触发升级到人或生成式推理,把 anytime 特性变成真正的自适应计算;构建 System 1/1.5/2 可切换的频谱,用 SanSi 概率决定何时升级到显式思维链;探索每环差异化训练而非共享同一套 LoRA;在多语言和真实生产数据(内容审核、合规检查)上验证套件结论;对循环中学到的迭代算法(如谎者链上真值逐链翻转)做机制可解释性分析;以及把循环骨干与外部检索、工具调用结合以弥补知识短板。

复现评估

复现条件相当友好。骨干 Ouro-1.4B/2.6B 是公开的预训练模型;评估套件由公开数据集(AG News、MNLI、FOLIO、BBH、MMLU、QuALITY 等 59 个测试源)组装,附录 A/B 给出完整来源、数量与提示渲染格式;两个深度任务为程序生成并附防捷径检验;深度可控、无答案项与众包分布目标都有精确定义。训练只动 61M 参数(LoRA rank 64 加读出),1,000 步、每步 16 条,单张 RTX A6000 约 5.2 小时(313 GPU 分钟)即可复现主模型,Kev-4B 对照每种子约 3.9 小时;测试集一次完整前向约 43 分钟(2,571 秒)。三种子均值、标准差与分组自助法区间都完整报告。论文列有项目页但正文未明确承诺开源代码与数据套件;若放出则复现难度很低,即便只按论文实现,主要工程量也在数据拼装与格式对齐,总体属于小算力可复现的研究。

相似论文推荐