← Search

Chongjian GE

23 accepted papers

2026

Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Models

ICML 2026poster

Reinforcement Learning (RL) has emerged as a central paradigm for advancing Large Language Models (LLMs), where both pre-training and RL post-training stages are grounded in the same log-likelihood formulation. In contrast, recent RL approaches for diffusion models, most notably Denoising Diffusion …

Cited by 0SourceScholar
2026

Both Semantics and Reconstruction Matter: Making Representation Encoders Ready for Text-to-Image Generation and Editing

ICML 2026poster

Modern Latent Diffusion Models (LDMs) typically operate in low-level Variational Autoencoder (VAE) latent spaces that are primarily optimized for pixel-level reconstruction. To unify vision generation and understanding, a burgeoning trend is to adopt high-dimensional features from representation enc…

Cited by 0SourceScholar
2026

FlashVideo: Flowing Fidelity to Detail for Efficient High-Resolution Video Generation

AAAI 2026technical

DiT models have achieved great success in text-to-video generation, leveraging their scalability in model capacity and data scale. High content and motion fidelity aligned with text prompts, however, often require large model parameters and a substantial number of function evaluations (NFEs). Realis

Cited by 0SourcePDFScholar
2025

CompGS: Unleashing 2D Compositionality for Compositional Text-to-3D via Dynamically Optimizing 3D Gaussians

CVPR 2025poster

Recent breakthroughs in text-guided image generation have significantly advanced the field of 3D generation. While generating a single high-quality 3D object is now feasible, generating multiple objects with reasonable interactions within a 3D space, a.k.a. compositional 3D generation, presents subs…

Cited by 4SourcePDFScholar
2025

Goku: Flow Based Video Generative Foundation Models

CVPR 2025highlight

This paper introduces Goku, a state-of-the-art family of joint image-and-video generation models leveraging rectified flow Transformers to achieve industry-leading performance. We detail the foundational elements enabling high-quality visual generation, including the data curation pipeline, model ar…

Cited by 15SourcePDFScholar
2025

Prompt-A-Video: Prompt Your Video Diffusion Model via Preference-Aligned LLM

ICCV 2025poster

Text-to-video models have made remarkable advancements through optimization on high-quality text-video pairs, where the textual prompts play a pivotal role in determining quality of output videos. However, achieving the desired output often entails multiple revisions and iterative inference to refin…

2025

WOMD-Reasoning: A Large-Scale Dataset for Interaction Reasoning in Driving

ICML 2025poster

Language models uncover unprecedented abilities in analyzing driving scenarios, owing to their limitless knowledge accumulated from text-based pre-training. Naturally, they should particularly excel in analyzing rule-based interactions, such as those triggered by traffic laws, which are well documen…

2024

DeepAccident: A Motion and Accident Prediction Benchmark for V2X Autonomous Driving

AAAI 2024technical

Safety is the primary priority of autonomous driving. Nevertheless, no published dataset currently supports the direct and explainable safety evaluation for autonomous driving. In this work, we propose DeepAccident, a large-scale dataset generated via a realistic simulator containing diverse acciden…

Cited by 65SourcePDFScholar
2024

InstructDET: Diversifying Referring Object Detection with Generalized Instructions

ICLR 2024poster

We propose InstructDET, a data-centric method for referring object detection (ROD) that localizes target objects based on user instructions. While deriving from referring expressions (REC), the instructions we leverage are greatly diversified to encompass common user intentions related to object det…

2024

Large Language Models as Automated Aligners for benchmarking Vision-Language Models

ICLR 2024poster

With the advancements in Large Language Models (LLMs), Vision-Language Models (VLMs) have reached a new level of sophistication, showing notable competence in executing intricate cognition and reasoning tasks. However, existing evaluation benchmarks, primarily relying on rigid, hand-crafted datasets…

Cited by 9SourcePDFScholar
2024

PixArt-$\alpha$: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

ICLR 2024spotlight

The most advanced text-to-image (T2I) models require significant training costs (e.g., millions of GPU hours), seriously hindering the fundamental innovation for the AIGC community while increasing CO2 emissions. This paper introduces PixArt-$\alpha$, a Transformer-based T2I diffusion model whose im…

Cited by 546SourcePDFScholar
2024

PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

ECCV 2024poster

"In this paper, we introduce , a Diffusion Transformer model (DiT) capable of directly generating images at 4K resolution. represents a significant advancement over its predecessor, , offering images of markedly higher fidelity and improved alignment with text prompts. A key feature of is its traini…

Cited by 178SourcePDFScholar
2024

RoboCodeX: Multimodal Code Generation for Robotic Behavior Synthesis

ICML 2024poster

Robotic behavior synthesis, the problem of understanding multimodal inputs and generating precise physical control for robots, is an important part of Embodied AI. Despite successes in applying multimodal large language models for high-level understanding, it remains challenging to translate these c…

Cited by 18SourcePDFScholar
2023

MetaBEV: Solving Sensor Failures for 3D Detection and Map Segmentation

ICCV 2023poster

Perception systems in modern autonomous driving vehicles typically take inputs from complementary multi-modal sensors, e.g., LiDAR and cameras. However, in real-world applications, sensor corruptions and failures lead to inferior performances, thus compromising autonomous safety. In this paper, we p…

Cited by 43PDFScholar
2023

Soft Neighbors are Positive Supporters in Contrastive Visual Representation Learning

ICLR 2023poster

Contrastive learning methods train visual encoders by comparing views (e.g., often created via a group of data augmentations on the same instance) from one instance to others. Typically, the views created from one instance are set as positive, while views from other instances are negative. This bina…

Cited by 38SourcePDFScholar
2022

AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentation

NeurIPS 2022accept

Despite the considerable progress in automatic abdominal multi-organ segmentation from CT/MRI scans in recent years, a comprehensive evaluation of the models' capabilities is hampered by the lack of a large-scale benchmark from diverse clinical scenarios. Constraint by the high cost of collecting an…

2022

AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognition

NeurIPS 2022accept

Pretraining Vision Transformers (ViTs) has achieved great success in visual recognition. A following scenario is to adapt a ViT to various image and video recognition tasks. The adaptation is challenging because of heavy computation and memory storage. Each model needs an independent and complete fi…

2022

CycleMLP: A MLP-like Architecture for Dense Prediction

ICLR 2022oral

This paper presents a simple MLP-like architecture, CycleMLP, which is a versatile backbone for visual recognition and dense predictions. As compared to modern MLP architectures, e.g. , MLP-Mixer, ResMLP, and gMLP, whose architectures are correlated to image size and thus are infeasible in object de…

2022

EViT: Expediting Vision Transformers via Token Reorganizations

ICLR 2022spotlight

Vision Transformers (ViTs) take all the image patches as tokens and construct multi-head self-attention (MHSA) among them. Complete leverage of these image tokens brings redundant computations since not all the tokens are attentive in MHSA. Examples include that tokens containing semantically meanin…

2021

Disentangled Cycle Consistency for Highly-Realistic Virtual Try-On

CVPR 2021poster

Image virtual try-on replaces the clothes on a person image with a desired in-shop clothes image. It is challenging because the person and the in-shop clothes are unpaired. Existing methods formulate virtual try-on as either in-painting or cycle consistency. Both of these two formulations encourage…

Cited by 132PDFcodeScholar
2021

Parser-Free Virtual Try-On via Distilling Appearance Flows

CVPR 2021poster

Image virtual try-on aims to fit a garment image (target clothes) to a person image. Prior methods are heavily based on human parsing. However, slightly-wrong segmentation results would lead to unrealistic try-on images with large artifacts. Inaccurate parsing misleads parser-based methods to produc…

Cited by 241PDFcodeScholar
2021

Revitalizing CNN Attention via Transformers in Self-Supervised Visual Representation Learning

NeurIPS 2021poster

Studies on self-supervised visual representation learning (SSL) improve encoder backbones to discriminate training samples without labels. While CNN encoders via SSL achieve comparable recognition performance to those via supervised learning, their network attention is under-explored for further imp…

2021

Watch Only Once: An End-to-End Video Action Detection Framework

ICCV 2021poster

We propose an end-to-end pipeline, named Watch Once Only (WOO), for video action detection. Current methods either decouple video action detection task into separated stages of actor localization and action classification or train two separated models within one stage. In contrast, our approach solv…

Cited by 78PDFScholar