arXiv:2508.05496cs.AI2025-08被引 1

用智能选数据+双阶段训练,让大模型推理更高效

InfiAlign: A Scalable and Sample-Efficient Framework for Aligning LLMs to Enhance Reasoning Capabilities

  • 自动筛选高质量推理数据,减少90%以上训练样本
  • 仅用12%数据达7B模型对齐效果,数学推理提升3.89%
  • 适合想低成本提升模型推理能力的研究者与开发者

大型语言模型在复杂任务中展现出强大推理能力,但通过后训练增强这一能力仍需大量数据和计算资源。现有方法多依赖启发式或任务特定的数据筛选策略,难以扩展。本文提出InfiAlign,一种可扩展且样本高效的后训练框架,结合监督微调(SFT)与直接偏好优化(DPO),实现大模型推理能力的增强。其核心是基于多维质量指标的自动化数据筛选管道,从开源推理数据集中自动提取高质量对齐数据。该方法显著提升性能,同时大幅降低数据需求,并可扩展至新数据源。应用于Qwen2.5-Math-7B-Base模型时,SFT模型性能媲美DeepSeek-R1-Distill-Qwen-7B,仅使用约12%训练数据;应用DPO后,在AIME 24/25基准上平均提升3.89%,且在多种推理任务中表现出强泛化能力。结果表明,结合原则性数据选择与全阶段后训练,可实现高效、可扩展的大模型对齐。

原文摘要 · Abstract (English)

Large language models (LLMs) have exhibited impressive reasoning abilities on a wide range of complex tasks. However, enhancing these capabilities through post-training remains resource intensive, particularly in terms of data and computational cost. Although recent efforts have sought to improve sample efficiency through selective data curation, existing methods often rely on heuristic or task-specific strategies that hinder scalability. In this work, we introduce InfiAlign, a scalable and sample-efficient post-training framework that integrates supervised fine-tuning (SFT) with Direct Preference Optimization (DPO) to align LLMs for enhanced reasoning. At the core of InfiAlign is a robust data selection pipeline that automatically curates high-quality alignment data from open-source reasoning datasets using multidimensional quality metrics. This pipeline enables significant performance gains while drastically reducing data requirements and remains extensible to new data sources. When applied to the Qwen2.5-Math-7B-Base model, our SFT model achieves performance on par with DeepSeek-R1-Distill-Qwen-7B, while using only approximately 12% of the training data, and demonstrates strong generalization across diverse reasoning tasks. Additional improvements are obtained through the application of DPO, with particularly notable gains in mathematical reasoning tasks. The model achieves an average improvement of 3.89% on AIME 24/25 benchmarks. Our results highlight the effectiveness of combining principled data selection with full-stage post-training, offering a practical solution for aligning large reasoning models in a scalable and data-efficient manner. The model checkpoints are available at https://huggingface.co/InfiX-ai/InfiAlign-Qwen-7B-SFT.

大模型对齐推理增强数据效率微调

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