← Search

Mingyuan Zhou

128 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

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

Mitigating Reward Hacking in RLHF via Bayesian Non-negative Reward Modeling

ICML 2026oral

Reward models learned from human preferences are central to aligning large language models (LLMs) via reinforcement learning from human feedback, yet they are often vulnerable to reward hacking due to noisy annotations and systematic biases such as response length or style. We propose Bayesian Non-N…

Cited by 0SourceScholar
2026

REAL: Regression-Aware Reinforcement Learning for LLM-as-a-Judge

ICML 2026poster

Large language models (LLMs) are increasingly deployed as automated evaluators that assign numeric scores to model outputs, a paradigm known as LLM-as-a-Judge. However, standard Reinforcement Learning (RL) methods typically rely on binary rewards (e.g., 0-1 accuracy), thereby ignoring the ordinal st…

Cited by 0SourceScholar
2026

Score Distillation Beyond Acceleration: Generative Modeling from Corrupted Data

ICLR 2026poster

Learning generative models directly from corrupted observations is a long-standing challenge across natural and scientific domains. We introduce *Distillation from Corrupted Data (DCD)*, a unified framework for learning high-fidelity, one-step generative models using **only** degraded data of the fo…

Cited by 0SourcecodeScholar
2025

Adversarial Score identity Distillation: Rapidly Surpassing the Teacher in One Step

ICLR 2025poster

Score identity Distillation (SiD) is a data-free method that has achieved state-of-the-art performance in image generation by leveraging only a pretrained diffusion model, without requiring any training data. However, the ultimate performance of SiD is constrained by the accuracy with which the pret…

2025

DRL: Decomposed Representation Learning for Tabular Anomaly Detection

ICLR 2025poster

Anomaly detection, indicating to identify the anomalies that significantly deviate from the majority normal instances of data, has been an important role in machine learning and related applications. Despite the significant success achieved in anomaly detection on image and text data, the accurate T…

Cited by 0SourcePDFScholar
2025

Enhancing Uncertainty Estimation and Interpretability with Bayesian Non-negative Decision Layer

ICLR 2025poster

Although deep neural networks have demonstrated significant success due to their powerful expressiveness, most models struggle to meet practical requirements for uncertainty estimation. Concurrently, the entangled nature of deep neural net- works leads to a multifaceted problem, where various locali…

Cited by 0SourcePDFScholar
2025

FedAWA: Adaptive Optimization of Aggregation Weights in Federated Learning Using Client Vectors

CVPR 2025poster

Federated Learning (FL) has emerged as a promising framework for distributed machine learning, enabling collaborative model training without sharing local data, thereby preserving privacy and enhancing security. However, data heterogeneity resulting from differences across user behaviors, preference…

2025

Generative Data Augmentation via Diffusion Distillation, Adversarial Alignment, and Importance Reweighting

NeurIPS 2025poster

Generative data augmentation (GDA) leverages generative models to enrich training sets with entirely new samples drawn from the modeled data distribution to achieve performance gains. However, the usage of the mighty contemporary diffusion models in GDA remains impractical: *i)* their thousand-step…

Cited by 0SourceScholar
2025

Generative Model Inversion Through the Lens of the Manifold Hypothesis

NeurIPS 2025poster

Model inversion attacks (MIAs) aim to reconstruct class-representative samples from trained models. Recent generative MIAs utilize generative adversarial networks to learn image priors that guide the inversion process, yielding reconstructions with high visual quality and strong fidelity to the priv…

Cited by 0SourceScholar
2025

Guided Score identity Distillation for Data-Free One-Step Text-to-Image Generation

ICLR 2025poster

Diffusion-based text-to-image generation models trained on extensive text-image pairs have demonstrated the ability to produce photorealistic images aligned with textual descriptions. However, a significant limitation of these models is their slow sample generation process, which requires iterative…

2025

Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replay

NeurIPS 2025poster

Reinforcement learning (RL) has become an effective approach for fine-tuning large language models (LLMs), particularly to enhance their reasoning capabilities. However, RL fine-tuning remains highly resource-intensive, and existing work has largely overlooked the problem of data efficiency. In this…

Cited by 0SourcecodeScholar
2025

KodCode: A Diverse, Challenging, and Verifiable Synthetic Dataset for Coding

ACL 2025finding

We introduce KodCode, a synthetic dataset that addresses the persistent challenge of acquiring high-quality, verifiable training data across diverse difficulties and domains for training Large Language Models for coding. Existing code-focused resources typically fail to ensure either the breadth of…

2025

OmiAD: One-Step Adaptive Masked Diffusion Model for Multi-class Anomaly Detection via Adversarial Distillation

ICML 2025poster

Diffusion models have demonstrated outstanding performance in industrial anomaly detection. However, their iterative denoising nature results in slow inference speed, limiting their practicality for real-time industrial deployment. To address this challenge, we propose OmiAD, a one-step masked diffu…

Cited by 0SourcePDFScholar
2025

One-Step Diffusion Policy: Fast Visuomotor Policies via Diffusion Distillation

ICML 2025poster

Diffusion models, praised for their success in generative tasks, are increasingly being applied to robotics, demonstrating exceptional performance in behavior cloning. However, their slow generation process stemming from iterative denoising steps poses a challenge for real-time applications in resou…

Cited by 11SourcePDFScholar
2025

Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Models

ICLR 2025poster

The machine learning community is increasingly recognizing the importance of fostering trust and safety in modern generative AI (GenAI) models. We posit machine unlearning (MU) as a crucial foundation for developing safe, secure, and trustworthy GenAI models. Traditional MU methods often rely on str…

2025

Self Forcing: Bridging the Train-Test Gap in Autoregressive Video Diffusion

NeurIPS 2025spotlight

We introduce Self Forcing, a novel training paradigm for autoregressive video diffusion models. It addresses the longstanding issue of exposure bias, where models trained on ground-truth context must generate sequences conditioned on their own imperfect outputs during inference. Unlike prior methods…

Cited by 0SourceScholar
2024

A Dense Reward View on Aligning Text-to-Image Diffusion with Preference

ICML 2024poster

Aligning text-to-image diffusion model (T2I) with preference has been gaining increasing research attention. While prior works exist on directly optimizing T2I by preference data, these methods are developed under the bandit assumption of a latent reward on the entire diffusion reverse chain, while…

2024

Diffusion Policies Creating a Trust Region for Offline Reinforcement Learning

NeurIPS 2024poster

Offline reinforcement learning (RL) leverages pre-collected datasets to train optimal policies. Diffusion Q-Learning (DQL), introducing diffusion models as a powerful and expressive policy class, significantly boosts the performance of offline RL. However, its reliance on iterative denoising samplin…

2024

Improving Unsupervised Hierarchical Representation with Reinforcement Learning

CVPR 2024poster

Learning representations to capture the very fundamental understanding of the world is a key challenge in machine learning. The hierarchical structure of explanatory factors hidden in data is such a general representation and could be potentially achieved with a hierarchical VAE. However training a…

2024

Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modeling

ICLR 2024poster

Diffusion models excel at generating photo-realistic images but come with significant computational costs in both training and sampling. While various techniques address these computational challenges, a less-explored issue is designing an efficient and adaptable network backbone for iterative refin…

2024

Long-tailed Diffusion Models with Oriented Calibration

ICLR 2024poster

Diffusion models are acclaimed for generating high-quality and diverse images. However, their performance notably degrades when trained on data with a long-tailed distribution. For long tail diffusion model generation, current works focus on the calibration and enhancement of the tail generation wit…

2024

OmniMotionGPT: Animal Motion Generation with Limited Data

CVPR 2024poster

Our paper aims to generate diverse and realistic animal motion sequences from textual descriptions without a large-scale animal text-motion dataset. While the task of text-driven human motion synthesis is already extensively studied and benchmarked it remains challenging to transfer this success to…

Cited by 7SourcePDFScholar
2024

Patch-Prompt Aligned Bayesian Prompt Tuning for Vision-Language Models

UAI 2024poster

For downstream applications of vision-language pre-trained models, there has been significant interest in constructing effective prompts. Existing works on prompt engineering, which either require laborious manual designs or optimize the prompt tuning as a point estimation problem, may fail to descr…

Cited by 3SourcePDFScholar
2024

Score identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generation

ICML 2024poster

We introduce Score identity Distillation (SiD), an innovative data-free method that distills the generative capabilities of pretrained diffusion models into a single-step generator. SiD not only facilitates an exponentially fast reduction in Fréchet inception distance (FID) during distillation but a…

2024

Switchable Decision: Dynamic Neural Generation Networks

ICML 2024poster

Auto-regressive generation models achieve competitive performance across many different NLP tasks such as summarization, question answering, and classifications. However, they are also known for being slow in inference, which makes them challenging to deploy in real-time applications. We propose a s…

Cited by 0SourcePDFScholar
2024

Transformer-Modulated Diffusion Models for Probabilistic Multivariate Time Series Forecasting

ICLR 2024poster

Transformers have gained widespread usage in multivariate time series (MTS) forecasting, delivering impressive performance. Nonetheless, these existing transformer-based methods often neglect an essential aspect: the incorporation of uncertainty into the predicted series, which holds significant val…

Cited by 21SourcePDFScholar
2024

UltrAvatar: A Realistic Animatable 3D Avatar Diffusion Model with Authenticity Guided Textures

CVPR 2024poster

Recent advances in 3D avatar generation have gained significant attention. These breakthroughs aim to produce more realistic animatable avatars narrowing the gap between virtual and real-world experiences. Most of existing works employ Score Distillation Sampling (SDS) loss combined with a different…

Cited by 4SourcePDFScholar
2024

Vague Prototype-Oriented Diffusion Model for Multi-Class Anomaly Detection

ICML 2024poster

Multi-class unsupervised anomaly detection aims to create a unified model for identifying anomalies in objects from multiple classes when only normal data is available. In such a challenging setting, widely used reconstruction-based networks persistently grapple with the "identical shortcut" problem…

Cited by 3SourcePDFScholar
2023

Bayesian Progressive Deep Topic Model with Knowledge Informed Textual Data Coarsening Process

ICML 2023poster

Deep topic models have shown an impressive ability to extract multi-layer document latent representations and discover hierarchical semantically meaningful topics.However, most deep topic models are limited to the single-step generative process, despite the fact that the progressive generative proce…

Cited by 6SourcePDFScholar
2023

Context-guided Embedding Adaptation for Effective Topic Modeling in Low-Resource Regimes

NeurIPS 2023poster

Embedding-based neural topic models have turned out to be a superior option for low-resourced topic modeling. However, current approaches consider static word embeddings learnt from source tasks as general knowledge that can be transferred directly to the target task, discounting the dynamically cha…

2023

DR2: Diffusion-Based Robust Degradation Remover for Blind Face Restoration

CVPR 2023poster

Blind face restoration usually synthesizes degraded low-quality data with a pre-defined degradation model for training, while more complex cases could happen in the real world. This gap between the assumed and actual degradation hurts the restoration performance where artifacts are often observed in…

2023

Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning

ICLR 2023poster

Offline reinforcement learning (RL), which aims to learn an optimal policy using a previously collected static dataset, is an important paradigm of RL. Standard RL methods often perform poorly in this regime due to the function approximation errors on out-of-distribution actions. While a variety of…

2023

Diffusion-GAN: Training GANs with Diffusion

ICLR 2023poster

Generative adversarial networks (GANs) are challenging to train stably, and a promising remedy of injecting instance noise into the discriminator input has not been very effective in practice. In this paper, we propose Diffusion-GAN, a novel GAN framework that leverages a forward diffusion chain to…

2023

Fantastic Rewards and How to Tame Them: A Case Study on Reward Learning for Task-oriented Dialogue Systems

ICLR 2023poster

When learning task-oriented dialogue (ToD) agents, reinforcement learning (RL) techniques can naturally be utilized to train dialogue strategies to achieve user-specific goals. Prior works mainly focus on adopting advanced RL techniques to train the ToD agents, while the design of the reward functio…

2023

Few-shot Generation via Recalling Brain-Inspired Episodic-Semantic Memory

NeurIPS 2023poster

Aimed at adapting a generative model to a novel generation task with only a few given data samples, the capability of few-shot generation is crucial for many real-world applications with limited data, \emph{e.g.}, artistic domains. Instead of training from scratch, recent works tend to leverage the…

Cited by 6SourcePDFScholar
2023

In-Context Learning Unlocked for Diffusion Models

NeurIPS 2023spotlight

We present Prompt Diffusion, a framework for enabling in-context learning in diffusion-based generative models. Given a pair of task-specific example images, such as depth from/to image and scribble from/to image, and a text guidance, our model automatically understands the underlying task and perfo…

2023

Learning to Jump: Thinning and Thickening Latent Counts for Generative Modeling

ICML 2023poster

Learning to denoise has emerged as a prominent paradigm to design state-of-the-art deep generative models for natural images. How to use it to model the distributions of both continuous real-valued data and categorical data has been well studied in recently proposed diffusion models. However, it is…

2023

POUF: Prompt-Oriented Unsupervised Fine-tuning for Large Pre-trained Models

ICML 2023poster

Through prompting, large-scale pre-trained models have become more expressive and powerful, gaining significant attention in recent years. Though these big models have zero-shot capabilities, in general, labeled data are still required to adapt them to downstream tasks. To overcome this critical lim…

2023

Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Models

NeurIPS 2023poster

Diffusion models are powerful, but they require a lot of time and data to train. We propose Patch Diffusion, a generic patch-wise training framework, to significantly reduce the training time costs while improving data efficiency, which thus helps democratize diffusion model training to broader user…

2023

PatchCT: Aligning Patch Set and Label Set with Conditional Transport for Multi-Label Image Classification

ICCV 2023poster

Multi-label image classification is a prediction task that aims to identify more than one label from a given image. This paper considers the semantic consistency of the latent space between the visual patch and linguistic label domains and introduces the conditional transport (CT) theory to bridge t…

Cited by 22PDFcodeScholar
2023

Preference-grounded Token-level Guidance for Language Model Fine-tuning

NeurIPS 2023poster

Aligning language models (LMs) with preferences is an important problem in natural language generation. A key challenge is that preferences are typically provided at the *sequence level* while LM training and generation both occur at the *token level*. There is, therefore, a *granularity mismatch* b…

2023

Probabilistic Conformal Prediction Using Conditional Random Samples

AISTATS 2023poster

This paper proposes probabilistic conformal prediction (PCP), a predictive inference algorithm that estimates a target variable by a discontinuous predictive set. Given inputs, PCP constructs the predictive set based on random samples from an estimated generative model. It is efficient and compatibl…

2023

Prototype-oriented unsupervised anomaly detection for multivariate time series

ICML 2023poster

Unsupervised anomaly detection (UAD) of multivariate time series (MTS) aims to learn robust representations of normal multivariate temporal patterns. Existing UAD methods try to learn a fixed set of mappings for each MTS, entailing expensive computation and limited model adaptation. To address this…

Cited by 26SourcePDFScholar
2023

Truncated Diffusion Probabilistic Models and Diffusion-based Adversarial Auto-Encoders

ICLR 2023poster

Employing a forward diffusion chain to gradually map the data to a noise distribution, diffusion-based generative models learn how to generate the data by inferring a reverse diffusion chain. However, this approach is slow and costly because it needs many forward and reverse steps. We propose a fas…

2022

A Unified Framework for Alternating Offline Model Training and Policy Learning

NeurIPS 2022accept

In offline model-based reinforcement learning (offline MBRL), we learn a dynamic model from historically collected data, and subsequently utilize the learned model and fixed datasets for policy learning, without further interacting with the environment. Offline MBRL algorithms can improve the effici…

2022

A Variational Edge Partition Model for Supervised Graph Representation Learning

NeurIPS 2022accept

Graph neural networks (GNNs), which propagate the node features through the edges and learn how to transform the aggregated features under label supervision, have achieved great success in supervised feature extraction for both node-level and graph-level classification tasks. However, GNNs typicall…

2022

ALLSH: Active Learning Guided by Local Sensitivity and Hardness

NAACL 2022findings

Active learning, which effectively collects informative unlabeled data for annotation, reduces the demand for labeled data. In this work, we propose to retrieve unlabeled samples with a local sensitivity and hardness-aware acquisition function. The proposed method generates data copies through local…

Cited by 39SourcePDFScholar
2022

Adaptive Distribution Calibration for Few-Shot Learning with Hierarchical Optimal Transport

NeurIPS 2022accept

Few-shot classification aims to learn a classifier to recognize unseen classes during training, where the learned model can easily become over-fitted based on the biased distribution formed by only a few training examples. A recent solution to this problem is calibrating the distribution of these fe…

Cited by 31SourcePDFScholar
2022

Alleviating "Posterior Collapse'' in Deep Topic Models via Policy Gradient

NeurIPS 2022accept

Deep topic models have been proven as a promising way to extract hierarchical latent representations from documents represented as high-dimensional bag-of-words vectors. However, the representation capability of existing deep topic models is still limited by the phenomenon of "posterior collapse", w…

Cited by 10SourcePDFScholar
2022

Bayesian Deep Embedding Topic Meta-Learner

ICML 2022spotlight

Existing deep topic models are effective in capturing the latent semantic structures in textual data but usually rely on a plethora of documents. This is less than satisfactory in practical applications when only a limited amount of data is available. In this paper, we propose a novel framework that…

Cited by 6SourcePDFScholar
2022

Deep Variational Graph Convolutional Recurrent Network for Multivariate Time Series Anomaly Detection

ICML 2022spotlight

Anomaly detection within multivariate time series (MTS) is an essential task in both data mining and service quality management. Many recent works on anomaly detection focus on designing unsupervised probabilistic models to extract robust normal patterns of MTS. In this paper, we model sensor depend…

2022

HyperMiner: Topic Taxonomy Mining with Hyperbolic Embedding

NeurIPS 2022accept

Embedded topic models are able to learn interpretable topics even with large and heavy-tailed vocabularies. However, they generally hold the Euclidean embedding space assumption, leading to a basic limitation in capturing hierarchical relations. To this end, we present a novel framework that introdu…

2022

Knowledge-Aware Bayesian Deep Topic Model

NeurIPS 2022accept

We propose a Bayesian generative model for incorporating prior domain knowledge into hierarchical topic modeling. Although embedded topic models (ETMs) and its variants have gained promising performance in text analysis, they mainly focus on mining word co-occurrence patterns, ignoring potentially e…

2022

Learning Prototype-oriented Set Representations for Meta-Learning

ICLR 2022poster

Learning from set-structured data is a fundamental problem that has recently attracted increasing attention, where a series of summary networks are introduced to deal with the set input. In fact, many meta-learning problems can be treated as set-input tasks. Most existing summary networks aim to des…

Cited by 26SourcePDFScholar
2022

Learning to Re-weight Examples with Optimal Transport for Imbalanced Classification

NeurIPS 2022accept

Imbalanced data pose challenges for deep learning based classification models. One of the most widely-used approaches for tackling imbalanced data is re-weighting, where training samples are associated with different weights in the loss function. Most of existing re-weighting approaches treat the ex…

2022

Meta Discovery: Learning to Discover Novel Classes given Very Limited Data

ICLR 2022spotlight

In novel class discovery (NCD), we are given labeled data from seen classes and unlabeled data from unseen classes, and we train clustering models for the unseen classes. However, the implicit assumptions behind NCD are still unclear. In this paper, we demystify assumptions behind NCD and find that…

2022

Regularizing a Model-based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning (RL) extends the paradigm of classical RL algorithms to purely learning from static datasets, without interacting with the underlying environment during the learning process. A key challenge of offline RL is the instability of policy training, caused by the mismatch be…

2022

Representing Mixtures of Word Embeddings with Mixtures of Topic Embeddings

ICLR 2022poster

A topic model is often formulated as a generative model that explains how each word of a document is generated given a set of topics and document-specific topic proportions. It is focused on capturing the word co-occurrences in a document and hence often suffers from poor performance in analyzing s…

2021

A Prototype-Oriented Framework for Unsupervised Domain Adaptation

NeurIPS 2021poster

Existing methods for unsupervised domain adaptation often rely on minimizing some statistical distance between the source and target samples in the latent space. To avoid the sampling variability, class imbalance, and data-privacy concerns that often plague these methods, we instead provide a memory…

2021

ARMS: Antithetic-REINFORCE-Multi-Sample Gradient for Binary Variables

ICML 2021spotlight

Estimating the gradients for binary variables is a task that arises frequently in various domains, such as training discrete latent variable models. What has been commonly used is a REINFORCE based Monte Carlo estimation method that uses either independent samples or pairs of negatively correlated s…

2021

Adversarially Adaptive Normalization for Single Domain Generalization

CVPR 2021poster

Single domain generalization aims to learn a model that performs well on many unseen domains with only one domain data for training. Existing works focus on studying the adversarial domain augmentation (ADA) to improve the model's generalization capability. The impact on domain generalization from t…

Cited by 161PDFScholar
2021

Alignment Attention by Matching Key and Query Distributions

NeurIPS 2021poster

The neural attention mechanism has been incorporated into deep neural networks to achieve state-of-the-art performance in various domains. Most such models use multi-head self-attention which is appealing for the ability to attend to information from different perspectives. This paper introduces ali…

2021

CARMS: Categorical-Antithetic-REINFORCE Multi-Sample Gradient Estimator

NeurIPS 2021poster

Accurately backpropagating the gradient through categorical variables is a challenging task that arises in various domains, such as training discrete latent variable models. To this end, we propose CARMS, an unbiased estimator for categorical random variables based on multiple mutually negatively co…

2021

Contextual Dropout: An Efficient Sample-Dependent Dropout Module

ICLR 2021poster

Dropout has been demonstrated as a simple and effective module to not only regularize the training process of deep neural networks, but also provide the uncertainty estimation for prediction. However, the quality of uncertainty estimation is highly dependent on the dropout probabilities. Most curren…

2021

EnsLM: Ensemble Language Model for Data Diversity by Semantic Clustering

ACL 2021long

Natural language processing (NLP) often faces the problem of data diversity such as different domains, themes, styles, and so on. Therefore, a single language model (LM) is insufficient to learn all knowledge from diverse samples. To solve this problem, we firstly propose an autoencoding topic model…

2021

Exploiting Chain Rule and Bayes' Theorem to Compare Probability Distributions

NeurIPS 2021poster

To measure the difference between two probability distributions, referred to as the source and target, respectively, we exploit both the chain rule and Bayes' theorem to construct conditional transport (CT), which is constituted by both a forward component and a backward one. The forward CT is the e…

2021

Hyperbolic graph embedding with enhanced semi-implicit variational inference.

AISTATS 2021poster

Efficient modeling of relational data arising in physical, social, and information sciences is challenging due to complicated dependencies within the data. In this work we build off of semi-implicit graph variational auto-encoders to capture higher order statistics in a low-dimensional graph latent…

2021

Probabilistic Margins for Instance Reweighting in Adversarial Training

NeurIPS 2021poster

Reweighting adversarial data during training has been recently shown to improve adversarial robustness, where data closer to the current decision boundaries are regarded as more critical and given larger weights. However, existing methods measuring the closeness are not very reliable: they are discr…

2021

Sawtooth Factorial Topic Embeddings Guided Gamma Belief Network

ICML 2021spotlight

Hierarchical topic models such as the gamma belief network (GBN) have delivered promising results in mining multi-layer document representations and discovering interpretable topic taxonomies. However, they often assume in the prior that the topics at each layer are independently drawn from the Diri…

2021

TopicNet: Semantic Graph-Guided Topic Discovery

NeurIPS 2021poster

Existing deep hierarchical topic models are able to extract semantically meaningful topics from a text corpus in an unsupervised manner and automatically organize them into a topic hierarchy. However, it is unclear how to incorporate prior belief such as knowledge graph to guide the learning of th…

2020

Adaptive Correlated Monte Carlo for Contextual Categorical Sequence Generation

ICLR 2020poster

Sequence generation models are commonly refined with reinforcement learning over user-defined metrics. However, high gradient variance hinders the practical use of this method. To stabilize this method, we adapt to contextual generation of categorical sequences a policy gradient estimator, which eva…

Cited by 4SourcecodeScholar
2020

Arsm Gradient Estimator for Supervised Learning to Rank

ICASSP 2020accepted

We propose a new model for supervised learning to rank. In our model, the relevance labels are assumed to follow a categorical distribution whose probabilities are constructed based on a scoring function. We optimize the training objective with respect to the multivariate categorical variables with…

Cited by 0SourceScholar
2020

Bayesian Graph Neural Networks with Adaptive Connection Sampling

ICML 2020poster

We propose a unified framework for adaptive connection sampling in graph neural networks (GNNs) that generalizes existing stochastic regularization methods for training GNNs. The proposed framework not only alleviates over-smoothing and over-fitting tendencies of deep GNNs, but also enables learning…

Cited by 160SourcePDFScholar
2020

Bidirectional Convolutional Poisson Gamma Dynamical Systems

NeurIPS 2020poster

Incorporating the natural document-sentence-word structure into hierarchical Bayesian modeling, we propose convolutional Poisson gamma dynamical systems (PGDS) that introduce not only word-level probabilistic convolutions, but also sentence-level stochastic temporal transitions. With word-level conv…

2020

Deep Relational Topic Modeling via Graph Poisson Gamma Belief Network

NeurIPS 2020poster

To analyze a collection of interconnected documents, relational topic models (RTMs) have been developed to describe both the link structure and document content, exploring their underlying relationships via a single-layer latent representation with limited expressive capability. To better utilize th…

2020

Discrete Action On-Policy Learning with Action-Value Critic

AISTATS 2020poster

Reinforcement learning (RL) in discrete action space is ubiquitous in real-world applications, but its complexity grows exponentially with the action-space dimension, making it challenging to apply existing on-policy gradient based deep RL algorithms efficiently. To effectively operate in multidimen…

2020

Learnable Bernoulli Dropout for Bayesian Deep Learning

AISTATS 2020poster

In this work, we propose learnable Bernoulli dropout (LBD), a new model-agnostic dropout scheme that considers the dropout rates as parameters jointly optimized with other model parameters. By probabilistic modeling of Bernoulli dropout, our method enables more robust prediction and uncertainty quan…

Cited by 53SourcePDFScholar
2020

Learning Dynamic Hierarchical Topic Graph with Graph Convolutional Network for Document Classification

AISTATS 2020poster

Constructing a graph with graph convolutional network (GCN) to explore the relational structure of the data has attracted lots of interests in various tasks. However, for document classification, existing graph based methods often focus on the straightforward word-word and word-document relations,…

2020

Meta-Learning without Memorization

ICLR 2020spotlight

The ability to learn new concepts with small amounts of data is a critical aspect of intelligence that has proven challenging for deep learning methods. Meta-learning has emerged as a promising technique for leveraging data from previous tasks to enable efficient learning of new tasks. However, most…

Cited by 249SourcecodeScholar
2020

Mutual Information Gradient Estimation for Representation Learning

ICLR 2020poster

Mutual Information (MI) plays an important role in representation learning. However, MI is unfortunately intractable in continuous and high-dimensional settings. Recent advances establish tractable and scalable MI estimators to discover useful representation. However, most of the existing methods ar…

Cited by 28SourceScholar
2020

Pairwise Supervised Hashing with Bernoulli Variational Auto-Encoder and Self-Control Gradient Estimator

UAI 2020poster

Semantic hashing has become a crucial component of fast similarity search in many large-scale information retrieval systems, in particular, for text data. Variational auto-encoders (VAEs) with binary latent variables as hashing codes provide state-of-the-art performance in terms of precision for doc…

Cited by 26SourcePDFScholar
2020

Recurrent Hierarchical Topic-Guided RNN for Language Generation

ICML 2020poster

To simultaneously capture syntax and global semantics from a text corpus, we propose a new larger-context recurrent neural network (RNN) based language model, which extracts recurrent hierarchical semantic structure via a dynamic deep topic model to guide natural language generation. Moving beyond a…

2020

Semi-Implicit Stochastic Recurrent Neural Networks

ICASSP 2020accepted

Stochastic recurrent neural networks with latent random variables of complex dependency structures have shown to be more successful in modeling sequential data than deterministic deep models. However, the majority of existing methods have limited expressive power due to the Gaussian assumption of la…

Cited by 0SourceScholar
2020

Variational Autoencoders for Sparse and Overdispersed Discrete Data

AISTATS 2020poster

Many applications, such as text modelling, high-throughput sequencing, and recommender systems, require analysing sparse, high-dimensional, and overdispersed discrete (count or binary) data. Recent deep probabilistic models based on variational autoencoders (VAE) have shown promising results on disc…

2020

Variational Hetero-Encoder Randomized GANs for Joint Image-Text Modeling

ICLR 2020poster

For bidirectional joint image-text modeling, we develop variational hetero-encoder (VHE) randomized generative adversarial network (GAN), a versatile deep generative model that integrates a probabilistic text decoder, probabilistic image encoder, and GAN into a coherent end-to-end multi-modality lea…

Cited by 1SourcecodeScholar
2019

ARSM: Augment-REINFORCE-Swap-Merge Estimator for Gradient Backpropagation Through Categorical Variables

ICML 2019oral

To address the challenge of backpropagating the gradient through categorical variables, we propose the augment-REINFORCE-swap-merge (ARSM) gradient estimator that is unbiased and has low variance. ARSM first uses variable augmentation, REINFORCE, and Rao-Blackwellization to re-express the gradient a…

2019

Locally Private Bayesian Inference for Count Models

ICML 2019oral

We present a general and modular method for privacy-preserving Bayesian inference for Poisson factorization, a broad class of models that includes some of the most widely used models in the social sciences. Our method satisfies limited-precision local privacy, a generalization of local differential…

Cited by 43SourcePDFScholar
2019

Poisson-Randomized Gamma Dynamical Systems

NeurIPS 2019poster

This paper presents the Poisson-randomized gamma dynamical system (PRGDS), a model for sequentially observed count tensors that encodes a strong inductive bias toward sparsity and burstiness. The PRGDS is based on a new motif in Bayesian latent variable modeling, an alternating chain of discrete Poi…

2019

Semi-Implicit Graph Variational Auto-Encoders

NeurIPS 2019poster

Semi-implicit graph variational auto-encoder (SIG-VAE) is proposed to expand the flexibility of variational graph auto-encoders (VGAE) to model graph data. SIG-VAE employs a hierarchical variational framework to enable neighboring node sharing for better generative modeling of graph dependency struc…

2019

Variational Graph Recurrent Neural Networks

NeurIPS 2019poster

Representation learning over graph structured data has been mostly studied in static graph settings while efforts for modeling dynamic graphs are still scant. In this paper, we develop a novel hierarchical variational model that introduces additional latent random variables to jointly model the hidd…

2018

Bayesian multi-domain learning for cancer subtype discovery from next-generation sequencing count data

NeurIPS 2018poster

Precision medicine aims for personalized prognosis and therapeutics by utilizing recent genome-scale high-throughput profiling techniques, including next-generation sequencing (NGS). However, translating NGS data faces several challenges. First, NGS count data are often overdispersed, requiring appr…

Cited by 79SourcePDFScholar
2018

Inter and Intra Topic Structure Learning with Word Embeddings

ICML 2018oral

One important task of topic modeling for text analysis is interpretability. By discovering structured topics one is able to yield improved interpretability as well as modeling accuracy. In this paper, we propose a novel topic model with a deep structure that explores both inter-topic and intra-topic…

2018

Masking: A New Perspective of Noisy Supervision

NeurIPS 2018poster

It is important to learn various types of classifiers given training data with noisy labels. Noisy labels, in the most popular noise model hitherto, are corrupted from ground-truth labels by an unknown noise transition matrix. Thus, by estimating this matrix, classifiers can escape from overfitting…

2018

Nonparametric Bayesian Lomax delegate racing for survival analysis with competing risks

NeurIPS 2018poster

We propose Lomax delegate racing (LDR) to explicitly model the mechanism of survival under competing risks and to interpret how the covariates accelerate or decelerate the time to event. LDR explains non-monotonic covariate effects by racing a potentially infinite number of sub-risks, and consequent…

2018

WHAI: Weibull Hybrid Autoencoding Inference for Deep Topic Modeling

ICLR 2018poster

To train an inference network jointly with a deep generative topic model, making it both scalable to big corpora and fast in out-of-sample prediction, we develop Weibull hybrid autoencoding inference (WHAI) for deep latent Dirichlet allocation, which infers posterior samples via a hybrid of stochast…

Cited by 126SourcePDFScholar
2017

Deep Latent Dirichlet Allocation with Topic-Layer-Adaptive Stochastic Gradient Riemannian MCMC

ICML 2017poster

It is challenging to develop stochastic gradient based scalable inference for deep discrete latent variable models (LVMs), due to the difficulties in not only computing the gradients, but also adapting the step sizes to different latent factors and hidden layers. For the Poisson gamma belief network…

Cited by 71SourcePDFScholar
2016

Bayesian Poisson Tucker Decomposition for Learning the Structure of International Relations

ICML 2016poster

We introduce Bayesian Poisson Tucker decomposition (BPTD) for modeling country–country interaction event data. These data consist of interaction events of the form “country i took action a toward country j at time t.” BPTD discovers overlapping country–community memberships, including the number of…

Cited by 105SourcePDFScholar