← Search

Wolfgang Lehrach

6 accepted papers

2026

Code World Models for General Game Playing

ICLR 2026poster

Large Language Models (LLMs) reasoning abilities are increasingly being applied to classical board and card games, but the dominant approach---involving prompting for direct move generation---has significant drawbacks. It relies on the model's implicit fragile pattern-matching capabilities, leading…

Cited by 0SourceScholar
2025

Improving Transformer World Models for Data-Efficient RL

ICML 2025poster

We present an approach to model-based RL that achieves a new state of the art performance on the challenging Craftax-classic benchmark, an open-world 2D survival game that requires agents to exhibit a wide range of general abilities---such as strong generalization, deep exploration, and long-term re…

Cited by 0SourcePDFScholar
2024

DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractors

NeurIPS 2024poster

Learning from previously collected data via behavioral cloning or offline reinforcement learning (RL) is a powerful recipe for scaling generalist agents by avoiding the need for expensive online learning. Despite strong generalization in some respects, agents are often remarkably brittle to minor vi…

2024

Learning Cognitive Maps from Transformer Representations for Efficient Planning in Partially Observed Environments

ICML 2024poster

Despite their stellar performance on a wide range of tasks, including in-context tasks only revealed during inference, vanilla transformers and variants trained for next-token predictions (a) do not learn an explicit world model of their environment which can be flexibly queried and (b) cannot be us…

Cited by 2SourcePDFScholar
2021

Query Training: Learning a Worse Model to Infer Better Marginals in Undirected Graphical Models with Hidden Variables

AAAI 2021technical

Probabilistic graphical models (PGMs) provide a compact representation of knowledge that can be queried in a flexible way: after learning the parameters of a graphical model once, new probabilistic queries can be answered at test time without retraining. However, when using undirected PGMS with hidd…

2016

Generative Shape Models: Joint Text Recognition and Segmentation with Very Little Training Data

NeurIPS 2016poster

We demonstrate that a generative model for object shapes can achieve state of the art results on challenging scene text recognition tasks, and with orders of magnitude fewer training images than required for competing discriminative methods. In addition to transcribing text from challenging images,…

Cited by 13SourcePDFScholar