promptbench

[![Contributors][contributors-shield]][contributors-url] [![Forks][forks-shield]][forks-url] [![Stargazers][stars-shield]][stars-url] [![Issues][issues-shield]][issues-url]
Table of Contents

1. News and Updates
2. Introduction
3. Installation
4. Usage
5. Datasets and Models
6. Benchmark Results
7. Acknowledgments

## News and Updates - [19/08/2024] Add [DyVal 2 (ICML 2024)](https://openreview.net/pdf?id=DwTgy1hXXo). - [19/08/2024] Merge [PromptEval](https://github.com/felipemaiapolo/prompteval), an efficient multi-prompt evaluation method, into this repository. - [26/05/2024] Add support for GPT-4o. - [13/03/2024] Add support for multi-modal models and datasets. - [05/01/2024] Add support for BigBench Hard, DROP, ARC datasets. - [16/12/2023] Add support for Gemini, Mistral, Mixtral, Baichuan, Yi models. - [15/12/2023] Add detailed instructions for users to add new modules (models, datasets, etc.) [examples/add_new_modules.md](examples/add_new_modules.md). - [05/12/2023] Published promptbench 0.0.1. ## Introduction **PromptBench** is a Pytorch-based Python package for Evaluation of Large Language Models (LLMs). It provides user-friendly APIs for researchers to conduct evaluation on LLMs. Check the technical report: https://arxiv.org/abs/2312.07910. ### What does promptbench currently provide? 1. **Quick model performance assessment:** We offer a user-friendly interface that allows for quick model building, dataset loading, and evaluation of model performance. 2. **Prompt Engineering:** We implemented several prompt engineering methods. For example: [Few-shot Chain-of-Thought](https://arxiv.org/abs/2201.11903) [1], [Emotion Prompt](https://arxiv.org/abs/2307.11760) [2], [Expert Prompting](https://arxiv.org/abs/2305.14688) [3] and so on. 3. **Evaluating adversarial prompts:** promptbench integrated [prompt attacks](https://arxiv.org/abs/2306.04528) [4], enabling researchers to simulate black-box adversarial prompt attacks on models and evaluate their robustness (see details [here](promptbench/prompt_attack/README.md)). 4. **Dynamic evaluation to mitigate potential test data contamination:** we integrated the dynamic evaluation framework [DyVal](https://arxiv.org/pdf/2309.17167) [5], which generates evaluation samples on-the-fly with controlled complexity. 5. **Efficient multi-prompt evaluation**: We integrated the efficient multi-prompt evaluation method [PromptEval](https://arxiv.org/abs/2405.17202) [8]. This method uses the performance of LLMs on a small amount of data to build an IRT-like model. This model is then used to predict the performance of LLMs on unseen data. Tests on MMLU, BBH, and LMentry show that this method requires sampling only 5% of the data to reduce the error between estimated and actual performance to around 2%. ## Installation ### Install via `pip` We provide a Python package *promptbench* for users who want to start evaluation quickly. Simply run: ```sh pip install promptbench ``` Note that the pip installation could be behind the recent updates. So, if you want to use the latest features or develop based on our code, you should install via GitHub. ### Install via GitHub First, clone the repo: ```sh git clone [email protected]:microsoft/promptbench.git ``` Then, ```sh cd promptbench ``` To install the required packages, you can create a conda environment: ```sh conda create --name promptbench python=3.9 conda activate promptbench ``` then use pip to install required packages: ```sh pip install -r requirements.txt ``` Note that this only installed basic python packages. For Prompt Attacks, you will also need to install [TextAttack](https://github.com/QData/TextAttack). ## Usage promptbench is easy to use and extend. Going through the examples below will help you get familiar with promptbench for quick use, evaluate existing datasets and LLMs, or create your own datasets and models. Please see [Installation](#installation) to install promptbench first. If promptbench is installed via `pip`, you can simply do: ```python import promptbench as pb ``` If you installed promptbench from `git` and want to use it in other projects: ```python import sys # Add the directory of promptbench to the Python path sys.path.append('/home/xxx/promptbench') # Now you can import promptbench by name import promptbench as pb ``` We provide tutorials for: 1. **evaluate models on existing benchmarks:** please refer to the [examples/basic.ipynb](examples/basic.ipynb) for constructing your evaluation pipeline. For a multi-modal evaluation pipeline, please refer to [examples/multimodal.ipynb](examples/multimodal.ipynb) 2. **test the effects of different prompting techniques:** 3. **examine the robustness for prompt attacks**, please refer to [examples/prompt_attack.ipynb](examples/prompt_attack.ipynb) to construct the attacks. 4. **use DyVal for evaluation:** please refer to [examples/dyval.ipynb](examples/dyval.ipynb) to construct DyVal datasets. 5. **efficient multi-prompt evaluation using PromptEval**: please refer to [examples/efficient_multi_prompt_eval.ipynb](examples/efficient_multi_prompt_eval.ipynb) ## Implemented Components PromptBench currently supports different datasets, models, prompt engineering methods, adversarial attacks, and more. You are welcome to add more. ### Datasets - Language datasets: - GLUE: SST-2, CoLA, QQP, MRPC, MNLI, QNLI, RTE, WNLI - MMLU - BIG-Bench Hard (Bool logic, valid parentheses, date...) - Math - GSM8K - SQuAD V2 - IWSLT 2017 - UN Multi - CSQA (CommonSense QA) - Numersense - QASC - Last Letter Concatenate - Multi-modal datasets: - VQAv2 - NoCaps - MMMU - MathVista - AI2D - ChartQA - ScienceQA ### Models Language models: - Open-source models: - google/flan-t5-large - databricks/dolly-v1-6b - Llama2 series - vicuna-13b, vicuna-13b-v1.3 - Cerebras/Cerebras-GPT-13B - EleutherAI/gpt-neox-20b - Google/flan-ul2 - phi-1.5 and phi-2 - Proprietary models - PaLM 2 - GPT-3.5 - GPT-4 - Gemini Pro Multi-modal models: - Open-source models: - BLIP2 - LLaVA - Qwen-VL, Qwen-VL-Chat - InternLM-XComposer2-VL - Proprietary models - GPT-4v - Gemini Pro Vision - Qwen-VL-Max, Qwen-VL-Plus ### Prompt Engineering - Chain-of-thought (COT) [1] - EmotionPrompt [2] - Expert prompting [3] - Zero-shot chain-of-thought - Generated knowledge [6] - Least to most [7] ### Adversarial Attacks - Character-level attack - DeepWordBug - TextBugger - Word-level attack - TextFooler - BertAttack - Sentence-level attack - CheckList - StressTest - Semantic-level attack - Human-crafted attack ### Protocols and Analysis - Standard evaluation - Dynamic evaluation - Semantic evaluation - Benchmark results - Visualization analysis - Transferability analysis - Word frequency analysis ## Benchmark Results Please refer to our [benchmark website](https://llm-eval.github.io/) for benchmark results on Prompt Attacks, Prompt Engineering and Dynamic Evaluation DyVal. ## Acknowledgements - [TextAttack](https://github.com/QData/TextAttack) - [README Template](https://github.com/othneildrew/Best-README-Template) - We thank the volunteers: Hanyuan Zhang, Lingrui Li, Yating Zhou for conducting the semantic preserving experiment in Prompt Attack benchmark. ## Reference [1] Jason Wei, et al. "Chain-of-Thought Prompting Elicits Reasoning in Large Language Models." arXiv preprint arXiv:2201.11903 (2022). [2] Cheng Li, et al. "Emotionprompt: Leveraging psychology for large language models enhancement via emotional stimulus." arXiv preprint arXiv:2307.11760 (2023). [3] BenFeng Xu, et al. "ExpertPrompting: Instructing Large Language Models to be Distinguished Experts" arXiv preprint arXiv:2305.14688 (2023). [4] Zhu, Kaijie, et al. "PromptBench: Towards Evaluating the Robustness of Large Language Models on Adversarial Prompts." arXiv preprint arXiv:2306.04528 (2023). [5] Zhu, Kaijie, et al. "DyVal: Graph-informed Dynamic Evaluation of Large Language Models." arXiv preprint arXiv:2309.17167 (2023). [6] Liu J, Liu A, Lu X, et al. Generated knowledge prompting for commonsense reasoning[J]. arXiv preprint arXiv:2110.08387, 2021. [7] Zhou D, Schärli N, Hou L, et al. Least-to-most prompting enables complex reasoning in large language models[J]. arXiv preprint arXiv:2205.10625, 2022. [8] Felipe Maia Polo, et al. "Prompteval: Efficient Multi-prompt Evaluation of Language Models." arXiv preprint arXiv:2405.17202. ## Citing promptbench and other research papers Please cite us if you find this project helpful for your project/paper: ``` … ``` ## Contributing This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.opensource.microsoft.com. When you submit a pull request, a CLA bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA. This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/). For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or contact [[email protected]](mailto:[email protected]) with any additional questions or comments. If you have a suggestion that would make promptbench better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again! 1. Fork the project 2. Create your branch (`git checkout -b your_name/your_branch`) 3. Commit your changes (`git commit -m 'Add some features'`) 4. Push to the branch (`git push origin your_name/your_branch`) 5. Open a Pull Request ## Trademarks This project may contain trademarks or logos for projects, products, or services. Authorized use of Microsoft trademarks or logos is subject to and must follow [Microsoft's Trademark & Brand Guidelines](https://www.microsoft.com/en-us/legal/intellectualproperty/trademarks/usage/general). Use of Microsoft trademarks or logos in modified versions of this project must not cause confusion or imply Microsoft sponsorship. Any use of third-party trademarks or logos are subject to those third-party's policies. [contributors-shield]: https://img.shields.io/github/contributors/microsoft/promptbench.svg?style=for-the-badge [contributors-url]: https://github.com/microsoft/promptbench/graphs/contributors [forks-shield]: https://img.shields.io/github/forks/microsoft/promptbench.svg?style=for-the-badge [forks-url]: https://github.com/microsoft/promptbench/network/members [stars-shield]: https://img.shields.io/github/stars/microsoft/promptbench.svg?style=for-the-badge [stars-url]: https://github.com/microsoft/promptbench/stargazers [issues-shield]: https://img.shields.io/github/issues/microsoft/promptbench.svg?style=for-the-badge [issues-url]: https://github.com/microsoft/promptbench/issues [license-shield]: https://img.shields.io/github/license/microsoft/promptbench.svg?style=for-the-badge [license-url]: https://github.com/microsoft/promptbench/blob/main/LICENSE.txt