← Search

Shixiang Shane Gu

16 accepted papers

2024

Geometric-Averaged Preference Optimization for Soft Preference Labels

NeurIPS 2024poster

Many algorithms for aligning LLMs with human preferences assume that human preferences are binary and deterministic. However, human preferences can vary across individuals, and therefore should be represented distributionally. In this work, we introduce the distributional soft preference labels and…

Cited by 7SourcePDFScholar
2024

Multimodal Web Navigation with Instruction-Finetuned Foundation Models

ICLR 2024poster

The progress of autonomous web navigation has been hindered by the dependence on billions of exploratory interactions via online reinforcement learning, and domain-specific model designs that make it difficult to leverage generalization from rich out-of-domain data. In this work, we study data-drive…

Cited by 111SourcePDFScholar
2023

A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation

ICLR 2023top-25%

The rise of generalist large-scale models in natural language and vision has made us expect that a massive data-driven approach could achieve broader generalization in other domains such as continuous control. In this work, we explore a method for learning a single policy that manipulates various fo…

2023

Collective Intelligence for 2D Push Manipulations With Mobile Robots

RA-L 2023

While natural systems often present collective intelligence that allows them to self-organize and adapt to changes, the equivalent is missing in most artificial systems. We explore the possibility of such a system in the context of cooperative 2D push manipulations using mobile robots. Although conv

Cited by 5SourcecodeScholar
2023

DreamSparse: Escaping from Plato’s Cave with 2D Diffusion Model Given Sparse Views

NeurIPS 2023poster

Synthesizing novel view images from a few views is a challenging but practical problem. Existing methods often struggle with producing high-quality results or necessitate per-object optimization in such few-view settings due to the insufficient information provided. In this work, we explore leverag…

Cited by 13SourcePDFScholar
2023

For SALE: State-Action Representation Learning for Deep Reinforcement Learning

NeurIPS 2023poster

In reinforcement learning (RL), representation learning is a proven tool for complex image-based tasks, but is often overlooked for environments with low-level states, such as physical control problems. This paper introduces SALE, a novel approach for learning embeddings that model the nuanced inte…

2023

Large Language Models Can Self-Improve

EMNLP 2023long main

Large Language Models (LLMs) have achieved excellent performances in various tasks. However, fine-tuning an LLM requires extensive supervision. Human, on the other hand, may improve their reasoning abilities by self-thinking without external inputs. In this work, we demonstrate that an LLM is also c…

Cited by 0SourceScholar
2023

Mind's Eye: Grounded Language Model Reasoning through Simulation

ICLR 2023poster

Successful and effective communication between humans and AI relies on a shared experience of the world. By training solely on written text, current language models (LMs) miss the grounded experience of humans in the real-world---their failure to relate language to the physical world causes knowledg…

Cited by 84SourcePDFScholar
2022

Blocks Assemble! Learning to Assemble with Large-Scale Structured Reinforcement Learning

ICML 2022spotlight

Assembly of multi-part physical structures is both a valuable end product for autonomous robotics, as well as a valuable diagnostic task for open-ended training of embodied intelligent agents. We introduce a naturalistic physics-based environment with a set of connectable magnet blocks inspired by c…

2022

Generalized Decision Transformer for Offline Hindsight Information Matching

ICLR 2022spotlight

How to extract as much learning signal from each trajectory data has been a key problem in reinforcement learning (RL), where sample inefficiency has posed serious challenges for practical applications. Recent works have shown that using expressive policy function approximators and conditioning on f…

2022

Large Language Models are Zero-Shot Reasoners

NeurIPS 2022accept

Pretrained large language models (LLMs) are widely used in many sub-fields of natural language processing (NLP) and generally known as excellent few-shot learners with task-specific exemplars. Notably, chain of thought (CoT) prompting, a recent technique for eliciting complex multi-step reasoning th…

2022

Why Should I Trust You, Bellman? The Bellman Error is a Poor Replacement for Value Error

ICML 2022spotlight

In this work, we study the use of the Bellman equation as a surrogate objective for value prediction accuracy. While the Bellman equation is uniquely solved by the true value function over all state-action pairs, we find that the Bellman error (the difference between both sides of the equation) is a…

Cited by 41SourcePDFScholar
2022

Why So Pessimistic? Estimating Uncertainties for Offline RL through Ensembles, and Why Their Independence Matters

NeurIPS 2022accept

Motivated by the success of ensembles for uncertainty estimation in supervised learning, we take a renewed look at how ensembles of $Q$-functions can be leveraged as the primary source of pessimism for offline reinforcement learning (RL). We begin by identifying a critical flaw in a popular algorith…

2021

EMaQ: Expected-Max Q-Learning Operator for Simple Yet Effective Offline and Online RL

ICML 2021spotlight

Off-policy reinforcement learning (RL) holds the promise of sample-efficient learning of decision-making policies by leveraging past experience. However, in the offline RL setting – where a fixed collection of interactions are provided and no further interactions are allowed – it has been shown that…

Cited by 144SourcePDFScholar
2021

Policy Information Capacity: Information-Theoretic Measure for Task Complexity in Deep Reinforcement Learning

ICML 2021spotlight

Progress in deep reinforcement learning (RL) research is largely enabled by benchmark task environments. However, analyzing the nature of those environments is often overlooked. In particular, we still do not have agreeable ways to measure the difficulty or solvability of a task, given that each has…

2021

Variational Empowerment as Representation Learning for Goal-Conditioned Reinforcement Learning

ICML 2021spotlight

Learning to reach goal states and learning diverse skills through mutual information maximization have been proposed as principled frameworks for unsupervised reinforcement learning, allowing agents to acquire broadly applicable multi-task policies with minimal reward engineering. In this paper, we…

Cited by 40SourcePDFScholar