Self-Refinement of Language Models from External Proxy Metrics Feedback

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ramji, Keshav, Lee, Young-Suk, Astudillo, Ramón Fernandez, Sultan, Md Arafat, Naseem, Tahira, Munawar, Asim, Florian, Radu, Roukos, Salim
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913251491577856
author Ramji, Keshav
Lee, Young-Suk
Astudillo, Ramón Fernandez
Sultan, Md Arafat
Naseem, Tahira
Munawar, Asim
Florian, Radu
Roukos, Salim
author_facet Ramji, Keshav
Lee, Young-Suk
Astudillo, Ramón Fernandez
Sultan, Md Arafat
Naseem, Tahira
Munawar, Asim
Florian, Radu
Roukos, Salim
contents It is often desirable for Large Language Models (LLMs) to capture multiple objectives when providing a response. In document-grounded response generation, for example, agent responses are expected to be relevant to a user's query while also being grounded in a given document. In this paper, we introduce Proxy Metric-based Self-Refinement (ProMiSe), which enables an LLM to refine its own initial response along key dimensions of quality guided by external metrics feedback, yielding an overall better final response. ProMiSe leverages feedback on response quality through principle-specific proxy metrics, and iteratively refines its response one principle at a time. We apply ProMiSe to open source language models Flan-T5-XXL and Llama-2-13B-Chat, to evaluate its performance on document-grounded question answering datasets, MultiDoc2Dial and QuAC, demonstrating that self-refinement improves response quality. We further show that fine-tuning Llama-2-13B-Chat on the synthetic dialogue data generated by ProMiSe yields significant performance improvements over the zero-shot baseline as well as a supervised fine-tuned model on human annotated data.
format Preprint
id arxiv_https___arxiv_org_abs_2403_00827
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Refinement of Language Models from External Proxy Metrics Feedback
Ramji, Keshav
Lee, Young-Suk
Astudillo, Ramón Fernandez
Sultan, Md Arafat
Naseem, Tahira
Munawar, Asim
Florian, Radu
Roukos, Salim
Computation and Language
Artificial Intelligence
Machine Learning
It is often desirable for Large Language Models (LLMs) to capture multiple objectives when providing a response. In document-grounded response generation, for example, agent responses are expected to be relevant to a user's query while also being grounded in a given document. In this paper, we introduce Proxy Metric-based Self-Refinement (ProMiSe), which enables an LLM to refine its own initial response along key dimensions of quality guided by external metrics feedback, yielding an overall better final response. ProMiSe leverages feedback on response quality through principle-specific proxy metrics, and iteratively refines its response one principle at a time. We apply ProMiSe to open source language models Flan-T5-XXL and Llama-2-13B-Chat, to evaluate its performance on document-grounded question answering datasets, MultiDoc2Dial and QuAC, demonstrating that self-refinement improves response quality. We further show that fine-tuning Llama-2-13B-Chat on the synthetic dialogue data generated by ProMiSe yields significant performance improvements over the zero-shot baseline as well as a supervised fine-tuned model on human annotated data.
title Self-Refinement of Language Models from External Proxy Metrics Feedback
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2403.00827