← 返回资源分享
NEFTune: Noisy Embeddings Improve Instruction Finetuning
论文
论文
发布时间2023-10-09
发表arXiv:2310.05914
作者:Bhavya Kailkhura,Tom Goldstein,Micah Goldblum,Jonas Geiping,John Kirchenbauer,Yuxin Wen,Neel Jain,Aniruddha Saha,Ping-Yeh Chiang,Hong-Min Chu,Gowthami Somepalli,Brian R. Bartoldson,Avi Schwarzschild
详细介绍
We show that language model finetuning can be improved, sometimes dramatically, with a simple augmentation. NEFTune adds noise to the embedding vectors during training. Standard finetuning of LLaMA-2-7B using Alpaca achieves 29.79% on AlpacaEval, which rises to 64.69% using noisy embeddings. NEFTune also improves over strong baselines on modern instruction datasets. Models trained with Evol-Instruct see a 10% improvement, with ShareGPT an 8% improvement, and with OpenPlatypus an 8% improvement. Even powerful models further refined with RLHF such as LLaMA-2-Chat benefit from additional training with NEFTune.
代码仓库 (4)
rijgersberg/geitje官方PyTorch
neelsjain/neftune官方PyTorch
openaccess-ai-collective/axolotlPyTorch
akjindal53244/arithmoPyTorch
