--- library_name: transformers tags: - text-generation - pytorch - Lynx - Patronus AI - evaluation - hallucination-detection license: cc-by-nc-4.0 language: - en --- # Model Card for Model ID Lynx is an open-source hallucination evaluation model. Patronus-Lynx-70B-Instruct was trained on a mix of datasets including CovidQA, PubmedQA, DROP, RAGTruth. The datasets contain a mix of hand-annotated and synthetic data. The maximum sequence length is 8000 tokens. ## Model Details - **Model Type:** This is a quantized version of Patronus-Lynx-70B-Instruct. - **Language:** Primarily English - **Developed by:** Patronus AI - **Paper:** [https://arxiv.org/abs/2407.08488](https://arxiv.org/abs/2407.08488) - **License:** [ https://creativecommons.org/licenses/by-nc/4.0/](https://creativecommons.org/licenses/by-nc/4.0/) ### Model Sources - **Repository:** [https://github.com/patronus-ai/Lynx-hallucination-detection](https://github.com/patronus-ai/Lynx-hallucination-detection) ## How to Get Started with the Model The model is fine-tuned to be used to detect hallucinations in a RAG setting. Provided a document, question and answer, the model can evaluate whether the answer is faithful to the document. To use the model, we recommend using the prompt we used for fine-tuning: ``` PROMPT = """ Given the following QUESTION, DOCUMENT and ANSWER you must analyze the provided answer and determine whether it is faithful to the contents of the DOCUMENT. The ANSWER must not offer new information beyond the context provided in the DOCUMENT. The ANSWER also must not contradict information provided in the DOCUMENT. Output your final verdict by strictly following this format: "PASS" if the answer is faithful to the DOCUMENT and "FAIL" if the answer is not faithful to the DOCUMENT. Show your reasoning. -- QUESTION (THIS DOES NOT COUNT AS BACKGROUND INFORMATION): {question} -- DOCUMENT: {context} -- ANSWER: {answer} -- Your output should be in JSON FORMAT with the keys "REASONING" and "SCORE": {{"REASONING": , "SCORE": }} """ ``` The model will output the score as 'PASS' if the answer is faithful to the document or FAIL if the answer is not faithful to the document. ## Training Details The model was finetuned for 3 epochs using 32 H100s on dataset of size 2400. We use [lion](https://github.com/lucidrains/lion-pytorch) optimizer with lr=5.0e-7. For more details on data generation, please check out our Github repo. ### Training Data We train on 2400 samples consisting of CovidQA, PubmedQA, DROP and RAGTruth samples. For datasets that do not contain hallucinated samples, we generate perturbations to introduce hallucinations in the data. For more details about the data generation process, refer to the paper. ## Evaluation The model was evaluated on [PatronusAI/HaluBench](https://huggingface.co/datasets/PatronusAI/HaluBench). It outperforms GPT-3.5-Turbo, GPT-4-Turbo, GPT-4o and Claude-3-Sonnet. ## Citation If you are using the model, cite using ``` @article{ravi2024lynx, title={Lynx: An Open Source Hallucination Evaluation Model}, author={Ravi, Selvan Sunitha and Mielczarek, Bartosz and Kannappan, Anand and Kiela, Douwe and Qian, Rebecca}, journal={arXiv preprint arXiv:2407.08488}, year={2024} } ``` ## Model Card Contact [@sunitha-ravi](https://huggingface.co/sunitha-ravi)