← Search

Steven Zheng

7 accepted papers

2025

Speculative RAG: Enhancing Retrieval Augmented Generation through Drafting

ICLR 2025poster

Retrieval augmented generation (RAG) combines the generative abilities of large language models (LLMs) with external knowledge sources to provide more accurate and up-to-date responses. Recent RAG advancements focus on improving retrieval outcomes through iterative LLM refinement or self-critique ca…

Cited by 13SourcePDFScholar
2024

In-Context Principle Learning from Mistakes

ICML 2024poster

In-context learning (ICL, also known as few-shot prompting) has been the standard method of adapting LLMs to downstream tasks, by learning from a few input-output examples. Nonetheless, all ICL-based approaches only learn from correct input-output pairs. In this paper, we revisit this paradigm, by l…

Cited by 29SourcePDFScholar
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
2023

Transcending Scaling Laws with 0.1% Extra Compute

EMNLP 2023long main

Scaling language models improves performance but comes with significant computational costs. This paper proposes UL2R, a method that substantially improves existing language models and their scaling curves with a relatively tiny amount of extra compute. The key idea is to continue training a state-o…

Cited by 0SourceScholar
2023

UL2: Unifying Language Learning Paradigms

ICLR 2023poster

Existing pre-trained models are generally geared towards a particular class of problems. To date, there seems to be still no consensus on what the right architecture and pre-training setup should be. This paper presents a unified framework for pre-training models that are universally effective acros…

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…

2017

A cloud robot system using the dexterity network and berkeley robotics and automation as a service (Brass)

ICRA 2017poster

In support of Cloud Robotics, Robotics and Automation as a Service (RAaaS) frameworks have the potential to reduce the complexity of software development, simplify software installation and maintenance, and facilitate data sharing for machine learning. In this proof-of-concept paper, we describe Ber…

Cited by 47SourceScholar