← Search

Zonghao Chen

10 accepted papers

2026

(De)-regularized Maximum Mean Discrepancy Gradient Flow

ICML 2026poster

We introduce a (de)-regularization of the Maximum Mean Discrepancy (DrMMD) and its Wasserstein gradient flow. Existing gradient flows that transport samples from source distribution to target distribution with only target samples, either lack tractable numerical implementation ($f$-divergence flows)…

Cited by 0SourcecodeScholar
2026

From Interaction Trajectories to Prompt Rules: Credit Assignment for Multi-Agent Prompt Optimization

ICML 2026poster

Large language model (LLM)-based multi-agent systems commonly rely on natural-language prompts to specify agent behavior, yet optimizing these prompts remains challenging when agent roles and interaction structures are fixed by design. In such systems, behaviors emerge over long, noisy interaction t…

Cited by 0SourceScholar
2026

TOPOGRAPH: Topology-Preserving Graph Reduction with Adaptive Structure for Persistent Homology

AAAI 2026technical

Topological Data Analysis (TDA) provides artificial intelligence (AI) systems with mathematically rigorous geometric descriptors through Persistent Homology (PH), capturing essential shape characteristics in high-dimensional data. Yet, PH’s combinatorial complexity and sensitivity to outliers hinder

Cited by 0SourcePDFScholar
2026

Thinned Mean Field Langevin Dynamics

ICML 2026poster

Several important learning tasks can be formulated as minimizing an entropy-regularized objective over an appropriate space of probability distributions. Mean-field Langevin dynamics (MFLD) facilitate computation in this general context, casting the minimizer as the invariant distribution of a McKea…

Cited by 0SourceScholar
2022

Probabilistic Bilevel Coreset Selection

ICML 2022spotlight

The goal of coreset selection in supervised learning is to produce a weighted subset of data, so that training only on the subset achieves similar performance as training on the entire dataset. Existing methods achieved promising results in resource-constrained scenarios such as continual learning a…

Cited by 40SourcePDFScholar
2022

Tractable Function-Space Variational Inference in Bayesian Neural Networks

NeurIPS 2022accept

Reliable predictive uncertainty estimation plays an important role in enabling the deployment of neural networks to safety-critical settings. A popular approach for estimating the predictive uncertainty of neural networks is to define a prior distribution over the network parameters, infer an approx…

2021

Efficient Neural Network Training via Forward and Backward Propagation Sparsification

NeurIPS 2021poster

Sparse training is a natural idea to accelerate the training speed of deep neural networks and save the memory usage, especially since large modern neural networks are significantly over-parameterized. However, most of the existing methods cannot achieve this goal in practice because the chain rule…