arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~

表示对齐的辅助监督用于语言模型适配

Representation-Aligned Auxiliary Supervision for Language Model Adaptation

Kyuyoung Kim, Peiyao Sheng, Ashwin Hebbar, Peiyang Xu, Yunfei Xie, Kevin Wang, Rui Xin, Chen Wei, Zhangyang Wang, Jinwoo Shin, Pramod Viswanath, Sewoong Oh

arXiv 2610.04098首次发表:更新:

发表机构

KAIST AI; Sentient Labs; Princeton University; Rice University; University of Texas, Austin; University of Washington(韩国科学技术院人工智能研究院; Sentient实验室; 普林斯顿大学; 莱斯大学; 德克萨斯大学奥斯汀分校; 华盛顿大学)

机构由 AI 辅助整理,请以论文原文为准。

AI 中文总结

本研究提出表示对齐的辅助监督方法,利用环境派生任务提升语言模型在结构化领域(如国际象棋)的适配性能,显著改善最优动作预测和评论生成。

AI 中文摘要

语言模型展现出强大的推理能力,但将其适配到结构化领域仍然具有挑战性,且可能产生不一致的结果。我们识别出表示兼容性——模型有效处理结构化任务表示的程度——是适配中的关键因素。我们在国际象棋中对此进行研究,国际象棋提供了一个受控的测试平台,具有精确的语义、可计算的最优动作和多种状态表示,包括符号编码(FEN)和空间格式(ASCII)。我们发现,模型通常对语义等价的输入处理方式差异显著,影响学习和泛化。基于这一观察,我们提出表示对齐的辅助监督,利用以兼容表示表达的环境派生任务来改善对结构化领域的适配。跨模型和表示,辅助监督在相同目标数据下,相对于仅目标训练,持续提高最优动作预测。暴露环境动态的任务比表面或静态监督提供更大且更一致的收益,同时与大幅增加目标任务数据量保持竞争力。此外,ASCII训练的模型比FEN训练的模型更有效地迁移到FEN,甚至在FEN评估上超过了仅FEN目标训练的基线。这些收益还扩展到开放式、基于事实的评论生成,而不仅限于最优动作预测。总体而言,我们的结果表明,在模型兼容表示中的辅助监督能够实现结构化领域的有效适配。

英文摘要

Language models exhibit strong reasoning capabilities, yet adapting them to structured domains remains challenging and can yield inconsistent outcomes. We identify representation compatibility, the extent to which a model effectively processes a representation for a structured task, as a key factor in adaptation. We study this in chess, which provides a controlled testbed with precise semantics, computable optimal actions, and multiple state representations, including a symbolic encoding (FEN) and a spatial format (ASCII). We find that models often process semantically equivalent inputs substantially differently, affecting both learning and generalization. Building on this observation, we propose representation-aligned auxiliary supervision, which uses environment-derived tasks expressed in compatible representations to improve adaptation to structured domains. Across models and representations, auxiliary supervision consistently improves optimal-move prediction relative to target-only training under identical target data. Tasks that expose environment dynamics provide larger and most consistent gains than surface-level or static supervision, while remaining competitive with substantially increasing the amount of target-task data. Moreover, ASCII-trained models transfer more effectively to FEN than FEN-trained models do to ASCII, even surpassing the FEN target-only baseline on FEN evaluation. The gains also extend beyond optimal-move prediction to open-ended, factually grounded commentary generation. Overall, our results show that auxiliary supervision in model-compatible representations can enable effective adaptation in structured domains.

CommentsNeurIPS 2026 Workshop on LP4FM Oral

论文原文

arXiv 摘要页 · PDF 原文 · HTML 原文

↑