Harness-Aware Distillation 框架通过对比教师模型在有无 harness 信息下的动作差异训练小模型智能体,无需任务奖励或成功标签。该方法将学生在 ALFWorld 上使用 harness 信息的比例从 65.7% 提至 73.1%,未见任务成功率达 63.4%,高于最佳基线 47.0% 并超过 8B 教师模型,还能摆脱 59.7% 的停滞情况。
Really nice paper on harness-aware distillation for small language model agents.
(bookmark it)
It's a super interesting distillation framework for small language model agents that are deployed with a harness.
In simple terms, you run the big model with and without harness info, then train the small model on cases where its action changes.
Researchers show that adding the harness to on-policy distillation raises how often the student uses harness information on ALFWorld (65.7% to 73.1%) but leaves success flat (43.1% to 43.5%).
Their method, Harness-Aware Distillation, queries the same teacher with and without the harness information and trains the student to prefer the action chosen with it. A filter drops pairs whose preferred action contradicts the harness records. The method uses no task rewards or success labels.
The student reaches 63.4% on unseen ALFWorld tasks, compared with 47.0% for the best baseline, and exceeds its 8B teacher. It also escapes 59.7% of stalls, while the baselines stay near the untrained student's 46.8%.
Paper: https://arxiv.org/abs/2610.02858
Chat with Paper: https://academy.dair.ai/papers/harness-aware-distillation-for-small-language-model-agents-2610.02858
来源:elvis · x.com