让大模型学会识别作者的语言风格,提升文本生成与理解能力。
Addressing the Ecological Fallacy in Larger LMs with Human Context
- 用作者历史文本构建上下文,设计新预训练任务HuLM。
- 在80亿参数模型上,微调时引入作者上下文可显著提效。
- 适合需要个性化语言建模的场景,如对话系统、内容创作。
语言模型训练与推理常忽略一个基本语言事实:同一作者写的多段文本间存在依赖关系。先前研究显示,纠正这种‘生态谬误’能大幅提升小型(约124M)GPT类模型性能。本文探讨在更大规模的80亿参数Llama模型中,通过特定任务HuLM建模作者语言上下文是否也能带来类似收益。我们探索了利用作者历史文本时间顺序信息进行预训练与微调的变体。实验表明,仅在微调阶段引入作者上下文(即HuFT),结合QLoRA方法,即可显著优于标准微调;此外,基于QLoRA的持续HuLM预训练,使模型在八项下游任务上仅用线性分类器即可实现性能提升。结果表明,将语言置于其原始生成者——作者的上下文中建模,具有重要价值。
原文摘要 · Abstract (English)
Language model training and inference ignore a fundamental linguistic fact -- there is a dependence between multiple sequences of text written by the same person. Prior work has shown that addressing this form of \textit{ecological fallacy} can greatly improve the performance of multiple smaller (~124M) GPT-based models. In this work, we ask if addressing the ecological fallacy by modeling the author's language context with a specific LM task (called HuLM) can provide similar benefits for a larger-scale model, an 8B Llama model. To this end, we explore variants that process an author's language in the context of their other temporally ordered texts. We study the effect of pre-training with this author context using the HuLM objective, as well as using it during fine-tuning with author context (\textit{HuFT:Human-aware Fine-Tuning}). Empirical comparisons show that addressing the ecological fallacy during fine-tuning alone using QLoRA improves the performance of the larger 8B model over standard fine-tuning. Additionally, QLoRA-based continued HuLM pre-training results in a human-aware model generalizable for improved performance over eight downstream tasks with linear task classifier training alone. These results indicate the utility and importance of modeling language in the context of its original generators, the authors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。