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 (NIPS, 2016)

Learning higher-order sequential structure with cloned HMMs. - Antoine Dedieu; Nishad Gothoskar; Scott Swingle; Wolfgang Lehrach; Miguel Lázaro-Gredilla; Dileep George (CoRR, 2019)

Learning undirected models via query training. - Miguel Lázaro-Gredilla; Wolfgang Lehrach; Dileep George (CoRR, 2019)

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)

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 (AAAI, 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)

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 (ICML, 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 (NeurIPS, 2024)

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; Wolfgang Lehrach; J. Swaroop Guntupalli; Miguel Lázaro-Gredilla; Kevin Patrick Murphy (CoRR, 2025)