← Search

Heng-Tze Cheng

4 accepted papers

2024

Controlled Decoding from Language Models

ICML 2024poster

KL-regularized reinforcement learning (RL) is a popular alignment framework to control the language model responses towards high reward outcomes. We pose a tokenwise RL objective and propose a modular solver for it, called *controlled decoding (CD)*. CD exerts control through a separate *prefix scor…

Cited by 86SourcePDFScholar
2024

SELF-DISCOVER: Large Language Models Self-Compose Reasoning Structures

NeurIPS 2024poster

We introduce SELF-DISCOVER, a general framework for LLMs to self-discover the task-intrinsic reasoning structures to tackle complex reasoning problems that are challenging for typical prompting methods. Core to the framework is a self-discovery process where LLMs select multiple atomic reasoning mod…

Cited by 47SourcePDFScholar
2024

Take a Step Back: Evoking Reasoning via Abstraction in Large Language Models

ICLR 2024poster

We present STEP-BACK PROMPTING, a simple prompting technique that enables LLMs to do abstractions to derive high-level concepts and first principles from instances containing specific details. Using the concepts and principles to guide reasoning, LLMs significantly improve their abilities in followi…

Cited by 157SourcePDFScholar
2022

HyperPrompt: Prompt-based Task-Conditioning of Transformers

ICML 2022spotlight

Prompt-Tuning is a new paradigm for finetuning pre-trained language models in a parameter efficient way. Here, we explore the use of HyperNetworks to generate hyper-prompts: we propose HyperPrompt, a novel architecture for prompt-based task-conditioning of self-attention in Transformers. The hyper-p…