Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Code and documentation to train Stanford's Alpaca models, and generate the data.

License

NotificationsYou must be signed in to change notification settings

snapshot-repo/stanford_alpaca

 
 

Repository files navigation

Stanford-Alpaca

Stanford Alpaca: An Instruction-following LLaMA model

LicensePython 3.8+Code style: black

This is the repo for the Stanford Alpaca project, which aims to build and share an instruction-following LLaMA model. The repo contains:

Overview

The current Alpaca model is fine-tuned from a 7B LLaMA model [1] on 52K instruction-following data generated by the techniques in the Self-Instruct [2] paper, with some modifications that we discuss in the next section.In a preliminary human evaluation, we found that the Alpaca 7B model behaves similarly to thetext-davinci-003 model on the Self-Instruct instruction-following evaluation suite [2].

Alpaca is still under development, and there are many limitations that have to be addressed.Importantly, we have not yet fine-tuned the Alpaca model to be safe and harmless.We thus encourage users to be cautious when interacting with Alpaca, and to report any concerning behavior to help improve the safety and ethical considerations of the model.

Our initial release contains the data generation procedure, dataset, and training recipe. We intend to release the model weights if we are given permission to do so by the creators of LLaMA. For now, we have chosen to host a live demo to help readers better understand the capabilities and limits of Alpaca, as well as a way to help us better evaluate Alpaca's performance on a broader audience.

Please read our releaseblog post for more details about the model, our discussion of the potential harm and limitations of Alpaca models, and our thought process of an open-source release.

[1]: LLaMA: Open and Efficient Foundation Language Models. Hugo Touvron, Thibaut Lavril, Gautier Izacard, Xavier Martinet, Marie-Anne Lachaux, Timothée Lacroix, Baptiste Rozière, Naman Goyal, Eric Hambro, Faisal Azhar, Aurelien Rodriguez, Armand Joulin, Edouard Grave, Guillaume Lample.https://arxiv.org/abs/2302.13971v1

[2]: Self-Instruct: Aligning Language Model with Self Generated Instructions. Yizhong Wang, Yeganeh Kordi, Swaroop Mishra, Alisa Liu, Noah A. Smith, Daniel Khashabi, Hannaneh Hajishirzi.https://arxiv.org/abs/2212.10560

Data Release

alpaca_data.json contains 52K instruction-following data we used for fine-tuning the Alpaca model.This JSON file is a list of dictionaries, each dictionary contains the following fields:

  • instruction:str, describes the task the model should perform. Each of the 52K instructions is unique.
  • input:str, optional context or input for the task. For example, when the instruction is "Summarize the following article", the input is the article. Around 40% of the examples have an input.
  • output:str, the answer to the instruction as generated bytext-davinci-003.

We used the following prompts for fine-tuning the Alpaca model:

  • for examples with a non-empty input field:
Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.### Instruction:{instruction}### Input:{input}### Response:
  • for examples with an empty input field:
Below is an instruction that describes a task. Write a response that appropriately completes the request.### Instruction:{instruction}### Response:

Data Generation Process

Running the code
  1. Set environment variablesOPENAI_API_KEY to your OpenAI API key.
  2. Install the dependencies withpip install -r requirements.txt.
  3. Runpython -m generate_instruction generate_instruction_following_data to generate the data.

We built on the data generation pipeline fromself-instruct and made the following modifications:

  • We usedtext-davinci-003 to generate the instruction data instead ofdavinci.
  • We wrote a new prompt (prompt.txt) that explicitly gave the requirement of instruction generation totext-davinci-003.
  • We adopted much more aggressive batch decoding, i.e., generating 20 instructions at once, which significantly reduced the cost of data generation.
  • We simplified the data generation pipeline by discarding the difference between classification and non-classification instructions.
  • We only generated a single instance for each instruction, instead of 2 to 3 instances as in [1].

This produced an instruction-following dataset with 52K examples obtained at a much lower cost (less than $500).In a preliminary study, we also find our 52K generated data to be much more diverse than the data released byself-instruct.We plot the below figure (in the style of Figure 2 in theself-instruct paper to demonstrate the diversity of our data.The inner circle of the plot represents the root verb of the instructions, and the outer circle represents the direct objects.

Fine-tuning

We fine-tune our model using standard huggingface training code with the following hyperparameters:

HyperparameterValue
Batch size128
Learning rate2e-5
Epochs3
Max length512
Weight decay1

We are waiting for huggingface to officially support the llama models (i.e. thisPR to be merged) before we release a stable version of the finetuning code.

Authors

All grad students below contributed equally and the order is determined by random draw.

All advised byTatsunori B. Hashimoto. Yann is also advised byPercy Liang and Xuechen is also advised byCarlos Guestrin.

Citation

Please cite the repo if you use the data or code in this repo.

@misc{alpaca,  author = {Rohan Taori and Ishaan Gulrajani and Tianyi Zhang and Yann Dubois and Xuechen Li and Carlos Guestrin and Percy Liang and Tatsunori B. Hashimoto },  title = {Stanford Alpaca: An Instruction-following LLaMA model},  year = {2023},  publisher = {GitHub},  journal = {GitHub repository},  howpublished = {\url{https://github.com/tatsu-lab/stanford_alpaca}},}

Naturally, you should also cite the original LLaMA paper [1] and the Self-Instruct paper [2].

Acknowledgements

We thank Yizhong Wang for his help in explaining the data generation pipeline in Self-Instruct and providing the code for the parse analysis plot.

About

Code and documentation to train Stanford's Alpaca models, and generate the data.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python100.0%

[8]ページ先頭

©2009-2025 Movatter.jp