← Search

Alexander Cong Li

5 accepted papers

2024

On the Surprising Effectiveness of Attention Transfer for Vision Transformers

NeurIPS 2024poster

Conventional wisdom suggests that pre-training Vision Transformers (ViT) improves downstream performance by learning useful representations. Is this actually true? We investigate this question and find that the features and representations learned during pre-training are not essential. Surprisingly…

2023

Diffusion-TTA: Test-time Adaptation of Discriminative Models via Generative Feedback

NeurIPS 2023poster

The advancements in generative modeling, particularly the advent of diffusion models, have sparked a fundamental question: how can these models be effectively used for discriminative tasks? In this work, we find that generative models can be great test-time adapters for discriminative models. Our me…

2023

Internet Explorer: Targeted Representation Learning on the Open Web

ICML 2023poster

Vision models typically rely on fine-tuning general-purpose models pre-trained on large, static datasets. These general-purpose models only capture the knowledge within their pre-training datasets, which are tiny, out-of-date snapshots of the Internet---where billions of images are uploaded each day…

2021

Functional Regularization for Reinforcement Learning via Learned Fourier Features

NeurIPS 2021poster

We propose a simple architecture for deep reinforcement learning by embedding inputs into a learned Fourier basis and show that it improves the sample efficiency of both state-based and image-based RL. We perform infinite-width analysis of our architecture using the Neural Tangent Kernel and theoret…