← Search

Kaiming He

65 accepted papers

2026

Bidirectional Normalizing Flow: From Data to Noise and Back

CVPR 2026

Normalizing Flows (NFs) have been established as a principled framework for generative modeling. Standard NFs consist of a forward process and a reverse process: the forward process maps data to noise, while the reverse process generates samples by inverting it. Typical NF forward transformations ar

Cited by 0SourcecodeScholar
2026

GeoPT: Scaling Physics Simulation via Lifted Geometric Pre-Training

ICML 2026poster

Neural simulators promise efficient surrogates for physics simulation, but scaling them is bottlenecked by the prohibitive cost of generating high-fidelity training data. Pre-training on abundant off-the-shelf geometries offers a natural alternative, yet faces a fundamental gap: supervision on stati…

Cited by 0SourceScholar
2026

Improved Mean Flows: On the Challenges of Fastforward Generative Models

CVPR 2026

MeanFlow (MF) has recently been established as a framework for one-step generative modeling. However, its "fastforward" nature introduces key challenges in both the training objective and the guidance mechanism. First, the original MF's training target depends not only on the underlying ground-truth

Cited by 0SourcecodeScholar
2026

In Pursuit of Pixel Supervision for Visual Pre-training

CVPR 2026

Data matters. In computer vision, data (or pixels) are the primary source of information containing signals that span from low-level attributes to high-level concepts. At scale, the success of modern vision systems has been closely tied to how data is curated for semantic understanding (e.g., ImageN

Cited by 0SourcecodeScholar
2026

One-step Latent-free Image Generation with Pixel Mean Flows

ICML 2026poster

Modern diffusion/flow-based models for image generation typically exhibit two core characteristics: (i) using multi-step sampling, and (ii) operating in a latent space. Recent advances have made encouraging progress on each aspect individually, paving the way toward one-step diffusion/flow without l…

Cited by 0SourceScholar
2025

Deconstructing Denoising Diffusion Models for Self-Supervised Learning

ICLR 2025poster

In this study, we examine the representation learning abilities of Denoising Diffusion Models (DDM) that were originally purposed for image generation. Our philosophy is to deconstruct a DDM, gradually transforming it into a classical Denoising Autoencoder (DAE). This deconstructive process allows u…

Cited by 75SourcePDFScholar
2025

Fluid: Scaling Autoregressive Text-to-image Generative Models with Continuous Tokens

ICLR 2025poster

Scaling up autoregressive models in vision has not proven as beneficial as in large language models. In this work, we investigate this scaling problem in the context of text-to-image generation, focusing on two critical factors: whether models use discrete or continuous tokens, and whether tokens ar…

Cited by 35SourcePDFScholar
2025

Highly Compressed Tokenizer Can Generate Without Training

ICML 2025poster

Commonly used image tokenizers produce a 2D grid of spatially arranged tokens. In contrast, so-called *1D* image tokenizers represent images as highly compressed one-dimensional sequences of as few as 32 discrete tokens. We find that the high degree of compression achieved by a 1D tokenizer with vec…

2025

Is Noise Conditioning Necessary for Denoising Generative Models?

ICML 2025poster

It is widely believed that noise conditioning is indispensable for denoising diffusion models to work successfully. This work challenges this belief. Motivated by research on blind image denoising, we investigate a variety of denoising-based generative models in the absence of noise conditioning. To…

Cited by 4SourcePDFScholar
2025

TetSphere Splatting: Representing High-Quality Geometry with Lagrangian Volumetric Meshes

ICLR 2025oral

We introduce TetSphere Splatting, a Lagrangian geometry representation designed for high-quality 3D shape modeling. TetSphere splatting leverages an underused yet powerful geometric primitive -- volumetric tetrahedral meshes. It represents 3D shapes by deforming a collection of tetrahedral spheres,…

Cited by 4SourcePDFScholar
2024

Autoregressive Image Generation without Vector Quantization

NeurIPS 2024spotlight

Conventional wisdom holds that autoregressive models for image generation are typically accompanied by vector-quantized tokens. We observe that while a discrete-valued space can facilitate representing a categorical distribution, it is not a necessity for autoregressive modeling. In this work, we pr…

2024

Physically Compatible 3D Object Modeling from a Single Image

NeurIPS 2024spotlight

We present a computational framework that transforms single images into 3D physical objects. The visual geometry of a physical object in an image is determined by three orthogonal attributes: mechanical properties, external forces, and rest-shape geometry. Existing single-view 3D reconstruction meth…

Cited by 9SourcePDFScholar
2024

Return of Unconditional Generation: A Self-supervised Representation Generation Method

NeurIPS 2024oral

Unconditional generation -- the problem of modeling data distribution without relying on human-annotated labels -- is a long-standing and fundamental challenge in generative models, creating a potential of learning from large-scale unlabeled data. In the literature, the generation quality of an unco…

2024

Scaling Proprioceptive-Visual Learning with Heterogeneous Pre-trained Transformers

NeurIPS 2024spotlight

One of the roadblocks for training generalist robotic models today is heterogeneity. Previous robot learning methods often collect data to train with one specific embodiment for one task, which is expensive and prone to overfitting. This work studies the problem of learning policy representations th…

2023

Scaling Language-Image Pre-Training via Masking

CVPR 2023poster

We present Fast Language-Image Pre-training (FLIP), a simple and more efficient method for training CLIP. Our method randomly masks out and removes a large portion of image patches during training. Masking allows us to learn from more image-text pairs given the same wall-clock time and contrast more…

2022

Exploring Plain Vision Transformer Backbones for Object Detection

ECCV 2022poster

"We explore the plain, non-hierarchical Vision Transformer (ViT) as a backbone network for object detection. This design enables the original ViT architecture to be fine-tuned for object detection without needing to redesign a hierarchical backbone for pre-training. With minimal adaptations for fine…

2022

Masked Autoencoders Are Scalable Vision Learners

CVPR 2022oral

This paper shows that masked autoencoders (MAE) are scalable self-supervised learners for computer vision. Our MAE approach is simple: we mask random patches of the input image and reconstruct the missing pixels. It is based on two core designs. First, we develop an asymmetric encoder-decoder archit…

Cited by 12331PDFcodeScholar
2022

Masked Autoencoders As Spatiotemporal Learners

NeurIPS 2022accept

This paper studies a conceptually simple extension of Masked Autoencoders (MAE) to spatiotemporal representation learning from videos. We randomly mask out spacetime patches in videos and learn an autoencoder to reconstruct them in pixels. Interestingly, we show that our MAE method can learn strong…

2021

A Large-Scale Study on Unsupervised Spatiotemporal Representation Learning

CVPR 2021poster

We present a large-scale study on unsupervised spatiotemporal representation learning from videos. With a unified perspective on four recent image-based frameworks, we study a simple objective that can easily generalize all these methods to space-time. Our objective encourages temporally-persistent…

Cited by 324PDFcodeScholar
2020

A Multigrid Method for Efficiently Training Video Models

CVPR 2020oral

Training competitive deep video models is an order of magnitude slower than training their counterpart image models. Slow training causes long research cycles, which hinders progress in video understanding research. Following standard practice for training image models, video model training has used…

Cited by 123PDFcodeScholar
2020

Are Labels Necessary for Neural Architecture Search?

ECCV 2020poster

Existing neural network architectures in computer vision --- whether designed by humans or by machines --- were typically found using both images and their associated labels. In this paper, we ask the question: can we find high-quality neural architectures using only images, but no human-annotated l…

2020

Designing Network Design Spaces

CVPR 2020poster

In this work, we present a new network design paradigm. Our goal is to help advance the understanding of network design and discover design principles that generalize across settings. Instead of focusing on designing individual network instances, we design network design spaces that parametrize popu…

Cited by 2278PDFcodeScholar
2020

Momentum Contrast for Unsupervised Visual Representation Learning

CVPR 2020oral

We present Momentum Contrast (MoCo) for unsupervised visual representation learning. From a perspective on contrastive learning as dictionary look-up, we build a dynamic dictionary with a queue and a moving-averaged encoder. This enables building a large and consistent dictionary on-the-fly that fac…

Cited by 15734PDFcodeScholar
2019

Deep Hough Voting for 3D Object Detection in Point Clouds

ICCV 2019oral

Current 3D object detection methods are heavily influenced by 2D detectors. In order to leverage architectures in 2D detectors, they often convert 3D point clouds to regular grids (i.e., to voxel grids or to bird's eye view images), or rely on detection in 2D images to propose 3D boxes. Few works ha…

Cited by 1587PDFcodeScholar
2019

Exploring Randomly Wired Neural Networks for Image Recognition

ICCV 2019oral

Neural networks for image recognition have evolved through extensive manual design from simple chain-like models to structures with multiple wiring paths. The success of ResNets and DenseNets is due in large part to their innovative wiring plans. Now, neural architecture search (NAS) studies are exp…

Cited by 472PDFcodeScholar
2019

Feature Denoising for Improving Adversarial Robustness

CVPR 2019poster

Adversarial attacks to image classification systems present challenges to convolutional networks and opportunities for understanding them. This study suggests that adversarial perturbations on images lead to noise in the features constructed by these networks. Motivated by this observation, we devel…

Cited by 1130PDFcodeScholar
2019

Long-Term Feature Banks for Detailed Video Understanding

CVPR 2019oral

To understand the world, we humans constantly need to relate the present to the past, and put events in context. In this paper, we enable existing video models to do the same. We propose a long-term feature bank--supportive information extracted over the entire span of a video--to augment state-of-t…

Cited by 621PDFcodeScholar
2018

Data Distillation: Towards Omni-Supervised Learning

CVPR 2018poster

We investigate omni-supervised learning, a special regime of semi-supervised learning in which the learner exploits all available labeled data plus internet-scale sources of unlabeled data. Omni-supervised learning is lower-bounded by performance on existing labeled datasets, offering the potential…

2018

Exploring the Limits of Weakly Supervised Pretraining

ECCV 2018poster

State-of-the-art visual perception models for a wide range of tasks rely on supervised pretraining. ImageNet classification is the de facto pretraining task for these models. Yet, ImageNet is now nearly ten years old and is by modern standards "small". Even so, relatively little is known about the b…

2018

GLoMo: Unsupervised Learning of Transferable Relational Graphs

NeurIPS 2018poster

Modern deep transfer learning approaches have mainly focused on learning generic feature vectors from one task that are transferable to other tasks, such as word embeddings in language and pretrained convolutional features in vision. However, these approaches usually transfer unary features and larg…

Cited by 27SourcePDFScholar
2018

Group Normalization

ECCV 2018poster

Batch Normalization (BN) is a milestone technique in the development of deep learning, enabling various networks to train. However, normalizing along the batch dimension introduces problems --- BN's error increases rapidly when the batch size becomes smaller, caused by inaccurate batch statistics es…

2017

Aggregated Residual Transformations for Deep Neural Networks

CVPR 2017poster

We present a simple, highly modularized network architecture for image classification. Our network is constructed by repeating a building block that aggregates a set of transformations with the same topology. Our simple design results in a homogeneous, multi-branch architecture that has only a few h…

Cited by 14783PDFcodeScholar
2017

Feature Pyramid Networks for Object Detection

CVPR 2017poster

Feature pyramids are a basic component in recognition systems for detecting objects at different scales. But pyramid representations have been avoided in recent object detectors that are based on deep convolutional networks, partially because they are slow to compute and memory intensive. In this pa…

Cited by 33044PDFcodeScholar
2017

Transitive Invariance for Self-Supervised Visual Representation Learning

ICCV 2017poster

Learning visual representations with self-supervised learning has become popular in computer vision. The idea is to design auxiliary tasks where labels are free to obtain. Most of these tasks end up providing data to learn specific kinds of invariance useful for recognition. In this paper, we propos…

Cited by 182PDFcodeScholar
2016

R-FCN: Object Detection via Region-based Fully Convolutional Networks

NeurIPS 2016poster

We present region-based, fully convolutional networks for accurate and efficient object detection. In contrast to previous region-based detectors such as Fast/Faster R-CNN that apply a costly per-region subnetwork hundreds of times, our region-based detector is fully convolutional with almost all co…

2016

ScribbleSup: Scribble-Supervised Convolutional Networks for Semantic Segmentation

CVPR 2016oral

Large-scale data are of crucial importance for learning semantic segmentation models, but annotating per-pixel masks is a tedious and inefficient procedure. We note that for the topic of interactive image segmentation, scribbles are very widely used in academic research and commercial software, and…

Cited by 1333PDFScholar
2015

BoxSup: Exploiting Bounding Boxes to Supervise Convolutional Networks for Semantic Segmentation

ICCV 2015poster

Recent leading approaches to semantic segmentation rely on deep convolutional networks trained with human-annotated, pixel-level segmentation masks. Such pixel-accurate supervision demands expensive labeling effort and limits the performance of deep networks that usually benefit from more training d…

Cited by 1368PDFScholar
2015

Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification

ICCV 2015poster

Rectified activation units (rectifiers) are essential for state-of-the-art neural networks. In this work, we study rectifier neural networks for image classification from two aspects. First, we propose a Parametric Rectified Linear Unit (PReLU) that generalizes the traditional rectified unit. PReLU…

Cited by 27645PDFScholar
2015

Efficient and Accurate Approximations of Nonlinear Convolutional Networks

CVPR 2015poster

This paper aims to accelerate the test-time computation of deep convolutional neural networks (CNNs). Unlike existing methods that are designed for approximating linear filters or linear responses, our method takes the nonlinear units into account. We minimize the reconstruction error of the nonline…

Cited by 344SourcePDFScholar
2015

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

NeurIPS 2015poster

State-of-the-art object detection networks depend on region proposal algorithms to hypothesize object locations. Advances like SPPnet and Fast R-CNN have reduced the running time of these detection networks, exposing region proposal computation as a bottleneck. In this work, we introduce a Region Pr…