Lehrach, Wolfgang

8 publications

TMLR 2025 Diffusion Model Predictive Control Guangyao Zhou, Sivaramakrishnan Swaminathan, Rajkumar Vasudeva Raju, J Swaroop Guntupalli, Wolfgang Lehrach, Joseph Ortiz, Antoine Dedieu, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

ICML 2025 Improving Transformer World Models for Data-Efficient RL Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, J Swaroop Guntupalli, Wolfgang Lehrach, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

ICLRW 2025 Improving Transformer World Models for Data-Efficient RL Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, Wolfgang Lehrach, J Swaroop Guntupalli, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

NeurIPS 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, Guangyao Zhou, Sivaramakrishnan Swaminathan, Miguel Lázaro-Gredilla, Kevin Murphy

ICML 2024 Learning Cognitive Maps from Transformer Representations for Efficient Planning in Partially Observed Environments Antoine Dedieu, Wolfgang Lehrach, Guangyao Zhou, Dileep George, Miguel Lazaro-Gredilla

MLOSS 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

AAAI 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

NeurIPS 2016 Generative Shape Models: Joint Text Recognition and Segmentation with Very Little Training Data Xinghua Lou, Ken Kansky, Wolfgang Lehrach, Cc Laan, Bhaskara Marthi, D. Phoenix, Dileep George