← Search

Rishabh Kabra

11 accepted papers

2026

A Mixed Diet Makes DINO An Omnivorous Vision Encoder

CVPR 2026

Pre-trained vision encoders like DINOv2 have demonstrated exceptional performance on unimodal tasks. However, we observe that their features are poorly aligned across different modalities. For instance, the feature embedding for an RGB image and its corresponding depth map of the same scene exhibit

Cited by 0SourcecodeScholar
2025

From Image to Video: An Empirical Study of Diffusion Representations

ICCV 2025poster

Diffusion models have revolutionized generative modeling, enabling unprecedented realism in image and video synthesis.This success has sparked interest in leveraging their representations for visual understanding tasks. While recent works have explored this potential for image generation, the visual…

Cited by 0SourcePDFScholar
2025

OpenWorldSAM: Extending SAM2 for Universal Image Segmentation with Language Prompts

NeurIPS 2025spotlight

The ability to segment objects based on open-ended language prompts remains a critical challenge, requiring models to ground textual semantics into precise spatial masks while handling diverse and unseen categories. We present OpenWorldSAM, a framework that extends the prompt-driven Segment Anything…

Cited by 0SourcecodeScholar
2024

Leveraging VLM-Based Pipelines to Annotate 3D Objects

ICML 2024poster

Pretrained vision language models (VLMs) present an opportunity to caption unlabeled 3D objects at scale. The leading approach to summarize VLM descriptions from different views of an object (Luo et al., 2023) relies on a language model (GPT4) to produce the final output. This text-based aggregation…

Cited by 6SourcePDFScholar
2024

Moving Off-the-Grid: Scene-Grounded Video Representations

NeurIPS 2024spotlight

Current vision models typically maintain a fixed correspondence between their representation structure and image space. Each layer comprises a set of tokens arranged “on-the-grid,” which biases patches or tokens to encode information at a specific spatio(-temporal) location. In this work we present…

Cited by 2SourcePDFScholar
2024

Neural Assets: 3D-Aware Multi-Object Scene Synthesis with Image Diffusion Models

NeurIPS 2024spotlight

We address the problem of multi-object 3D pose control in image diffusion models. Instead of conditioning on a sequence of text tokens, we propose to use a set of per-object representations, *Neural Assets*, to control the 3D pose of individual objects in a scene. Neural Assets are obtained by pooli…

Cited by 13SourcePDFScholar
2021

PARTS: Unsupervised Segmentation With Slots, Attention and Independence Maximization

ICCV 2021poster

From an early age, humans perceive the visual world as composed of coherent objects with distinctive properties such as shape, size, and color. There is great interest in building models that are able to learn similar structure, ideally in an unsupervised manner. Learning such structure from complex…

Cited by 53PDFScholar
2021

SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decomposition

NeurIPS 2021poster

To help agents reason about scenes in terms of their building blocks, we wish to extract the compositional structure of any given scene (in particular, the configuration and characteristics of objects comprising the scene). This problem is especially difficult when scene structure needs to be inferr…

Cited by 82SourcePDFScholar
2021

Unsupervised Object-Based Transition Models For 3D Partially Observable Environments

NeurIPS 2021poster

We present a slot-wise, object-based transition model that decomposes a scene into objects, aligns them (with respect to a slot-wise object memory) to maintain a consistent order across time, and predicts how those objects evolve over successive frames. The model is trained end-to-end without superv…

Cited by 27SourcePDFScholar
2019

An Investigation of Model-Free Planning

ICML 2019oral

The field of reinforcement learning (RL) is facing increasingly challenging domains with combinatorial complexity. For an RL agent to address these challenges, it is essential that it can plan effectively. Prior work has typically utilized an explicit model of the environment, combined with a specif…

2019

Multi-Object Representation Learning with Iterative Variational Inference

ICML 2019oral

Human perception is structured around objects which form the basis for our higher-level cognition and impressive systematic generalization abilities. Yet most work on representation learning focuses on feature learning without even considering multiple objects, or treats segmentation as an (often su…