← Search

Baifeng Shi

17 accepted papers

2026

Attend Before Attention: Efficient and Scalable Video Understanding via Autoregressive Gazing

CVPR 2026

Multi-modal large language models (MLLMs) have advanced general-purpose video understanding but struggle with long, high-resolution videos---they process every pixel equally in their vision transformers (ViTs) or LLMs despite significant spatiotemporal redundancy. We introduce AutoGaze, a lightweigh

Cited by 0SourcecodeScholar
2026

Learning to Grasp Anything By Playing with Random Toys

ICLR 2026poster

Robotic manipulation policies often struggle to generalize to novel objects, limiting their real-world utility. In contrast, cognitive science suggests that children develop generalizable dexterous manipulation skills by mastering a small set of simple toys and then applying that knowledge to more c…

Cited by 0SourceScholar
2025

NVILA: Efficient Frontier Visual Language Models

CVPR 2025poster

Visual language models (VLMs) have made significant advances in accuracy in recent years. However, their efficiency has received much less attention. This paper introduces NVILA, a family of open VLMs designed to optimize both efficiency and accuracy. Building on top of VILA, we improve its model ar…

Cited by 43SourcePDFScholar
2025

Scaling Vision Pre-Training to 4K Resolution

CVPR 2025highlight

High-resolution perception of visual details is crucial for daily tasks. Current vision pre-training, however, is still limited to low resolutions (e.g., 378 x 378 pixels) due to the quadratic cost of processing larger images. We introduce PS3 that scales CLIP-style vision pre-training to 4K resolut…

Cited by 0SourcePDFScholar
2024

Humanoid Locomotion as Next Token Prediction

NeurIPS 2024spotlight

We cast real-world humanoid control as a next token prediction problem, akin to predicting the next word in language. Our model is a causal transformer trained via autoregressive prediction of sensorimotor sequences. To account for the multi-modal nature of the data, we perform prediction in a modal…

Cited by 55SourcePDFScholar
2024

LLARVA: Vision-Action Instruction Tuning Enhances Robot Learning

CoRL 2024poster

In recent years, instruction-tuned Large Multimodal Models (LMMs) have been successful at several tasks, including image captioning and visual question answering; yet leveraging these models remains an open question for robotics. Prior LMMs for robotics applications have been extensively trained on…

Cited by 22SourceScholar
2023

Robot Learning with Sensorimotor Pre-training

CoRL 2023oral

We present a self-supervised sensorimotor pre-training approach for robotics. Our model, called RPT, is a Transformer that operates on sequences of sensorimotor tokens. Given a sequence of camera images, proprioceptive robot states, and actions, we encode the sequence into tokens, mask out a subset,…

Cited by 54SourceScholar
2022

Visual Attention Emerges from Recurrent Sparse Reconstruction

ICML 2022spotlight

Visual attention helps achieve robust perception under noise, corruption, and distribution shifts in human vision, which are areas where modern neural networks still fall short. We present VARS, Visual Attention from Recurrent Sparse reconstruction, a new attention formulation built on two prominent…

2021

Temporal Action Detection With Multi-Level Supervision

ICCV 2021poster

Training temporal action detection in videos requires large amounts of labeled data, yet such annotation is expensive to collect. Incorporating unlabeled or weakly-labeled data to train action detection model could help reduce annotation cost. In this work, we first introduce the Semi-supervised Act…

Cited by 16PDFcodeScholar
2020

Auxiliary Task Reweighting for Minimum-data Learning

NeurIPS 2020poster

Supervised learning requires a large amount of training data, limiting its application where labeled data is scarce. To compensate for data scarcity, one possible method is to utilize auxiliary tasks to provide additional supervision for the main task. Assigning and optimizing the importance weights…

Cited by 39SourcePDFScholar
2020

Informative Dropout for Robust Representation Learning: A Shape-bias Perspective

ICML 2020poster

Convolutional Neural Networks (CNNs) are known to rely more on local texture rather than global shape when making decisions. Recent work also indicates a close relationship between CNN’s texture-bias and its robustness against distribution shift, adversarial perturbation, random corruption, etc. In…

2020

Weakly-Supervised Action Localization by Generative Attention Modeling

CVPR 2020poster

Weakly-supervised temporal action localization is a problem of learning an action localization model with only video-level action labeling available. The general framework largely relies on the classification activation, which employs an attention model to identify the action-related frames and then…

Cited by 196PDFcodeScholar
2020

Weakly-Supervised Action Localization with Expectation-Maximization Multi-Instance Learning

ECCV 2020poster

Weakly-supervised action localization requires training a model to localize the action segments in the video given only video level action label. It can be solved under the Multiple Instance Learning (MIL) framework, where a bag (video) contains multiple instances (action segments). Since only the b…