← Search

Tongzhou Wang

15 accepted papers

2024

Characterizing Model Robustness via Natural Input Gradients

ECCV 2024poster

"Adversarially robust models are locally smooth around each data sample so that small perturbations cannot drastically change model outputs. In modern systems, such smoothness is usually obtained via Adversarial Training, which explicitly enforces models to perform well on perturbed examples. In thi…

2023

Generalizing Dataset Distillation via Deep Generative Prior

CVPR 2023poster

Dataset Distillation aims to distill an entire dataset's knowledge into a few synthetic images. The idea is to synthesize a small number of synthetic data points that, when given to a learning algorithm as training data, result in a model approximating one trained on the original data. Despite a rec…

2023

Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning

ICML 2023poster

In goal-reaching reinforcement learning (RL), the optimal value function has a particular geometry, called quasimetrics structure. This paper introduces Quasimetric Reinforcement Learning (QRL), a new RL method that utilizes quasimetric models to learn optimal value functions. Distinct from prior ap…

2022

Dataset Distillation by Matching Training Trajectories

CVPR 2022oral

Dataset distillation is the task of synthesizing a small dataset such that a model trained on the synthetic set will match the test accuracy of the model trained on the full dataset. The task is extremely challenging as it often involves backpropagating through the full training process or assuming…

Cited by 451PDFcodeScholar
2022

Denoised MDPs: Learning World Models Better Than the World Itself

ICML 2022spotlight

The ability to separate signal from noise, and reason with clean abstractions, is critical to intelligence. With this ability, humans can efficiently perform real world tasks without considering all possible nuisance factors. How can artificial agents do the same? What kind of information can agents…

2022

Procedural Image Programs for Representation Learning

NeurIPS 2022accept

Learning image representations using synthetic data allows training neural networks without some of the concerns associated with real images, such as privacy and bias. Existing work focuses on a handful of curated generative processes which require expert knowledge to design, making it hard to scale…

2022

Totems: Physical Objects for Verifying Visual Integrity

ECCV 2022poster

"We introduce a new approach to image forensics: placing physical refractive objects, which we call totems, into a scene so as to protect any photograph taken of that scene. Totems bend and redirect light rays, thus providing multiple, albeit distorted, views of the scene within a single image. A de…

Cited by 3SourcePDFScholar
2021

Learning to See by Looking at Noise

NeurIPS 2021spotlight

Current vision systems are trained on huge datasets, and these datasets come with costs: curation is expensive, they inherit human biases, and there are concerns over privacy and usage rights. To counter these costs, interest has surged in learning from cheaper data sources, such as unlabeled images…

2020

Diverse Image Generation via Self-Conditioned GANs

CVPR 2020poster

We introduce a simple but effective unsupervised method for generating diverse images. We train a class-conditional GAN model without using manually annotated class labels. Instead, our model is conditional on labels automatically derived from clustering in the discriminator's feature space. Our clu…

Cited by 132PDFcodeScholar
2020

Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere

ICML 2020poster

Contrastive representation learning has been outstandingly successful in practice. In this work, we identify two key properties related to the contrastive loss: (1) alignment (closeness) of features from positive pairs, and (2) uniformity of the induced distribution of the (normalized) features on t…

2017

Learning to Synthesize a 4D RGBD Light Field From a Single Image

ICCV 2017spotlight

We present a machine learning algorithm that takes as input a 2D RGB image and synthesizes a 4D RGBD light field (color and depth of the scene in each ray direction). For training, we introduce the largest public light field dataset, consisting of over 3300 plenoptic camera light fields of scenes co…

Cited by 297PDFScholar