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

论文提出 SanSi,通过让 Ouro-1.4B 模型重复运行同一批层最多 8 次并读取每轮选项概率,将小模型决策准确率从 58.4% 提升至 72.0%(10,027 条测试决策,算力为 7.7 倍)。

正文

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