← Search

Yuning Chai

20 accepted papers

2025

Cohere3D: Exploiting Temporal Coherence for Unsupervised Representation Learning of Vision-Based Autonomous Driving

ICRA 2025

Multi-frame temporal inputs are important for vision-based autonomous driving. Observations from different angles enable the recovery of 3 D object states from 2 D images as long as we can identify the same instance from different input frames. However, the dynamic nature of driving scenes leads to

Cited by 3SourceScholar
2025

DriveGPT: Scaling Autoregressive Behavior Models for Driving

ICML 2025poster

We present DriveGPT, a scalable behavior model for autonomous driving. We model driving as a sequential decision-making task, and learn a transformer model to predict future agent states as tokens in an autoregressive fashion. We scale up our model parameters and training data by multiple orders of…

Cited by 1SourcePDFScholar
2025

Generative Data Mining with Longtail-Guided Diffusion

ICML 2025poster

It is difficult to anticipate the myriad challenges that a predictive model will encounter once deployed. Common practice entails a reactive, cyclical approach: model deployment, data mining, and retraining. We instead develop a proactive longtail discovery process by imagining additional data durin…

Cited by 0SourcePDFScholar
2025

PROFIT: A Specialized Optimizer for Deep Fine Tuning

NeurIPS 2025poster

The fine-tuning of pre-trained models has become ubiquitous in generative AI, computer vision, and robotics. Although much attention has been paid to improving the efficiency of fine-tuning model, there has been less scholarship around fine-tuning specifically for improved model performance. To reme…

Cited by 0SourceScholar
2025

Reading Recognition in the Wild

NeurIPS 2025poster

To enable egocentric contextual AI in always-on smart glasses, it is crucial to be able to keep a record of the user's interactions with the world, including during reading. In this paper, we introduce a new task of reading recognition to determine when the user is reading. We first introduce the fi…

Cited by 0SourceScholar
2024

ViP-LLaVA: Making Large Multimodal Models Understand Arbitrary Visual Prompts

CVPR 2024poster

While existing large vision-language multimodal models focus on whole image understanding there is a prominent gap in achieving region-specific comprehension. Current approaches that use textual coordinates or spatial encodings often fail to provide a user-friendly interface for visual prompting. To…

Cited by 106SourcePDFScholar
2023

Efficient Transformer-based 3D Object Detection with Dynamic Token Halting

ICCV 2023poster

Balancing efficiency and accuracy is a long-standing problem for deploying deep learning models. The trade-off is even more important for real-time safety-critical systems like autonomous vehicles. In this paper, we propose an effective approach for accelerating transformer-based 3D object detectors…

Cited by 8PDFScholar
2023

SHIFT3D: Synthesizing Hard Inputs For Tricking 3D Detectors

ICCV 2023poster

We present SHIFT3D, a differentiable pipeline for generating 3D shapes that are structurally plausible yet challenging to 3D object detectors. In safety-critical applications like autonomous driving, discovering such novel challenging objects can offer insight into unknown vulnerabilities of 3D dete…

Cited by 1PDFScholar
2022

Occupancy Flow Fields for Motion Forecasting in Autonomous Driving

RA-L 2022

We propose Occupancy Flow Fields, a new representation for motion forecasting of multiple agents, an important task in autonomous driving.Our representation is a spatio-temporal grid with each grid cell containing both the probability of the cell being occupied by any agent, and a two-dimensional fl

Cited by 99SourceScholar
2021

HDMapGen: A Hierarchical Graph Generative Model of High Definition Maps

CVPR 2021poster

High Definition (HD) maps are maps with precise definitions of road lanes with rich semantics of the traffic rules. They are critical for several key stages in an autonomous driving system, including motion forecasting and planning. However, there are only a small amount of real-world road topologie…

Cited by 71PDFScholar
2021

Large Scale Interactive Motion Forecasting for Autonomous Driving: The Waymo Open Motion Dataset

ICCV 2021poster

As autonomous driving systems mature, motion forecasting has received increasing attention as a critical requirement for planning. Of particular importance are interactive situations such as merges, unprotected turns, etc., where predicting individual object motion is not sufficient. Joint predictio…

Cited by 624PDFScholar
2021

RSN: Range Sparse Net for Efficient, Accurate LiDAR 3D Object Detection

CVPR 2021poster

The detection of 3D objects from LiDAR data is a critical component in most autonomous driving systems. Safe, high speed driving needs larger detection ranges, which are enabled by new LiDARs. These larger detection ranges require more efficient and accurate detection models. Towards this goal, we p…

Cited by 204PDFScholar
2021

To the Point: Efficient 3D Object Detection in the Range Image With Graph Convolution Kernels

CVPR 2021poster

3D object detection is vital for many robotics applications. For tasks where a 2D perspective range image exists, we propose to learn a 3D representation directly from this range image view. To this end, we designed a 2D convolutional network architecture that carries the 3D spherical coordinates of…

Cited by 87PDFScholar
2020

Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropout

NeurIPS 2020poster

The vast majority of deep models use multiple gradient signals, typically corresponding to a sum of multiple loss terms, to update a shared set of trainable weights. However, these multiple updates can impede optimal training by pulling the model in conflicting directions. We present Gradient Sign D…

2020

Scalability in Perception for Autonomous Driving: Waymo Open Dataset

CVPR 2020poster

The research community has increasing interest in autonomous driving research, despite the resource intensity of obtaining representative real world data. Existing self-driving datasets are limited in the scale and variation of the environments they capture, even though generalization within and bet…

Cited by 3735PDFScholar
2020

SurfelGAN: Synthesizing Realistic Sensor Data for Autonomous Driving

CVPR 2020oral

Autonomous driving system development is critically dependent on the ability to replay complex and diverse traffic scenarios in simulation. In such scenarios, the ability to accurately simulate the vehicle sensors such as cameras, lidar or radar is hugely helpful. However, current sensor simulators…

Cited by 125PDFScholar
2019

FEELVOS: Fast End-To-End Embedding Learning for Video Object Segmentation

CVPR 2019poster

Many of the recent successful methods for video object segmentation (VOS) are overly complicated, heavily rely on fine-tuning on the first frame, and/or are slow, and are hence of limited practical use. In this work, we propose FEELVOS as a simple and fast method which does not rely on fine-tuning.…

Cited by 528PDFcodeScholar
2019

MultiPath: Multiple Probabilistic Anchor Trajectory Hypotheses for Behavior Prediction

CoRL 2019

Predicting human behavior is a difficult and crucial task required for motion planning. It is challenging in large part due to the highly uncertain and multimodal set of possible outcomes in real-world domains such as autonomous driving. Beyond single MAP trajectory prediction [1, 2], obtaining an a

Cited by 0SourcePDFScholar