arXiv:2410.04715cs.CLcs.AI2024-10AAAI被引 11

用正交性挑选互补规则,提升大模型微调数据质量

Selection of LLM Fine-Tuning Data based on Orthogonal Rules

  • 通过正交性度量筛选互补规则,避免冗余
  • 在多个领域上提升评分准确率和微调性能
  • 适合需要高质量数据的模型训练场景

高质量训练数据对大语言模型性能至关重要。现有方法依赖人工设计的规则,通过大模型打分选数据,但常受启发式方法限制,缺乏可解释的规则评估指标,且泛化能力差。本文提出一种基于正交性的规则驱动数据选择框架:首先用大模型生成覆盖多维度的数据质量规则,再利用行列式点过程(DPP)筛选出最独立的规则,最后用这些规则对全量数据打分并选取高分样本用于下游任务(如模型微调)。在IMDB、Medical、Math和Code四个领域进行的实验表明,该方法在与真实评分对齐及微调后模型表现两方面均优于强基线,显著提升数据选择效果。

原文摘要 · Abstract (English)

High-quality training data is critical to the performance of large language models (LLMs). Recent work has explored using LLMs to rate and select data based on a small set of human-designed criteria (rules), but these approaches often rely heavily on heuristics, lack principled metrics for rule evaluation, and generalize poorly to new tasks. We propose a novel rule-based data selection framework that introduces a metric based on the orthogonality of rule score vectors to evaluate and select complementary rules. Our automated pipeline first uses LLMs to generate diverse rules covering multiple aspects of data quality, then rates samples according to these rules and applies the determinantal point process (DPP) to select the most independent rules. These rules are then used to score the full dataset, and high-scoring samples are selected for downstream tasks such as LLM fine-tuning. We evaluate our framework in two experiment setups: (1) alignment with ground-truth ratings and (2) performance of LLMs fine-tuned on the selected data. Experiments across IMDB, Medical, Math, and Code domains demonstrate that our DPP-based rule selection consistently improves both rating accuracy and downstream model performance over strong baselines.

数据筛选大模型微调规则系统DPP

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