← Search

Mingtian Zhang

12 accepted papers

2025

Improving Probabilistic Diffusion Models With Optimal Diagonal Covariance Matching

ICLR 2025oral

The probabilistic diffusion model has become highly effective across various domains. Typically, sampling from a diffusion model involves using a denoising distribution characterized by a Gaussian with a learned mean and either fixed or learned covariances. In this paper, we leverage the recently pr…

Cited by 0SourcePDFScholar
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

Constructing Semantics-Aware Adversarial Examples with a Probabilistic Perspective

NeurIPS 2024poster

We propose a probabilistic perspective on adversarial examples, allowing us to embed subjective understanding of semantics as a distribution into the process of generating adversarial examples, in a principled manner. Despite significant pixel-level modifications compared to traditional adversarial…

2024

Diffusive Gibbs Sampling

ICML 2024poster

The inadequate mixing of conventional Markov Chain Monte Carlo (MCMC) methods for multi-modal distributions presents a significant challenge in practical applications such as Bayesian inference and molecular dynamics. Addressing this, we propose Diffusive Gibbs Sampling (DiGS), an innovative family…

2024

Reparameterized Multi-Resolution Convolutions for Long Sequence Modelling

NeurIPS 2024poster

Global convolutions have shown increasing promise as powerful general-purpose sequence models. However, training long convolutions is challenging, and kernel parameterizations must be able to learn long-range dependencies without overfitting. This work introduces reparameterized multi-resolution con…

Cited by 1SourcePDFScholar
2023

Moment Matching Denoising Gibbs Sampling

NeurIPS 2023poster

Energy-Based Models (EBMs) offer a versatile framework for modelling complex data distributions. However, training and sampling from EBMs continue to pose significant challenges. The widely-used Denoising Score Matching (DSM) method for scalable EBM training suffers from inconsistency issues, causin…

2021

AFEC: Active Forgetting of Negative Transfer in Continual Learning

NeurIPS 2021poster

Continual learning aims to learn a sequence of tasks from dynamic data distributions. Without accessing to the old training samples, knowledge transfer from the old tasks to each new task is difficult to determine, which might be either positive or negative. If the old knowledge interferes with the…

2021

On the Out-of-distribution Generalization of Probabilistic Image Modelling

NeurIPS 2021poster

Out-of-distribution (OOD) detection and lossless compression constitute two problems that can be solved by the training of probabilistic models on a first dataset with subsequent likelihood evaluation on a second dataset, where data distributions differ. By defining the generalization of probabilist…