arXiv:2502.11062cs.CL2025-02ACL被引 5

用梯度图捕捉指令间关系,提升小样本领域模型训练效果

Beyond Similarity: A Gradient-based Graph Method for Instruction Tuning Data Selection

  • 构建指令梯度图,建模指令间的联合分布与依赖关系
  • 在数据稀缺场景下,性能显著优于传统相似性筛选方法
  • 适合需要高效微调的领域专用模型开发者

大语言模型(LLMs)在指令微调中展现出强大泛化能力,但领域专用数据稀缺严重制约其在特定任务上的表现。现有方法多依赖目标领域与通用数据集间的相似性进行数据筛选,却忽视了指令之间的联合分布,导致学习效率低、知识迁移不充分。为此,我们提出G2IS(基于梯度的指令选择方法),通过构建混合梯度指令图,捕捉指令间的联合分布与相互依赖关系,提升领域适应效率。进一步设计梯度游走算法优化数据选择过程,增强训练效果与效率。实验表明,G2IS在多个领域适应任务中均优于传统方法,尤其在复杂且数据稀疏的场景下表现突出,验证了其在发展大规模领域专用模型中的潜力。

原文摘要 · Abstract (English)

Large language models (LLMs) have shown great potential across various industries due to their remarkable ability to generalize through instruction tuning. However, the limited availability of domain-specific data significantly hampers their performance on specialized tasks. While existing methods primarily focus on selecting training data from general datasets that are similar to the target domain, they often fail to consider the joint distribution of instructions, resulting in inefficient learning and suboptimal knowledge transfer. To address these challenges, we introduce G2IS (Gradient-based Graph Instruction Selection), a novel method that constructs a mixed gradient-based instruction graph to capture the joint distribution and interdependencies between instructions. By accounting for the relationships between instructions, G2IS improves domain adaptation efficiency. Additionally, we propose a gradient walk algorithm to refine the data selection process, enhancing both training effectiveness and efficiency. Our experiments demonstrate that G2IS outperforms traditional methods across various domain adaptation tasks, yielding significant performance gains, particularly in complex, data-scarce scenarios. These results underscore the potential of G2IS in advancing the development of large, domain-specific models.

指令微调数据选择领域适应

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。