RAVEN: In-Context Learning with Retrieval-Augmented Encoder-Decoder Language Models
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2023
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866917751595991040 |
|---|---|
| author | Huang, Jie Ping, Wei Xu, Peng Shoeybi, Mohammad Chang, Kevin Chen-Chuan Catanzaro, Bryan |
| author_facet | Huang, Jie Ping, Wei Xu, Peng Shoeybi, Mohammad Chang, Kevin Chen-Chuan Catanzaro, Bryan |
| contents | In this paper, we investigate the in-context learning ability of retrieval-augmented encoder-decoder language models. We first conduct a comprehensive analysis of existing models and identify their limitations in in-context learning, primarily due to a mismatch between pretraining and inference, as well as a restricted context length. To address these issues, we propose RAVEN, a model that combines retrieval-augmented masked language modeling and prefix language modeling. We further introduce Fusion-in-Context Learning to enhance the few-shot performance by enabling the model to leverage more in-context examples without requiring additional training. Through extensive experiments, we demonstrate that our simple yet effective design significantly improves performance, achieving results comparable to the most advanced language models in certain scenarios, despite having substantially fewer parameters. Our work underscores the potential of retrieval-augmented encoder-decoder language models for in-context learning and encourages further research in this direction. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2308_07922 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | RAVEN: In-Context Learning with Retrieval-Augmented Encoder-Decoder Language Models Huang, Jie Ping, Wei Xu, Peng Shoeybi, Mohammad Chang, Kevin Chen-Chuan Catanzaro, Bryan Computation and Language Artificial Intelligence Machine Learning In this paper, we investigate the in-context learning ability of retrieval-augmented encoder-decoder language models. We first conduct a comprehensive analysis of existing models and identify their limitations in in-context learning, primarily due to a mismatch between pretraining and inference, as well as a restricted context length. To address these issues, we propose RAVEN, a model that combines retrieval-augmented masked language modeling and prefix language modeling. We further introduce Fusion-in-Context Learning to enhance the few-shot performance by enabling the model to leverage more in-context examples without requiring additional training. Through extensive experiments, we demonstrate that our simple yet effective design significantly improves performance, achieving results comparable to the most advanced language models in certain scenarios, despite having substantially fewer parameters. Our work underscores the potential of retrieval-augmented encoder-decoder language models for in-context learning and encourages further research in this direction. |
| title | RAVEN: In-Context Learning with Retrieval-Augmented Encoder-Decoder Language Models |
| topic | Computation and Language Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2308.07922 |