← Search

Jia-Jie Zhu

8 accepted papers

2024

Interaction-Force Transport Gradient Flows

NeurIPS 2024poster

This paper presents a new gradient flow dissipation geometry over non-negative and probability measures. This is motivated by a principled construction that combines the unbalanced optimal transport and interaction forces modeled by reproducing kernels. Using a precise connection between the Helling…

2023

Estimation Beyond Data Reweighting: Kernel Method of Moments

ICML 2023poster

Moment restrictions and their conditional counterparts emerge in many areas of machine learning and statistics ranging from causal inference to reinforcement learning. Estimators for these tasks, generally called methods of moments, include the prominent generalized method of moments (GMM) which has…

2022

Adversarially Robust Kernel Smoothing

AISTATS 2022poster

We propose a scalable robust learning algorithm combining kernel smoothing and robust optimization. Our method is motivated by the convex analysis perspective of distributionally robust optimization based on probability metrics, such as the Wasserstein distance and the maximum mean discrepancy. We a…

2022

Functional Generalized Empirical Likelihood Estimation for Conditional Moment Restrictions

ICML 2022spotlight

Important problems in causal inference, economics, and, more generally, robust machine learning can be expressed as conditional moment restrictions, but estimation becomes challenging as it requires solving a continuum of unconditional moment restrictions. Previous works addressed this problem by ex…

2021

Kernel Distributionally Robust Optimization: Generalized Duality Theorem and Stochastic Approximation

AISTATS 2021poster

We propose kernel distributionally robust optimization (Kernel DRO) using insights from the robust optimization theory and functional analysis. Our method uses reproducing kernel Hilbert spaces (RKHS) to construct a wide range of convex ambiguity sets, which can be generalized to sets based on integ…