用轻量模型集成实现高效对话偏好对齐,性能接近大模型。
LoRA-LiteE: A Computationally Efficient Framework for Chatbot Preference-Tuning
- 结合SFT、LoRA与集成学习,用小模型组实现高效对齐。
- 在有限资源下,性能超越单个大模型,接近未微调GPT-4。
- 适合算力受限场景,提升对话系统可部署性。
有效偏好对齐对使聊天机器人响应符合人类期望至关重要,能提升用户满意度和参与度。传统方法如强化学习人类反馈(RLHF)在先进模型(如GPT-4)中表现优异,但计算开销大,限制了其可扩展性和普及性。为此,本文提出LoRA-Lite集成框架(LoRA-LiteE),融合监督微调(SFT)、低秩适配(LoRA)与集成学习技术,通过轻量级模型聚合预测,在性能与计算成本间取得平衡。基于Chatbot Arena基准数据集,我们对比了LoRA-LiteE模型、不同规模基线模型及使用RLHF训练的GPT-4。实证结果表明,所提模型在资源受限条件下性能优于单一大模型,并达到未微调GPT-4水平。研究验证了该框架在资源受限环境中实现人类偏好预测的可行性与高效性,显著提升了聊天机器人系统的可扩展性与可用性。
原文摘要 · Abstract (English)
Effective preference tuning is pivotal in aligning chatbot responses with human expectations, enhancing user satisfaction and engagement. Traditional approaches, notably Reinforcement Learning from Human Feedback (RLHF) as employed in advanced models like GPT-4, have demonstrated considerable success in this domain. However, RLHF methods are often computationally intensive and resource-demanding, limiting their scalability and accessibility for broader applications. To address these challenges, this study introduces LoRA-Lite Ensemble (LoRA-LiteE), an innovative framework that combines Supervised Fine-tuning (SFT) with Low-Rank Adaptation (LoRA) and Ensemble Learning techniques to effectively aggregate predictions of lightweight models, which aim to achieve a balance between the performance and computational cost. Utilizing the Chatbot Arena benchmark dataset, we conduct a comprehensive comparative analysis among our LoRA-LiteE model, corresponding base models at different scales, and GPT-4 trained with RLHF. Our empirical results demonstrate that the proposed LoRA-LiteE model achieves comparable performance to un-finetuned GPT-4 and outperforms the single larger-scale models under limited resource constraints. These findings highlight that our LoRA-LiteE provides a feasible and efficient methodology for human preference prediction in chatbot systems, enhancing scalability and accessibility, and thereby broadening the applicability of preference-tuned chatbots in resource-constrained environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。