← Search

Jiajun He

13 accepted papers

2026

Accelerated Parallel Tempering via Neural Transports

ICLR 2026poster

Markov Chain Monte Carlo (MCMC) algorithms are essential tools in computational statistics for sampling from unnormalised probability distributions, but can be fragile when targeting high-dimensional, multimodal, or complex target distributions. Parallel Tempering (PT) enhances MCMC's sample efficie…

Cited by 0SourceScholar
2026

CREPE: Controlling diffusion with REPlica Exchange

ICLR 2026poster

Inference-time control of diffusion models aims to steer model outputs to satisfy new constraints without retraining. Previous approaches have mostly relied on heuristic guidance or have been coupled with Sequential Monte Carlo (SMC) for bias correction. In this paper, we propose a flexible alternat…

Cited by 0SourcecodeScholar
2026

RNE: plug-and-play diffusion inference-time control and energy-based training

ICLR 2026poster

Diffusion models generate data by removing noise gradually, which corresponds to the time-reversal of a noising process. However, access to only the denoising kernels is often insufficient. In many applications, we need the knowledge of the marginal densities along the generation trajectory, which e…

Cited by 0SourceScholar
2026

Towards Diverse Scientific Hypothesis Search with Large Language Models

ICML 2026poster

Large language models are increasingly used to accelerate scientific discovery, especially in iteratively searching scientific hypotheses. Yet in many discovery settings the goal is not to identify a single ``best'' hypothesis: validation is noisy and expensive, multiple hypotheses can remain plausi…

Cited by 0SourceScholar
2026

Vector Quantization using Gaussian Variational Autoencoder

ICML 2026poster

Vector-quantized variational autoencoders (VQ-VAEs) are discrete autoencoders that compress images into discrete tokens. However, they are difficult to train due to discretization. In this paper, we propose a simple yet effective technique dubbed __Gaussian Quant (GQ)__, which first trains a Gaussia…

Cited by 0SourceScholar
2026

Vision in One Vector: Implicit Visual Compression with Diffusion Foundation Models

ICML 2026poster

Modern visual generative models acquire rich visual knowledge through large-scale training, yet existing visual representations (such as pixels, latents, or tokens) remain external to the model and cannot directly exploit this knowledge for compact storage or reuse. In this work, we introduce a new …

Cited by 0SourceScholar
2025

FEAT: Free energy Estimators with Adaptive Transport

NeurIPS 2025poster

We present Free energy Estimators with Adaptive Transport (FEAT), a novel framework for free energy estimation---a critical challenge across scientific domains. FEAT leverages learned transports implemented via stochastic interpolants and provides consistent, minimum-variance estimators based on esc…

Cited by 0SourcecodeScholar
2025

Progressive Tempering Sampler with Diffusion

ICML 2025poster

Recent research has focused on designing neural samplers that amortize the process of sampling from unnormalized densities. However, despite significant advancements, they still fall short of the state-of-the-art MCMC approach, Parallel Tempering (PT), when it comes to the efficiency of target eval…

2025

Training Neural Samplers with Reverse Diffusive KL Divergence

AISTATS 2025poster

Training generative models to sample from unnormalized density functions is an important and challenging task in machine learning. Traditional training methods often rely on the reverse Kullback-Leibler (KL) divergence due to its tractability. However, the mode-seeking behavior of reverse KL hinder…

Cited by 0SourcecodeScholar
2024

Accelerating Relative Entropy Coding with Space Partitioning

NeurIPS 2024poster

Relative entropy coding (REC) algorithms encode a random sample following a target distribution $Q$, using a coding distribution $P$ shared between the sender and receiver. Sadly, general REC algorithms suffer from prohibitive encoding times, at least on the order of $2^{D_{\text{KL}}[Q||P]}$, and…

Cited by 1SourcePDFScholar
2024

MF-AED-AEC: Speech Emotion Recognition by Leveraging Multimodal Fusion, Asr Error Detection, and Asr Error Correction

ICASSP 2024accepted

The prevalent approach in speech emotion recognition (SER) involves integrating both audio and textual information to comprehensively identify the speaker’s emotion, with the text generally obtained through automatic speech recognition (ASR). An essential issue of this approach is that ASR errors fr…

Cited by 0SourceScholar
2024

RECOMBINER: Robust and Enhanced Compression with Bayesian Implicit Neural Representations

ICLR 2024poster

COMpression with Bayesian Implicit NEural Representations (COMBINER) is a recent data compression method that addresses a key inefficiency of previous Implicit Neural Representation (INR)-based approaches: it avoids quantization and enables direct optimization of the rate-distortion performance. How…

2023

Compression with Bayesian Implicit Neural Representations

NeurIPS 2023spotlight

Many common types of data can be represented as functions that map coordinates to signal values, such as pixel locations to RGB values in the case of an image. Based on this view, data can be compressed by overfitting a compact neural network to its functional representation and then encoding the ne…