← Search

Yuge Shi

11 accepted papers

2024

Genie: Generative Interactive Environments

ICML 2024oral

We introduce Genie, the first *generative interactive environment* trained in an unsupervised manner from unlabelled Internet videos. The model can be prompted to generate an endless variety of action-controllable virtual worlds described through text, synthetic images, photographs, and even sketche…

Cited by 172SourcePDFScholar
2024

Memory Consolidation Enables Long-Context Video Understanding

ICML 2024spotlight

Most transformer-based video encoders are limited to short temporal contexts due to their quadratic complexity. While various attempts have been made to extend this context, this has often come at the cost of both conceptual and computational complexity. We propose to instead re-purpose existing pre…

Cited by 25SourcePDFScholar
2024

Position: Open-Endedness is Essential for Artificial Superhuman Intelligence

ICML 2024oral

In recent years there has been a tremendous surge in the general capabilities of AI systems, mainly fuelled by training foundation models on internet-scale data. Nevertheless, the creation of open-ended, ever self-improving AI remains elusive. **In this position paper, we argue that the ingredients…

Cited by 27SourcePDFScholar
2023

How robust is unsupervised representation learning to distribution shift?

ICLR 2023poster

The robustness of machine learning algorithms to distributions shift is primarily discussed in the context of supervised learning (SL). As such, there is a lack of insight on the robustness of the representations learned from unsupervised methods, such as self-supervised learning (SSL) and auto-enco…

Cited by 21SourcePDFScholar
2023

Tuning Computer Vision Models With Task Rewards

ICML 2023poster

Misalignment between model predictions and intended usage can be detrimental for the deployment of computer vision models. The issue is exacerbated when the task involves complex structured outputs, as it becomes harder to design procedures which address this misalignment. In natural language proces…

2022

Gradient Matching for Domain Generalization

ICLR 2022poster

Machine learning systems typically assume that the distributions of training and test sets match closely. However, a critical requirement of such systems in the real world is their ability to generalize to unseen domains. Here, we propose an _inter-domain gradient matching_ objective that targets do…

2022

Learning Multimodal VAEs through Mutual Supervision

ICLR 2022spotlight

Multimodal VAEs seek to model the joint distribution over heterogeneous data (e.g.\ vision, language), whilst also capturing a shared representation across such modalities. Prior work has typically combined information from the modalities by reconciling idiosyncratic representations directly in the…

2021

Relating by Contrasting: A Data-efficient Framework for Multimodal Generative Models

ICLR 2021poster

Multimodal learning for generative models often refers to the learning of abstract concepts from the commonality of information in multiple modalities, such as vision and language. While it has proven effective for learning generalisable representations, the training of such models often requires a…

Cited by 45SourcePDFScholar
2019

Variational Mixture-of-Experts Autoencoders for Multi-Modal Deep Generative Models

NeurIPS 2019poster

Learning generative models that span multiple data modalities, such as vision and language, is often motivated by the desire to learn more useful, generalisable representations that faithfully capture common underlying factors between the modalities. In this work, we characterise successful learning…