Sheared-LLaMA-1.3B-ShareGPT

Sheared-LLaMA-1.3B-ShareGPT项目介绍

项目背景

Sheared-LLaMA-1.3B-ShareGPT是一个由普林斯顿自然语言处理团队开发的项目。该项目专注于语言模型的加速预训练，通过结构化剪枝实现。研究人员为了提高模型的效率和推广能力，使用了来自ShareGPT数据集的指令和响应对进行调优训练。

项目内容

目标

项目的核心目标是优化语言模型的预训练效率，同时保证其在处理任务中的表现不受影响。通过这种方式，开发团队希望为传统语言模型提供更高效的选择，满足更广泛应用的需求。

数据集和训练方法

训练数据集选用了ShareGPT数据集中的10,000对指令-响应对，仅使用了初次对话的数据。此选择的目的是利用真实世界的对话数据对模型进行微调，使其更好地理解和生成自然语言。训练使用了以下提示:

你是一个乐于助人的助手。请写一个恰当完成请求的回复。\n\n### 输入：\n{input}\n\n### 回复：

这种方法确保模型能够生成上下文相关且准确的响应。

如何使用模型

研究者提供了一个简便的方法来加载和使用该模型。用户可以通过使用transformers库中的LlamaModelForCausalLM类来加载Sheared-LLaMA-1.3B-ShareGPT模型：

from transformers import LlamaModelForCausalLM
model = LlamaModelForCausalLM.from_pretrained("princeton-nlp/Sheared-LLaMA-1.3B-ShareGPT")

这一实现使得开发者能够更为方便地将模型集成到他们的应用中。

重要资源

学术论文: arXiv 预印本
项目代码仓库: GitHub - LLM-Shearing
预训练模型: Hugging Face - Sheared-LLaMA-1.3B, Sheared-LLaMA-2.7B

参考引文格式

如果该模型为你的研究或应用带来了帮助，请考虑引用以下文献：

@article{xia2023sheared,
  title={Sheared llama: Accelerating language model pre-training via structured pruning},
  author={Xia, Mengzhou and Gao, Tianyu and Zeng, Zhiyuan and Chen, Danqi},
  journal={arXiv preprint arXiv:2310.06694},
  year={2023}
}