跳到正文
Rohan Paul· @rohanpaul_ai · X·· 2 小时前AI 评分53
AI 导读

一篇论文提出 SanSi 方法,让预训练时就学会复用层的小模型 Ouro-1.4B 在给出答案前多次重跑同一批层,从而提升决策准确率。在 10,027 条测试决策上,SanSi 得分 72.0%,同结构但不循环的模型为 58.4%,代价是 7.7 倍算力;在比训练时更长的 liar chains 上,它达到 72.6%,而 4.2B 单次前向模型只有 50.0%,即随机水平。

正文

This paper finds that small decision models get far more accurate by rerunning the same layers several times before giving an answer.

Many LLM calls only need a choice, like spam or not spam. Typed decision models answer these in a single pass, which fails when a decision needs several dependent steps.

SanSi loops Ouro-1.4B, a model pretrained to reuse its layers, and reads option probabilities after each of 8 loops. Every loop is trained to answer, so the same model can stop anywhere from 1 to 8 loops.

On 10,027 test decisions, SanSi scored 72.0% versus 58.4% for a same-shape model without loops, at 7.7x the compute. On liar chains longer than any seen in training, it reached 72.6% while a 4.2B single-pass model scored 50.0%, the chance level.

If memory limits your decision-style calls, try a looped small model before moving to a model 3x larger.

来源:Rohan Paul · x.com