← Search

Dileep George

12 accepted papers

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
2024

What type of inference is planning?

NeurIPS 2024spotlight

Multiple types of inference are available for probabilistic graphical models, e.g., marginal, maximum-a-posteriori, and even marginal maximum-a-posteriori. Which one do researchers mean when they talk about ``planning as inference''? There is no consistency in the literature, different types are use…

2023

3D Neural Embedding Likelihood: Probabilistic Inverse Graphics for Robust 6D Pose Estimation

ICCV 2023poster

The ability to perceive and understand 3D scenes is crucial for many applications in computer vision and robotics. Inverse graphics is an appealing approach to 3D scene understanding that aims to infer the 3D scene structure from 2D images. In this paper, we introduce probabilistic modeling to the i…

Cited by 4PDFcodeScholar
2023

Learning Noisy OR Bayesian Networks with Max-Product Belief Propagation

ICML 2023poster

Noisy-OR Bayesian Networks (BNs) are a family of probabilistic graphical models which express rich statistical dependencies in binary data. Variational inference (VI) has been the main method proposed to learn noisy-OR BNs with complex latent structures (Jaakkola & Jordan, 1999; Ji et al., 2020; Buh…

Cited by 1SourcePDFScholar
2023

Schema-learning and rebinding as mechanisms of in-context learning and emergence

NeurIPS 2023spotlight

In-context learning (ICL) is one of the most powerful and most unexpected capabilities to emerge in recent transformer-based large language models (LLMs). Yet the mechanisms that underlie it are poorly understood. In this paper, we demonstrate that comparable ICL capabilities can be acquired by an a…

Cited by 20SourcePDFScholar
2022

DURableVS: Data-efficient Unsupervised Recalibrating Visual Servoing via online learning in a structured generative model

ICRA 2022poster

Visual servoing enables robotic systems to perform accurate closed-loop control, which is required in many applications. However, existing methods require either precise calibration of the robot kinematic model and cameras or use neural architectures that require large amounts of data to train. In t…

Cited by 2SourceScholar
2021

Perturb-and-max-product: Sampling and learning in discrete energy-based models

NeurIPS 2021poster

Perturb-and-MAP offers an elegant approach to approximately sample from a energy-based model (EBM) by computing the maximum-a-posteriori (MAP) configuration of a perturbed version of the model. Sampling in turn enables learning. However, this line of research has been hindered by the general intract…

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…

2021

Sample-Efficient L0-L2 Constrained Structure Learning of Sparse Ising Models

AAAI 2021technical

We consider the problem of learning the underlying graph of a sparse Ising model with p nodes from n i.i.d. samples. The most recent and best performing approaches combine an empirical loss (the logistic regression loss or the interaction screening loss) with a regularizer (an L1 penalty or an L1 co…

2017

Schema Networks: Zero-shot Transfer with a Generative Causal Model of Intuitive Physics

ICML 2017poster

The recent adaptation of deep neural network-based methods to reinforcement learning and planning domains has yielded remarkable progress on individual tasks. Nonetheless, progress on task-to-task transfer remains limited. In pursuit of efficient and robust generalization, we introduce the Schema Ne…

Cited by 300SourcePDFScholar
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