activity
20192026
most citedYou Only Propagate Once: Accelerating Adversarial Training via Maximal Principle

74 citations · 303 across the 37 of their papers we have counts for

collaborators

37 papers

cs.LG2026

Quantifying Explanation Quality in Graph Neural Networks using Out-of-Distribution Generalization

Ding Zhang, Siddharth Betala, Chirag Agarwal

Evaluating the quality of post-hoc explanations for Graph Neural Networks (GNNs) remains a significant challenge. While recent years have seen an increasing development of explaina…

cs.CE2026

Diffusion Large Language Models for Black-Box Optimization

Ye Yuan, Can, Chen +4

Offline black-box optimization (BBO) aims to find optimal designs based solely on an offline dataset of designs and their labels. Such scenarios frequently arise in domains like DN…

cs.CV2025

Generalized Dynamics Generation towards Scannable Physical World Model

Yichen Li, Zhiyi Li, Brandon Feng +2

Digital twin worlds with realistic interactive dynamics presents a new opportunity to develop generalist embodied agents in scannable environments with complex physical behaviors.…

cs.AI2025

Self-Evolving Curriculum for LLM Reasoning

Xiaoyin Chen, Jiarui Lu, Minsu Kim +6

Reinforcement learning (RL) has proven effective for fine-tuning large language models (LLMs), significantly enhancing their reasoning abilities in domains such as mathematics and…

cs.LG2025★ 2 cited

No Trick, No Treat: Pursuits and Challenges Towards Simulation-free Training of Neural Samplers

Jiajun He, Yuanqi Du, Francisco Vargas +5

We consider the sampling problem, where the aim is to draw samples from a distribution whose density is known only up to a normalization constant. Recent breakthroughs in generativ…

cs.LG2024

Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNets

Zhen Liu, Tim Z. Xiao, Weiyang Liu +2

While one commonly trains large diffusion models by collecting datasets on target downstream tasks, it is often desired to align and finetune pretrained diffusion models with some…

cs.LG2024★ 1 cited

Efficient and Robust Continual Graph Learning for Graph Classification in Biology

Ding Zhang, Jane Downer, Can Chen +1

Graph classification is essential for understanding complex biological systems, where molecular structures and interactions are naturally represented as graphs. Traditional graph n…

cs.LG2024★ 6 cited

EnzymeFlow: Generating Reaction-specific Enzyme Catalytic Pockets through Flow Matching and Co-Evolutionary Dynamics

Chenqing Hua, Yong Liu, Dinghuai Zhang +6

Enzyme design is a critical area in biotechnology, with applications ranging from drug development to synthetic biology. Traditional methods for enzyme function prediction or prote…

cs.LG2024

Baking Symmetry into GFlowNets

George Ma, Emmanuel Bengio, Yoshua Bengio +1

GFlowNets have exhibited promising performance in generating diverse candidates with high rewards. These networks generate objects incrementally and aim to learn a policy that assi…

q-bio.PE2023★ 2 cited

PhyloGFN: Phylogenetic inference with generative flow networks

Mingyang Zhou, Zichao Yan, Elliot Layne +5

Phylogenetics is a branch of computational biology that studies the evolutionary relationships among biological entities. Its long history and numerous applications notwithstanding…

cs.LG2023

Learning to Scale Logits for Temperature-Conditional GFlowNets

Minsu Kim, Joohwan Ko, Taeyoung Yun +6

GFlowNets are probabilistic models that sequentially generate compositional structures through a stochastic policy. Among GFlowNets, temperature-conditional GFlowNets can introduce…

cs.LG2023

Local Search GFlowNets

Minsu Kim, Taeyoung Yun, Emmanuel Bengio +4

Generative Flow Networks (GFlowNets) are amortized sampling methods that learn a distribution over discrete objects proportional to their rewards. GFlowNets exhibit a remarkable ab…

cs.LG2023

Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimization

Dinghuai Zhang, Ricky T. Q. Chen, Cheng-Hao Liu +2

We tackle the problem of sampling from intractable high-dimensional density functions, a fundamental task that often appears in machine learning and statistics. We extend recent sa…

cs.LG2023

Delta-AI: Local objectives for amortized inference in sparse graphical models

Jean-Pierre Falet, Hae Beom Lee, Esmeralda S. Whitammer +6

We present a new algorithm for amortized inference in sparse probabilistic graphical models (PGMs), which we call -amortized inference (-AI). Our approach is based on the obs…

cs.AI2023

Cooperation or Competition: Avoiding Player Domination for Multi-Target Robustness via Adaptive Budgets

Yimu Wang, Dinghuai Zhang, Yihan Wu +2

Despite incredible advances, deep learning has been shown to be susceptible to adversarial attacks. Numerous approaches have been proposed to train robust networks both empirically…

cs.LG2023★ 10 cited

Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNets

Dinghuai Zhang, Hanjun Dai, Esmeralda S. Whitammer +3

Combinatorial optimization (CO) problems are often NP-hard and thus out of reach for exact algorithms, making them a tempting domain to apply machine learning methods. The highly s…

cs.LG2023★ 3 cited

Stochastic Generative Flow Networks

Ling Pan, Dinghuai Zhang, Moksh Jain +2

Generative Flow Networks (or GFlowNets for short) are a family of probabilistic agents that learn to sample complex combinatorial structures through the lens of "inference as contr…

cs.LG2023★ 1 cited

Distributional GFlowNets with Quantile Flows

Dinghuai Zhang, Ling Pan, Ricky T. Q. Chen +2

Generative Flow Networks (GFlowNets) are a new family of probabilistic samplers where an agent learns a stochastic policy for generating complex combinatorial structure through a s…

cs.LG2023★ 2 cited

Better Training of GFlowNets with Local Credit and Incomplete Trajectories

Ling Pan, Nikolay Malkin, Dinghuai Zhang +1

Generative Flow Networks or GFlowNets are related to Monte-Carlo Markov chain methods (as they sample from a distribution specified by an energy function), reinforcement learning (…

cs.LG2023★ 6 cited

A theory of continuous generative flow networks

Salem Lahlou, Tristan Deleu, Pablo Lemos +6

Generative flow networks (GFlowNets) are amortized variational inference algorithms that are trained to sample from unnormalized target distributions over compositional objects. A…

cs.LG2022★ 2 cited

Generative Augmented Flow Networks

Ling Pan, Dinghuai Zhang, Aaron Courville +2

The Generative Flow Network is a probabilistic framework where an agent learns a stochastic policy for object generation, such that the probability of generating an object is propo…

cs.LG2022★ 3 cited

GFlowOut: Dropout with Generative Flow Networks

Dianbo Liu, Moksh Jain, Bonaventure Dossou +10

Bayesian Inference offers principled tools to tackle many critical problems with modern neural networks such as poor calibration and generalization, and data inefficiency. However,…

cs.LG2022★ 1 cited

Latent State Marginalization as a Low-cost Approach for Improving Exploration

Dinghuai Zhang, Aaron Courville, Yoshua Bengio +3

While the maximum entropy (MaxEnt) reinforcement learning (RL) framework -- often touted for its exploration and robustness capabilities -- is usually motivated from a probabilisti…

cs.LG2022

GFlowNets and variational inference

Esmeralda S. Whitammer, Salem Lahlou, Tristan Deleu +5

This paper builds bridges between two families of probabilistic algorithms: (hierarchical) variational inference (VI), which is typically used to model distributions over continuou…