← Search

Jiatao Gu

47 accepted papers

2026

Continuously Augmented Discrete Diffusion model for Categorical Generative Modeling

ICLR 2026poster

Standard discrete diffusion models treat all unobserved states the same way, typically mapping them to an absorbing [MASK] token. This creates an "information void" where global semantic information that may be inferred for the masked tokens from the unmasked tokens is not directly passed from one d…

Cited by 0SourcecodeScholar
2026

DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation

ICLR 2026poster

Diffusion large language models (dLLMs) are compelling alternatives to autoregressive (AR) models because their denoising models operate over the entire sequence. The global planning and iterative refinement features of dLLMs are particularly useful for code generation. However, current training and…

Cited by 0SourcecodeScholar
2026

ROVER: Benchmarking Reciprocal Cross-Modal Reasoning for Omnimodal Generation

ICLR 2026poster

Unified multimodal models (UMMs) have shown remarkable advances in jointly understanding and generating text and images. However, prevailing evaluations treat these abilities in isolation, such that tasks with multimodal inputs and outputs are scored primarily through unimodal reasoning: textual ben…

Cited by 0SourcecodeScholar
2026

STARFlow-V: End-to-End Video Generative Modeling with Autoregressive Normalizing Flows

CVPR 2026

Normalizing flows (NFs) are end-to-end likelihood-based generative models for continuous data, and have recently regained attention with encouraging progress on image generation. Yet in the video generation domain, where spatiotemporal complexity and computational cost are substantially higher, stat

Cited by 0SourcecodeScholar
2025

Denoising Autoregressive Transformers for Scalable Text-to-Image Generation

ICLR 2025poster

Diffusion models have become the dominant approach for visual generation. They are trained by denoising a Markovian process which gradually adds noise to the input. We argue that the Markovian property limits the model’s ability to fully utilize the generation trajectory, leading to inefficiencies d…

Cited by 0SourcePDFScholar
2025

Flexible Language Modeling in Continuous Space with Transformer-based Autoregressive Flows

NeurIPS 2025poster

Autoregressive models have driven remarkable progress in language modeling. Their foundational reliance on discrete tokens, unidirectional context, and single-pass decoding, while central to their success, also inspires the exploration of a design space that could offer new axes of modeling flexibil…

Cited by 0SourceScholar
2025

Normalizing Flows are Capable Generative Models

ICML 2025oral

Normalizing Flows (NFs) are likelihood-based models for continuous inputs. They have demonstrated promising results on both density estimation and generative modeling tasks, but have received relatively little attention in recent years. In this work, we demonstrate that NFs are more powerful than pr…

2025

PhysCtrl: Generative Physics for Controllable and Physics-Grounded Video Generation

NeurIPS 2025poster

Existing video generation models excel at producing photo-realistic videos from text or images, but often lack physical plausibility and 3D controllability. To overcome these limitations, we introduce PhysCtrl, a novel framework for physics-grounded image-to-video generation with physical parameters…

Cited by 0SourceScholar
2025

STARFlow: Scaling Latent Normalizing Flows for High-resolution Image Synthesis

NeurIPS 2025spotlight

We present STARFlow, a scalable generative model based on normalizing flows that achieves strong performance on high-resolution image synthesis. STARFlow's main building block is Transformer Autoregressive Flow (TARFlow), which combines normalizing flows with Autoregressive Transformer architecture…

Cited by 0SourcecodeScholar
2025

TADA: Improved Diffusion Sampling with Training-free Augmented DynAmics

NeurIPS 2025poster

Diffusion models have demonstrated exceptional capabilities in generating high-fidelity images but typically suffer from inefficient sampling. Many solver designs and noise scheduling strategies have been proposed to dramatically improve sampling speeds. In this paper, we introduce a new sampling m…

Cited by 0SourcecodeScholar
2025

Vid2Sim: Generalizable, Video-based Reconstruction of Appearance, Geometry and Physics for Mesh-free Simulation

CVPR 2025poster

Faithfully reconstructing textured shapes and physical properties from videos presents an intriguing yet challenging problem. Significant efforts have been dedicated to advancing such a system identification problem in this area. Previous methods often rely on heavy optimization pipelines with a dif…

Cited by 0SourcePDFScholar
2025

World-consistent Video Diffusion with Explicit 3D Modeling

CVPR 2025highlight

Recent advancements in diffusion models have set new benchmarks in image and video generation, enabling realistic visual synthesis across single- and multi-frame contexts. However, these models still struggle with efficiently and explicitly generating 3D-consistent content. To address this, we propo…

Cited by 7SourcePDFScholar
2024

Data-free Distillation of Diffusion Models with Bootstrapping

ICML 2024poster

Diffusion models have demonstrated great potential for generating diverse images. However, their performance often suffers from slow generation due to iterative denoising. Knowledge distillation has been recently proposed as a remedy which can reduce the number of inference steps to one or a few, wi…

Cited by 2SourcePDFScholar
2024

Divide-or-Conquer? Which Part Should You Distill Your LLM?

EMNLP 2024finding

Recent methods have demonstrated that Large Language Models (LLMs) can solve reasoning tasks better when they are encouraged to solve subtasks of the main task first. In this paper we devise a similar strategy that breaks down reasoning tasks into a problem decomposition phase and a problem solving…

2024

Generative Modeling with Phase Stochastic Bridge

ICLR 2024oral

Diffusion models (DMs) represent state-of-the-art generative models for continuous inputs. DMs work by constructing a Stochastic Differential Equation (SDE) in the input space (ie, position space), and using a neural network to reverse it. In this work, we introduce a novel generative modeling frame…

Cited by 14SourcePDFScholar
2024

Kaleido Diffusion: Improving Conditional Diffusion Models with Autoregressive Latent Modeling

NeurIPS 2024poster

Diffusion models have emerged as a powerful tool for generating high-quality images from textual descriptions. Despite their successes, these models often exhibit limited diversity in the sampled images, particularly when sampling with a high classifier-free guidance weight. To address this issue, w…

Cited by 8SourcePDFScholar
2023

Diffusion Probabilistic Fields

ICLR 2023poster

Diffusion probabilistic models have quickly become a major approach for generative modeling of images, 3D geometry, video and other domains. However, to adapt diffusion generative modeling to these domains the denoising network needs to be carefully designed for each domain independently, oftentimes…

Cited by 28SourcePDFScholar
2023

MAST: Masked Augmentation Subspace Training for Generalizable Self-Supervised Priors

ICLR 2023poster

Recent Self-Supervised Learning (SSL) methods are able to learn feature representations that are invariant to different data augmentations, which can then be transferred to downstream tasks of interest. However, different downstream tasks require different invariances for their best performance, so…

Cited by 11SourcePDFScholar
2023

NerfDiff: Single-image View Synthesis with NeRF-guided Distillation from 3D-aware Diffusion

ICML 2023poster

Novel view synthesis from a single image requires inferring occluded regions of objects and scenes whilst simultaneously maintaining semantic and physical consistency with the input. Existing approaches condition neural radiance fields (NeRF) on local image features, projecting points to the input i…

Cited by 182SourcePDFScholar
2023

PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Model

NeurIPS 2023poster

Autoregressive models for text sometimes generate repetitive and low-quality output because errors accumulate during the steps of generation. This issue is often attributed to exposure bias -- the difference between how a model is trained, and how it is used during inference. Denoising diffusion mod…

2023

Single-Stage Diffusion NeRF: A Unified Approach to 3D Generation and Reconstruction

ICCV 2023poster

3D-aware image synthesis encompasses a variety of tasks, such as scene generation and novel view synthesis from images. Despite numerous task-specific methods, developing a comprehensive model remains challenging. In this paper, we present SSDNeRF, a unified approach that employs an expressive diffu…

Cited by 156PDFcodeScholar
2023

Stabilizing Transformer Training by Preventing Attention Entropy Collapse

ICML 2023poster

Training stability is of great importance to Transformers. In this work, we investigate the training dynamics of Transformers by examining the evolution of the attention layers. In particular, we track the attention entropy for each attention head during the course of training, which is a proxy for…

2023

f-DM: A Multi-stage Diffusion Model via Progressive Signal Transformation

ICLR 2023poster

Diffusion models (DMs) have recently emerged as SoTA tools for generative modeling in various domains. Standard DMs can be viewed as an instantiation of hierarchical variational autoencoders (VAEs) where the latent variables are inferred from input-centered Gaussian distributions with fixed scales a…

Cited by 28SourcePDFScholar
2022

Detection, Disambiguation, Re-ranking: Autoregressive Entity Linking as a Multi-Task Problem

ACL 2022findings

We propose an autoregressive entity linking model, that is trained with two auxiliary tasks, and learns to re-rank generated samples at inference time. Our proposed novelties address two weaknesses in the literature. First, a recent method proposes to learn mention detection and then entity candidat…

Cited by 0SourcePDFScholar
2022

Direct Speech-to-Speech Translation With Discrete Units

ACL 2022long

We present a direct speech-to-speech translation (S2ST) model that translates speech from one language to speech in another language without relying on intermediate text generation. We tackle the problem by first applying a self-supervised discrete speech encoder on the target speech and then traini…

2022

IDPG: An Instance-Dependent Prompt Generation Method

NAACL 2022long

Prompt tuning is a new, efficient NLP transfer learning paradigm that adds a task-specific prompt in each input instance during the model training stage. It freezes the pre-trained language model and only optimizes a few task-specific prompts. In this paper, we propose a conditional prompt generatio…

Cited by 59SourcePDFScholar
2022

StyleNeRF: A Style-based 3D Aware Generator for High-resolution Image Synthesis

ICLR 2022poster

We propose StyleNeRF, a 3D-aware generative model for photo-realistic high-resolution image synthesis with high multi-view consistency, which can be trained on unstructured 2D images. Existing approaches either cannot synthesize high-resolution images with fine details or yield clearly noticeable 3…

2022

Textless Speech-to-Speech Translation on Real Data

NAACL 2022long

We present a textless speech-to-speech translation (S2ST) system that can translate speech from one language into another language and can be built without the need of any text data. Different from existing work in the literature, we tackle the challenge in modeling multi-speaker target speech and t…

Cited by 157SourcePDFScholar
2022

Unified Speech-Text Pre-training for Speech Translation and Recognition

ACL 2022long

In this work, we describe a method to jointly pre-train speech and text in an encoder-decoder modeling framework for speech translation and recognition. The proposed method utilizes multi-task learning to integrate four self-supervised and supervised subtasks for cross modality learning. A self-supe…

2022

data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language

ICML 2022oral

While the general idea of self-supervised learning is identical across modalities, the actual algorithms and objectives differ widely because they were developed with a single modality in mind. To get us closer to general self-supervised learning, we present data2vec, a framework that uses the same…

2021

Lightweight Adapter Tuning for Multilingual Speech Translation

ACL 2021short

Adapter modules were recently introduced as an efficient alternative to fine-tuning in NLP. Adapter tuning consists in freezing pre-trained parameters of a model and injecting lightweight modules between layers, resulting in the addition of only a small number of task-specific trainable parameters.…

2021

fairseq Sˆ2: A Scalable and Integrable Speech Synthesis Toolkit

EMNLP 2021system demonstrations

This paper presents fairseq Sˆ2, a fairseq extension for speech synthesis. We implement a number of autoregressive (AR) and non-AR text-to-speech models, and their multi-speaker variants. To enable training speech synthesis models with less curated data, a number of preprocessing tools are built and…

2020

Cross-lingual Retrieval for Iterative Self-Supervised Training

NeurIPS 2020spotlight

Recent studies have demonstrated the cross-lingual alignment ability of multilingual pretrained language models. In this work, we found that the cross-lingual alignment can be further improved by training seq2seq models on sentence pairs mined using their own encoder outputs. We utilized these findi…

2020

Dual-decoder Transformer for Joint Automatic Speech Recognition and Multilingual Speech Translation

COLING 2020main

We introduce dual-decoder Transformer, a new model architecture that jointly performs automatic speech recognition (ASR) and multilingual speech translation (ST). Our models are based on the original Transformer architecture (Vaswani et al., 2017) but consist of two decoders, each responsible for on…

2020

Neural Sparse Voxel Fields

NeurIPS 2020spotlight

Photo-realistic free-viewpoint rendering of real-world scenes using classical computer graphics techniques is challenging, because it requires the difficult step of capturing detailed appearance and geometry models. Recent studies have demonstrated promising results by learning scene representations…

2020

Non-autoregressive Machine Translation with Disentangled Context Transformer

ICML 2020poster

State-of-the-art neural machine translation models generate a translation from left to right and every step is conditioned on the previously generated tokens. The sequential nature of this generation process causes fundamental latency in inference since we cannot generate multiple tokens in each sen…

2020

PointContrast: Unsupervised Pre-training for 3D Point Cloud Understanding

ECCV 2020poster

Arguably one of the top success stories of deep learning is transfer learning. The finding that pre-training a network on a rich source set (g, ImageNet) can help boost performance once fine-tuned on a usually much smaller target set, has been instrumental to many applications in language and vision…

2020

Revisiting Self-Training for Neural Sequence Generation

ICLR 2020poster

Self-training is one of the earliest and simplest semi-supervised methods. The key idea is to augment the original labeled dataset with unlabeled data paired with the model's prediction (i.e. the pseudo-parallel data). While self-training has been extensively studied on classification problems, in c…

Cited by 282SourcecodeScholar
2020

Understanding Knowledge Distillation in Non-autoregressive Machine Translation

ICLR 2020poster

Non-autoregressive machine translation (NAT) systems predict a sequence of output tokens in parallel, achieving substantial improvements in generation speed compared to autoregressive models. Existing NAT models usually rely on the technique of knowledge distillation, which creates the training data…

Cited by 245SourceScholar
2018

Non-Autoregressive Neural Machine Translation

ICLR 2018poster

Existing approaches to neural machine translation condition each output word on previously generated outputs. We introduce a model that avoids this autoregressive property and produces its outputs in parallel, allowing an order of magnitude lower latency during inference. Through knowledge distillat…