模型跨格式检索能力弱,新方法CASCADE提升多模式知识获取效果。
CASCADE Your Datasets for Cross-Mode Knowledge Retrieval of Language Models
- 用分层数据集和后半段损失训练,捕捉多尺度知识
- 跨格式检索准确率提升显著,优于传统重写数据集方法
- 适合需要跨格式理解的智能问答与知识系统
语言模型在跨模式知识检索方面表现不佳——即在一种格式中学习的知识难以在另一种格式下被有效调用。本文通过控制实验研究了不同模式间随机标记序列记忆的表现,发现多源数据(如维基百科和TinyStories)训练的模型在格式不匹配时准确率显著下降。我们探索数据重写作为解决方案,发现其所需工作量呈类S型增长,代价过高。为此提出CASCADE算法:采用分层数据集,仅对每条序列后半部分计算损失,以捕捉不同尺度的知识。实验表明,CASCADE在统一模型与单一损失函数下仍优于数据重写方法,既揭示了跨模式检索的局限性,也提供了可落地的增强方案。
原文摘要 · Abstract (English)
Language models often struggle with cross-mode knowledge retrieval -- the ability to access knowledge learned in one format (mode) when queried in another. We demonstrate that models trained on multiple data sources (e.g., Wikipedia and TinyStories) exhibit significantly reduced accuracy when retrieving knowledge in a format different from its original training mode. This paper quantitatively investigates this phenomenon through a controlled study of random token sequence memorization across different modes. We first explore dataset rewriting as a solution, revealing that effective cross-mode retrieval requires prohibitively extensive rewriting efforts that follow a sigmoid-like relationship. As an alternative, we propose CASCADE, a novel pretraining algorithm that uses cascading datasets with varying sequence lengths and computing losses on only the second half of each training sequence to capture knowledge at different scales. Our experiments demonstrate that CASCADE outperforms dataset rewriting approaches, even when compressed into a single model with a unified loss function. This work provides both qualitative evidence of cross-mode retrieval limitations and a practical solution to enhance language models' ability to access knowledge independently of its presentational format.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。