跳到正文
原文
rohanpaul_ai· @rohanpaul_ai · X·· 4 小时前AI 评分51

论文:小决策模型循环复用同一层可显著提升准确率

AI 导读

论文提出 SanSi,通过循环 Ouro-1.4B(一个预训练为可复用自身层的模型)最多 8 次并在每轮读取选项概率来提升小决策模型准确率,每轮都训练为可作答,因此可在 1 到 8 轮间任意停止。

正文 · 原文

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.

来源:rohanpaul_ai · x.com