GReaTer: Gradients over Reasoning Makes Smaller Language Models Strong Prompt Optimizers

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Das, Sarkar Snigdha Sarathi, Kamoi, Ryo, Pang, Bo, Zhang, Yusen, Xiong, Caiming, Zhang, Rui
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913779601637376
author Das, Sarkar Snigdha Sarathi
Kamoi, Ryo
Pang, Bo
Zhang, Yusen
Xiong, Caiming
Zhang, Rui
author_facet Das, Sarkar Snigdha Sarathi
Kamoi, Ryo
Pang, Bo
Zhang, Yusen
Xiong, Caiming
Zhang, Rui
contents The effectiveness of large language models (LLMs) is closely tied to the design of prompts, making prompt optimization essential for enhancing their performance across a wide range of tasks. Many existing approaches to automating prompt engineering rely exclusively on textual feedback, refining prompts based solely on inference errors identified by large, computationally expensive LLMs. Unfortunately, smaller models struggle to generate high-quality feedback, resulting in complete dependence on large LLM judgment. Moreover, these methods fail to leverage more direct and finer-grained information, such as gradients, due to operating purely in text space. To this end, we introduce GReaTer, a novel prompt optimization technique that directly incorporates gradient information over task-specific reasoning. By utilizing task loss gradients, GReaTer enables self-optimization of prompts for open-source, lightweight language models without the need for costly closed-source LLMs. This allows high-performance prompt optimization without dependence on massive LLMs, closing the gap between smaller models and the sophisticated reasoning often needed for prompt refinement. Extensive evaluations across diverse reasoning tasks including BBH, GSM8k, and FOLIO demonstrate that GReaTer consistently outperforms previous state-of-the-art prompt optimization methods, even those reliant on powerful LLMs. Additionally, GReaTer-optimized prompts frequently exhibit better transferability and, in some cases, boost task performance to levels comparable to or surpassing those achieved by larger language models, highlighting the effectiveness of prompt optimization guided by gradients over reasoning. Code of GReaTer is available at https://github.com/psunlpgroup/GreaTer.
format Preprint
id arxiv_https___arxiv_org_abs_2412_09722
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GReaTer: Gradients over Reasoning Makes Smaller Language Models Strong Prompt Optimizers
Das, Sarkar Snigdha Sarathi
Kamoi, Ryo
Pang, Bo
Zhang, Yusen
Xiong, Caiming
Zhang, Rui
Computation and Language
The effectiveness of large language models (LLMs) is closely tied to the design of prompts, making prompt optimization essential for enhancing their performance across a wide range of tasks. Many existing approaches to automating prompt engineering rely exclusively on textual feedback, refining prompts based solely on inference errors identified by large, computationally expensive LLMs. Unfortunately, smaller models struggle to generate high-quality feedback, resulting in complete dependence on large LLM judgment. Moreover, these methods fail to leverage more direct and finer-grained information, such as gradients, due to operating purely in text space. To this end, we introduce GReaTer, a novel prompt optimization technique that directly incorporates gradient information over task-specific reasoning. By utilizing task loss gradients, GReaTer enables self-optimization of prompts for open-source, lightweight language models without the need for costly closed-source LLMs. This allows high-performance prompt optimization without dependence on massive LLMs, closing the gap between smaller models and the sophisticated reasoning often needed for prompt refinement. Extensive evaluations across diverse reasoning tasks including BBH, GSM8k, and FOLIO demonstrate that GReaTer consistently outperforms previous state-of-the-art prompt optimization methods, even those reliant on powerful LLMs. Additionally, GReaTer-optimized prompts frequently exhibit better transferability and, in some cases, boost task performance to levels comparable to or surpassing those achieved by larger language models, highlighting the effectiveness of prompt optimization guided by gradients over reasoning. Code of GReaTer is available at https://github.com/psunlpgroup/GreaTer.
title GReaTer: Gradients over Reasoning Makes Smaller Language Models Strong Prompt Optimizers
topic Computation and Language
url https://arxiv.org/abs/2412.09722