← Search

Jianwen Xie

48 accepted papers

2026

Align Your Structures: Generating Trajectories with Structure Pretraining for Molecular Dynamics

ICLR 2026poster

Generating molecular dynamics (MD) trajectories using deep generative models has attracted increasing attention, yet remains inherently challenging due to the limited availability of MD data and the complexities involved in modeling high-dimensional MD distributions. To overcome these challenges, we…

Cited by 0SourcecodeScholar
2026

EdiVal-Agent: An Object-Centric Framework for Automated, Fine-Grained Evaluation of Multi-Turn Editing

ICLR 2026poster

Instruction-based image editing has advanced rapidly, yet reliable and interpretable evaluation remains a bottleneck. Current protocols either (i) depend on paired reference images—resulting in limited coverage and inheriting biases from prior generative models—or (ii) rely *solely* on zero-shot vis…

Cited by 0SourcecodeScholar
2026

In-The-Flow Agentic System Optimization for Effective Planning and Tool Use

ICLR 2026oral

Outcome-driven reinforcement learning has advanced reasoning in large language models (LLMs), but prevailing tool-augmented approaches train a single, monolithic policy that interleaves thoughts and tool calls under full context; this scales poorly with long horizons and diverse tools and generalize…

Cited by 0SourcecodeScholar
2026

Mitigating Non-IID Drift in Zeroth-Order Federated LLM Fine-Tuning with Transferable Sparsity

ICLR 2026poster

Federated Learning enables collaborative fine-tuning of Large Language Models (LLMs) across decentralized Non-Independent and Identically Distributed (Non-IID) clients, but such models' massive parameter sizes lead to significant memory and communication challenges. This work introduces Meerkat, a s…

Cited by 0SourceScholar
2026

PixARMesh: Autoregressive Mesh-Native Single-View Scene Reconstruction

CVPR 2026

We introduce PixARMesh, a method to autoregressively reconstruct complete 3D indoor scene meshes directly from a single RGB image. Unlike prior methods that rely on implicit signed distance fields and post-hoc layout optimization, PixARMesh jointly predicts object layout and geometry within a unifie

Cited by 0SourcecodeScholar
2026

Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspective

ICLR 2026poster

Reinforcement Learning (RL) has proven highly effective for autoregressive language models, but adapting these methods to diffusion large language models (dLLMs) presents fundamental challenges. The core difficulty lies in likelihood approximation: while autoregressive models naturally provide token…

Cited by 0SourcecodeScholar
2026

Randomized Antipodal Search Done Right for Data Pareto Improvement of LLM Unlearning

ICLR 2026poster

Large language models (LLMs) sometimes memorize undesirable knowledge, which must be removed after deployment. Prior work on machine unlearning has focused largely on optimization methods that adjust parameters to enforce forgetting while preserving retention. However, these approaches assume that t…

Cited by 0SourceScholar
2026

To Compress or Not? Pushing the Frontier of Lossless GenAI Model Weights Compression with Exponent Concentration

ICLR 2026poster

The scaling of Generative AI (GenAI) models into the hundreds of billions of parameters makes low-precision computation indispensable for efficient deployment. We argue that the fundamental solution lies in developing low-precision \emph{floating-point} formats, which inherently provide numerical st…

Cited by 0SourcecodeScholar
2026

VideoNSA: Native Sparse Attention Scales Video Understanding

ICLR 2026poster

Video understanding in multimodal language models remains limited by context length: models often miss key transition frames and struggle to maintain coherence across long time scales. To address this, we adapt Native Sparse Attention (NSA) to video-language models. **Our method, VideoNSA, adapts Q…

Cited by 0SourcecodeScholar
2026

``Noisier'’ Noise Contrastive Estimation is (Almost) Maximum Likelihood

ICLR 2026poster

Noise Contrastive Estimation (NCE) has fueled major breakthroughs in representation learning and generative modeling. Yet a long-standing challenge remains: accurately estimating ratios between distributions that differ substantially, which significantly limits the applicability of NCE on modern hig…

Cited by 0SourcecodeScholar
2025

DEL-ToM: Inference-Time Scaling for Theory-of-Mind Reasoning via Dynamic Epistemic Logic

EMNLP 2025

Theory-of-Mind (ToM) tasks pose a unique challenge for large language models (LLMs), which often lack the capability for dynamic logical reasoning. In this work, we propose DEL-ToM, a framework that improves verifiable ToM reasoning through inference-time scaling rather than architectural changes. O

2025

DepR: Depth Guided Single-view Scene Reconstruction with Instance-level Diffusion

ICCV 2025poster

We propose DepR, a depth-guided single-view scene reconstruction framework that integrates instance-level diffusion within a compositional paradigm. Instead of reconstructing the entire scene holistically, DepR generates individual objects and subsequently composes them into a coherent 3D layout. Un…

Cited by 0SourcePDFScholar
2025

Latent Adaptive Planner for Dynamic Manipulation

CoRL 2025poster

This paper presents Latent Adaptive Planner (LAP), a novel approach for dynamic nonprehensile manipulation tasks that formulates planning as latent space inference, effectively learned from human demonstration videos. Our method addresses key challenges in visuomotor policy learning through a p…

Cited by 0SourceScholar
2025

Latent Thought Models with Variational Bayes Inference-Time Computation

ICML 2025poster

We propose a novel class of language models, Latent Thought Models (LTMs), which incorporate explicit latent thought vectors that follow an explicit prior model in latent space. These latent thought vectors guide the autoregressive generation of ground tokens through a Transformer decoder. Training…

2025

OverLayBench: A Benchmark for Layout-to-Image Generation with Dense Overlaps

NeurIPS 2025poster

Despite steady progress in layout-to-image generation, current methods still struggle with layouts containing significant overlap between bounding boxes. We identify two primary challenges: (1) large overlapping regions and (2) overlapping instances with minimal semantic distinction. Through both qu…

Cited by 0SourcecodeScholar
2025

SpatialReasoner: Towards Explicit and Generalizable 3D Spatial Reasoning

NeurIPS 2025poster

Despite recent advances on multi-modal models, 3D spatial reasoning remains a challenging task for state-of-the-art open-source and proprietary models. Recent studies explore data-driven approaches and achieve enhanced spatial reasoning performance by fine-tuning models on 3D-related visual question…

Cited by 0SourceScholar
2025

Tensor Decomposition Networks for Accelerating Machine Learning Force Field Computations

NeurIPS 2025poster

SO(3)-equivariant networks are the dominant models for machine learning interatomic potentials (MLIPs). The key operation of such networks is the Clebsch-Gordan (CG) tensor product, which is computationally expensive. To accelerate the computation, we develop tensor decomposition networks (TDNs) as…

Cited by 0SourcecodeScholar
2025

Word Salad Chopper: Reasoning Models Waste A Ton Of Decoding Budget On Useless Repetitions, Self-Knowingly

EMNLP 2025

Large Reasoning Models (LRMs) are often bottlenecked by the high cost of output tokens. We show that a significant portion of these tokens are useless self-repetitions — what we call “word salad” — that exhaust the decoding budget without adding value. Interestingly, we observe that LRMs are self-aw

Cited by 0SourcePDFScholar
2024

Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inference

NeurIPS 2024poster

In tasks aiming for long-term returns, planning becomes essential. We study generative modeling for planning with datasets repurposed from offline reinforcement learning. Specifically, we identify temporal consistency in the absence of step-wise rewards as one key technical challenge. We introduce t…

2024

Learning Energy-Based Models by Cooperative Diffusion Recovery Likelihood

ICLR 2024spotlight

Training energy-based models (EBMs) on high-dimensional data can be both challenging and time-consuming, and there exists a noticeable gap in sample quality between EBMs and other generative frameworks like GANs and diffusion models. To close this gap, inspired by the recent efforts of learning EBMs…

2024

Molecule Design by Latent Prompt Transformer

NeurIPS 2024spotlight

This work explores the challenging problem of molecule design by framing it as a conditional generative modeling task, where target biological properties or desired chemical constraints serve as conditioning variables. We propose the Latent Prompt Transformer (LPT), a novel generative model comprisi…

Cited by 2SourcePDFScholar
2023

A Tale of Two Latent Flows: Learning Latent Space Normalizing Flow with Short-Run Langevin Flow for Approximate Inference

AAAI 2023technical

We study a normalizing flow in the latent space of a top-down generator model, in which the normalizing flow model plays the role of the informative prior model of the generator. We propose to jointly learn the latent space normalizing flow prior model and the top-down generator model by a Markov ch…

Cited by 6SourcePDFScholar
2023

CoopInit: Initializing Generative Adversarial Networks via Cooperative Learning

AAAI 2023technical

Numerous research efforts have been made to stabilize the training of the Generative Adversarial Networks (GANs), such as through regularization and architecture design. However, we identify the instability can also arise from the fragile balance at the early stage of adversarial learning. This pape…

Cited by 4SourcePDFScholar
2023

Likelihood-Based Generative Radiance Field with Latent Space Energy-Based Model for 3D-Aware Disentangled Image Representation

AISTATS 2023poster

We propose the NeRF-LEBM, a likelihoodbased top-down 3D-aware 2D image generative model that incorporates 3D representation via Neural Radiance Fields (NeRF) and 2D imaging process via differentiable volume rendering. The model represents an image as a rendering process from 3D object to 2D image an…

Cited by 6SourcePDFScholar
2022

A Tale of Two Flows: Cooperative Learning of Langevin Flow and Normalizing Flow Toward Energy-Based Model

ICLR 2022poster

This paper studies the cooperative learning of two generative flow models, in which the two models are iteratively updated based on the jointly synthesized examples. The first flow model is a normalizing flow that transforms an initial simple density to a target density by applying a sequence of inv…

Cited by 54SourcePDFScholar
2022

Blind Image Super-Resolution With Elaborate Degradation Modeling on Noise and Kernel

CVPR 2022poster

While researches on model-based blind single image super-resolution (SISR) have achieved tremendous successes recently, most of them do not consider the image degradation sufficiently. Firstly, they always assume image noise obeys an independent and identically distributed (i.i.d.) Gaussian or Lapla…

Cited by 76PDFcodeScholar
2022

Energy-Based Generative Cooperative Saliency Prediction

AAAI 2022technical

Conventional saliency prediction models typically learn a deterministic mapping from an image to its saliency map, and thus fail to explain the subjective nature of human attention. In this paper, to model the uncertainty of visual saliency, we study the saliency prediction problem from the perspec…

2022

Learning V1 Simple Cells with Vector Representation of Local Content and Matrix Representation of Local Motion

AAAI 2022technical

This paper proposes a representational model for image pairs such as consecutive video frames that are related by local pixel displacements, in the hope that the model may shed light on motion perception in primary visual cortex (V1). The model couples the following two components: (1) the vector re…

Cited by 2SourcePDFScholar
2021

Generative PointNet: Deep Energy-Based Learning on Unordered Point Sets for 3D Generation, Reconstruction and Classification

CVPR 2021poster

We propose a generative model of unordered point sets, such as point clouds, in the forms of an energy-based model, where the energy function is parameterized by an input-permutation-invariant bottom-up neural network. The energy function learns a coordinate encoding of each point and then aggregate…

Cited by 91PDFcodeScholar
2021

Learning Cycle-Consistent Cooperative Networks via Alternating MCMC Teaching for Unsupervised Cross-Domain Translation

AAAI 2021technical

This paper studies the unsupervised cross-domain translation problem by proposing a generative framework, in which the probability distribution of each domain is represented by a generative cooperative network that consists of an energy-based model and a latent variable model. The use of generative…

Cited by 15SourcePDFScholar
2021

Learning Deep Latent Variable Models by Short-Run MCMC Inference With Optimal Transport Correction

CVPR 2021poster

Learning latent variable models with deep top-down architectures typically requires inferring the latent variables for each training example based on the posterior distribution of these latent variables. The inference step typically relies on either time-consuming long run Markov chain Monte Caro (M…

Cited by 14PDFScholar
2021

Learning Energy-Based Generative Models via Coarse-to-Fine Expanding and Sampling

ICLR 2021poster

Energy-based models (EBMs) parameterized by neural networks can be trained by the Markov chain Monte Carlo (MCMC) sampling-based maximum likelihood estimation. Despite the recent significant success of EBMs in image generation, the current approaches to train EBMs are unstable and have difficulty sy…

Cited by 50SourcePDFScholar
2021

Learning Energy-Based Model with Variational Auto-Encoder as Amortized Sampler

AAAI 2021technical

Due to the intractable partition function, training energy-based models (EBMs) by maximum likelihood requires Markov chain Monte Carlo (MCMC) sampling to approximate the gradient of the Kullback-Leibler divergence between data and model distributions. However, it is non-trivial to sample from an EBM…

Cited by 51SourcePDFScholar
2021

Learning Generative Vision Transformer with Energy-Based Latent Space for Saliency Prediction

NeurIPS 2021poster

Vision transformer networks have shown superiority in many computer vision tasks. In this paper, we take a step further by proposing a novel generative vision transformer with latent variables following an informative energy-based prior for salient object detection. Both the vision transformer netwo…

Cited by 110SourcePDFScholar
2021

On Path Integration of Grid Cells: Group Representation and Isotropic Scaling

NeurIPS 2021poster

Understanding how grid cells perform path integration calculations remains a fundamental problem. In this paper, we conduct theoretical analysis of a general representation model of path integration by grid cells, where the 2D self-position is encoded as a higher dimensional vector, and the 2D self-…

2021

Patchwise Generative ConvNet: Training Energy-Based Models From a Single Natural Image for Internal Learning

CVPR 2021poster

Exploiting internal statistics of a single natural image has long been recognized as a significant research paradigm where the goal is to learn the distribution of patches within the image without relying on external training data. Different from prior works that model such distributions implicitly…

Cited by 24PDFScholar
2020

Learning Noise-Aware Encoder-Decoder from Noisy Labels by Alternating Back-Propagation for Saliency Detection

ECCV 2020poster

In this paper, we propose a noise-aware encoder-decoder framework to disentangle a clean saliency predictor from noisy training examples, where the noisy labels are generated by unsupervised handcrafted feature-based methods. The proposed model consists of two sub-models parameterized by neural netw…

Cited by 60SourcePDFScholar
2019

Learning Feature-to-Feature Translator by Alternating Back-Propagation for Generative Zero-Shot Learning

ICCV 2019poster

We investigate learning feature-to-feature translator networks by alternating back-propagation as a general-purpose solution to zero-shot learning (ZSL) problems. It is a generative model-based ZSL framework. In contrast to models based on generative adversarial networks (GAN) or variational autoenc…

Cited by 127PDFcodeScholar
2019

Learning Grid Cells as Vector Representation of Self-Position Coupled with Matrix Representation of Self-Motion

ICLR 2019poster

This paper proposes a representational model for grid cells. In this model, the 2D self-position of the agent is represented by a high-dimensional vector, and the 2D self-motion or displacement of the agent is represented by a matrix that transforms the vector. Each component of the vector is a unit…

2019

Semantic-Guided Multi-Attention Localization for Zero-Shot Learning

NeurIPS 2019poster

Zero-shot learning extends the conventional object classification to the unseen class recognition by introducing semantic representations of classes. Existing approaches predominantly focus on learning the proper mapping function for visual-semantic embedding, while neglecting the effect of learning…

Cited by 180SourcePDFScholar
2018

A Causal And-Or Graph Model for Visibility Fluent Reasoning in Tracking Interacting Objects

CVPR 2018poster

Tracking humans that are interacting with the other subjects or environment remains unsolved in visual tracking, because the visibility of the human of interests in videos is unknown and might vary over time. In particular, it is still difficult for state-of-the-art human trackers to recover complet…

Cited by 36SourcePDFScholar
2018

Learning Descriptor Networks for 3D Shape Synthesis and Analysis

CVPR 2018poster

This paper proposes a 3D shape descriptor network, which is a deep convolutional energy-based model, for modeling volumetric shape patterns. The maximum likelihood training of the model follows an "analysis by synthesis" scheme and can be interpreted as a mode seeking and mode shifting process. The…

2018

Weakly and Semi Supervised Human Body Part Parsing via Pose-Guided Knowledge Transfer

CVPR 2018poster

Human body part parsing, or human semantic part segmentation, is fundamental to many computer vision tasks. In conventional semantic segmentation methods, the ground truth segmentations are provided, and fully convolutional networks (FCN) are trained in an end-to-end scheme. Although these methods h…