← Search

Jane X Wang

7 accepted papers

2024

CogBench: a large language model walks into a psychology lab

ICML 2024poster

Large language models (LLMs) have significantly advanced the field of artificial intelligence. Yet, evaluating them comprehensively remains challenging. We argue that this is partly due to the predominant focus on performance metrics in most benchmarks. This paper introduces *CogBench*, a benchmark…

2023

Learning to Induce Causal Structure

ICLR 2023poster

The fundamental challenge in causal induction is to infer the underlying graph structure given observational and/or interventional data. Most existing causal induction algorithms operate by generating candidate graphs and evaluating them using either score-based methods (including continuous optimiz…

Cited by 59SourcePDFScholar
2023

Meta-in-context learning in large language models

NeurIPS 2023poster

Large language models have shown tremendous performance in a variety of tasks. In-context learning -- the ability to improve at a task after being provided with a number of demonstrations -- is seen as one of the main contributors to their success. In the present paper, we demonstrate that the in-…

2023

Passive learning of active causal strategies in agents and language models

NeurIPS 2023poster

What can be learned about causality and experimentation from passive data? This question is salient given recent successes of passively-trained language models in interactive domains such as tool use. Passive learning is inherently limited. However, we show that purely passive learning can in fact a…

Cited by 23SourcePDFScholar
2022

Data Distributional Properties Drive Emergent In-Context Learning in Transformers

NeurIPS 2022accept

Large transformer-based models are able to perform in-context few-shot learning, without being explicitly trained for it. This observation raises the question: what aspects of the training regime lead to this emergent behavior? Here, we show that this behavior is driven by the distributions of the t…

2022

Semantic Exploration from Language Abstractions and Pretrained Representations

NeurIPS 2022accept

Effective exploration is a challenge in reinforcement learning (RL). Novelty-based exploration methods can suffer in high-dimensional state spaces, such as continuous partially-observable 3D environments. We address this challenge by defining novelty using semantically meaningful state abstractions,…

Cited by 72SourcePDFScholar
2021

Alchemy: A benchmark and analysis toolkit for meta-reinforcement learning agents

NeurIPS 2021poster

There has been rapidly growing interest in meta-learning as a method for increasing the flexibility and sample efficiency of reinforcement learning. One problem in this area of research, however, has been a scarcity of adequate benchmark tasks. In general, the structure underlying past benchmarks ha…

Cited by 38SourcecodeScholar