用少量高质量数据实现大模型对齐,节省超90%成本
Efficient Alignment of Large Language Models via Data Sampling
- 基于信息论筛选小而精的数据子集,替代全量数据对齐
- 仅用不到10%数据,效果接近全量数据对齐,性能提升显著
- 适合资源受限场景下的高效大模型对齐研究与应用
大语言模型对齐旨在使模型输出符合人类价值观、目标与意图。现有对齐方法依赖海量数据、计算资源与时间,且人工反馈数据采集成本高、耗时长。本文首次探究对齐性能随数据量的缩放规律,发现其呈指数饱和趋势——初期快速上升后趋于平缓。基于此,提出一种信息论驱动的数据子采样方法,识别出小规模高质量数据子集,显著降低对齐所需算力与时间。在多个数据集上验证表明,该方法仅使用不足10%的数据,即可达到与全量数据对齐相当的效果,实现超过90%的资源与成本节省,大幅加速大模型对齐进程。
原文摘要 · Abstract (English)
LLM alignment ensures that large language models behave safely and effectively by aligning their outputs with human values, goals, and intentions. Aligning LLMs employ huge amounts of data, computation, and time. Moreover, curating data with human feedback is expensive and takes time. Recent research depicts the benefit of data engineering in the fine-tuning and pre-training paradigms to bring down such costs. However, alignment differs from the afore-mentioned paradigms and it is unclear if data efficient alignment is feasible. In this work, we first aim to understand how the performance of LLM alignment scales with data. We find out that LLM alignment performance follows an exponential plateau pattern which tapers off post a rapid initial increase. Based on this, we identify data subsampling as a viable method to reduce resources required for alignment. Further, we propose an information theory-based methodology for efficient alignment by identifying a small high quality subset thereby reducing the computation and time required by alignment. We evaluate the proposed methodology over multiple datasets and compare the results. We find that the model aligned using our proposed methodology outperforms other sampling methods and performs comparable to the model aligned with the full dataset while using less than 10% data, leading to greater than 90% savings in costs, resources, and faster LLM alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。