Wolfgang Lehrach

According to our database, Wolfgang Lehrach authored at least 14 papers between 2016 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

AutoHarness: improving LLM agents by automatically synthesizing a code harness.

Xinghua Lou, Miguel Lázaro-Gredilla, Antoine Dedieu, Carter Wendelken, Wolfgang Lehrach, Kevin P. Murphy

CoRR, March, 2026

Joint Learning of Hierarchical Neural Options and Abstract World Model.

Wasu Top Piriyakulkij, Wolfgang Lehrach, Kevin Ellis, Kevin Murphy

CoRR, February, 2026

2025

Code World Models for General Game Playing.

Wolfgang Lehrach, Daniel Hennes, Miguel Lázaro-Gredilla, Xinghua Lou, Carter Wendelken, Zun Li, Antoine Dedieu, Jordi Grau-Moya, Marc Lanctot, Atil Iscen, John Schultz, Marcus Chiam, Ian Gemp, Piotr Zielinski, Satinder Singh, Kevin P. Murphy

CoRR, October, 2025

Diffusion Model Predictive Control.

Guangyao Zhou, Sivaramakrishnan Swaminathan, Rajkumar Vasudeva Raju, J. Swaroop Guntupalli, Wolfgang Lehrach, Joseph Ortiz, Antoine Dedieu, Miguel Lázaro-Gredilla, Kevin Patrick Murphy

Trans. Mach. Learn. Res., 2025

Improving Transformer World Models for Data-Efficient RL.

Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, J. Swaroop Guntupalli, Wolfgang Lehrach, Miguel Lázaro-Gredilla, Kevin Patrick Murphy

Proceedings of the Forty-second International Conference on Machine Learning, 2025

2024

PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX.

Guangyao Zhou, Antoine Dedieu, Nishanth Kumar, Wolfgang Lehrach, Shrinu Kushagra, Dileep George, Miguel Lázaro-Gredilla

J. Mach. Learn. Res., 2024

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

Joseph Ortiz, Antoine Dedieu, Wolfgang Lehrach, J. Swaroop Guntupalli, Carter Wendelken, Ahmad Humayun, Sivaramakrishnan Swaminathan, Guangyao Zhou, Miguel Lázaro-Gredilla, Kevin P. Murphy

Proceedings of the Advances in Neural Information Processing Systems 37: Annual Conference on Neural Information Processing Systems 2024, 2024

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

Antoine Dedieu, Wolfgang Lehrach, Guangyao Zhou, Dileep George, Miguel Lázaro-Gredilla

Proceedings of the Forty-first International Conference on Machine Learning, 2024

2021

Graphical Models with Attention for Context-Specific Independence and an Application to Perceptual Grouping.

Guangyao Zhou, Wolfgang Lehrach, Antoine Dedieu, Miguel Lázaro-Gredilla, Dileep George

CoRR, 2021

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

Miguel Lázaro-Gredilla, Wolfgang Lehrach, Nishad Gothoskar, Guangyao Zhou, Antoine Dedieu, Dileep George

Proceedings of the Thirty-Fifth AAAI Conference on Artificial Intelligence, 2021

2020

Query Training: Learning and inference for directed and undirected graphical models.

Miguel Lázaro-Gredilla, Wolfgang Lehrach, Nishad Gothoskar, Guangyao Zhou, Antoine Dedieu, Dileep George

CoRR, 2020

2019

Learning undirected models via query training.

Miguel Lázaro-Gredilla, Wolfgang Lehrach, Dileep George

CoRR, 2019

Learning higher-order sequential structure with cloned HMMs.

Antoine Dedieu, Nishad Gothoskar, Scott Swingle, Wolfgang Lehrach, Miguel Lázaro-Gredilla, Dileep George

CoRR, 2019

2016

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

Xinghua Lou, Ken Kansky, Wolfgang Lehrach, C. C. Laan, Bhaskara Marthi, D. Scott Phoenix, Dileep George

Proceedings of the Advances in Neural Information Processing Systems 29: Annual Conference on Neural Information Processing Systems 2016, 2016