transformers pytorch llama text-generation en dataset:yutaozhu94/INTERS arxiv:2401.06532 license:cc-by-sa-4.0 autotrain_compatible endpoints_compatible text-generation-inference region:us
---
license: cc-by-sa-4.0
datasets:
- yutaozhu94/INTERS
language:
- en
pipeline_tag: text-generation
---

This is the model trained by [INTERS](https://huggingface.co/datasets/yutaozhu94/INTERS). The backbone model is [LLaMA-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf).

## INTERS: Unlocking the Power of Large Language Models in Search with Instruction Tuning
**Authors**: Yutao Zhu, Peitian Zhang, Chenghao Zhang, Yifei Chen, Binyu Xie, Zhicheng Dou, Zheng Liu, and Ji-Rong Wen

<p>
📃 <a href="https://arxiv.org/abs/2401.06532">ArXiv Paper</a>
  •
💡 <a href="https://github.com/DaoD/INTERS">GitHub</a>
</p>

## Introduction
Large language models (LLMs) have demonstrated impressive capabilities in various natural language processing tasks. Despite this, their application to information retrieval (IR) tasks is still challenging due to the infrequent occurrence of many IR-specific concepts in natural language. While prompt-based methods can provide task descriptions to LLMs, they often fall short in facilitating a comprehensive understanding and execution of IR tasks, thereby limiting LLMs' applicability. To address this gap, in this work, we explore the potential of instruction tuning to enhance LLMs' proficiency in IR tasks. We introduce a novel instruction tuning dataset, INTERS, encompassing 20 tasks across three fundamental IR categories: query understanding, document understanding, and query-document relationship understanding. The data are derived from 43 distinct datasets with manually written templates. Our empirical results reveal that INTERS significantly boosts the performance of various publicly available LLMs, such as LLaMA, Mistral, and Phi, in IR tasks. Furthermore, we conduct extensive experiments to analyze the effects of instruction design, template diversity, few-shot demonstrations, and the volume of instructions on performance.

## Prompt
If you want to test the model on some IR tasks, please refer to [link](https://github.com/DaoD/INTERS/blob/main/instruct_templates.py) for writing prompts.