Publications

Andrea Karlova, Rishabh Kabra, Daniel Augusto de Souza, Brooks Paige (2026). COBALT: Censored Optimization and Bayesian Active Learning Techniques. UAI 2026.

PDF

Letian Wang, Chuhan Zhang, Rishabh Kabra, Jasper Uijlings, Steven Waslander, Andrew Zisserman, Joao Carreira, Kaiming He, Misha Andriluka, Eduard Gabriel Bazavan, Andrei Zanfir, Cristian Sminchisescu (2026). Video Generation Models are General-Purpose Vision Learners. ECCV 2026.

PDF Code

Rishabh Kabra, Maks Ovsjanikov, Drew A Hudson, Ye Xia, Skanda Koppula, Andre Araujo, Joao Carreira, Niloy J Mitra (2026). A Mixed Diet Makes DINO An Omnivorous Vision Encoder. arXiv.

PDF Code

Rishabh Kabra, Drew A Hudson, Sjoerd van Steenkiste, Joao Carreira, Niloy J Mitra (2026). How to Spin an Object: First, Get the Shape Right. CVPRW 2026.

PDF Code

Shiting Xiao, Rishabh Kabra, Yuhang Li, Donghyun Lee, Joao Carreira, Priyadarshini Panda (2025). OpenWorldSAM: Extending SAM2 for Universal Image Segmentation with Language Prompts. arXiv.

PDF Code

Jacob C Walker, Pedro Vélez, Luisa Polania Cabrera, Guangyao Zhou, Rishabh Kabra, Carl Doersch, Maks Ovsjanikov, Joao Carreira, Shiry Ginosar (2025). Generalist forecasting with frozen video models via latent diffusion. arXiv.

PDF

Pedro Vélez, Luisa F Polanía, Yi Yang, Chuhan Zhang, Rishabh Kabra, Anurag Arnab, Mehdi SM Sajjadi (2025). From Image to Video: An Empirical Study of Diffusion Representations. ICCV 2025.

PDF

Joao Carreira, Dilara Gokay, Michael King, Chuhan Zhang, Ignacio Rocco, Aravindh Mahendran, Thomas Albert Keck, Joseph Heyward, Skanda Koppula, Etienne Pot, Goker Erdogan, Yana Hasson, Yi Yang, Klaus Greff, Guillaume Le Moing, Sjoerd van Steenkiste, Daniel Zoran, Drew A Hudson, Pedro Vélez, Luisa Polanía, Luke Friedman, Chris Duvarney, Ross Goroshin, Kelsey Allen, Jacob Walker, Rishabh Kabra, Eric Aboussouan, Jennifer Sun, Thomas Kipf, Carl Doersch, Viorica Pătrăucean, Dima Damen, Pauline Luc, Mehdi SM Sajjadi, Andrew Zisserman (2024). Scaling 4D Representations. arXiv.

PDF Code

Ziyi Wu, Yulia Rubanova, Rishabh Kabra, Drew Hudson, Igor Gilitschenski, Yusuf Aytar, Sjoerd van Steenkiste, Kelsey Allen, Thomas Kipf (2024). Neural Assets: 3D-Aware Multi-Object Scene Synthesis with Image Diffusion Models. NeurIPS 2024.

PDF Project

Sjoerd van Steenkiste, Daniel Zoran, Yi Yang, Yulia Rubanova, Rishabh Kabra, Carl Doersch, Dilara Gokay, Etienne Pot, Klaus Greff, Drew Hudson, Thomas Keck, Joao Carreira, Alexey Dosovitskiy, Mehdi SM Sajjadi, Thomas Kipf (2024). Moving Off-the-Grid: Scene-Grounded Video Representations. NeurIPS 2024.

PDF

Lucas Beyer, Andreas Steiner, André Susano Pinto, Alexander Kolesnikov, Xiao Wang, Daniel Salz, Maxim Neumann, Ibrahim Alabdulmohsin, Michael Tschannen, Emanuele Bugliarello, Thomas Unterthiner, Daniel Keysers, Skanda Koppula, Fangyu Liu, Adam Grycner, Alexey Gritsenko, Neil Houlsby, Manoj Kumar, Keran Rong, Julian Eisenschlos, Rishabh Kabra, Matthias Bauer, Matko Bošnjak, Xi Chen, Matthias Minderer, Paul Voigtlaender, Ioana Bica, Ivana Balazevic, Joan Puigcerver, Pinelopi Papalampidi, Olivier Henaff, Xi Xiong, Radu Soricut, Jeremiah Harmsen, Xiaohua Zhai (2024). PaliGemma: A versatile 3B VLM for transfer. arXiv.

PDF Code

Rishabh Kabra, Loic Matthey, Alexander Lerchner, Niloy J. Mitra (2024). Leveraging VLM-Based Pipelines to Annotate 3D Objects. In ICML.

Cite Project

Rishabh Kabra, Daniel Zoran, Goker Erdogan, Loic Matthey, Antonia Creswell, Matt Botvinick, Alexander Lerchner, Chris Burgess (2021). SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decomposition. In NeurIPS.

Cite Project Slides Video

Antonia Creswell, Kyriacos Nikiforou, Oriol Vinyals, Andre Saraiva, Rishabh Kabra, Loic Matthey, Chris Burgess, Malcolm Reynolds, Richard Tanburn, Marta Garnelo, Murray Shanahan (2020). AlignNet: Unsupervised Entity Alignment. arXiv.

PDF

Klaus Greff, Raphaël Lopez Kaufmann, Rishabh Kabra, Nick Watters, Chris Burgess, Daniel Zoran, Loic Matthey, Matthew Botvinick, Alexander Lerchner (2019). Multi-Object Representation Learning with Iterative Variational Inference. ICML 2019.

PDF

Arthur Guez, Mehdi Mirza, Karol Gregor, Rishabh Kabra, Sébastien Racanière, Théophane Weber, David Raposo, Adam Santoro, Laurent Orseau, Tom Eccles, Greg Wayne, David Silver, Timothy Lillicrap (2019). An Investigation of Model-Free Planning. ICML 2019.

PDF

Christopher P Burgess, Loic Matthey, Nicholas Watters, Rishabh Kabra, Irina Higgins, Matt Botvinick, Alexander Lerchner (2019). MONet: Unsupervised Scene Decomposition and Representation. arXiv.

PDF