activity
20142025
most citedSliced Mutual Information: A Scalable Measure of Statistical Dependence

9 citations · 26 across the 18 of their papers we have counts for

collaborators

18 papers

cs.LG2025

Differentially Private Wasserstein Barycenters

Anming Gu, Sasidhar Kunapuli, Mark Bun +2

The Wasserstein barycenter is defined as the mean of a set of probability measures under the optimal transport metric, and has numerous applications spanning machine learning, stat…

cs.LG2025

Entropic Causal Inference: Graph Identifiability

Spencer Compton, Kristjan Greenewald, Dmitriy Katz +1

Entropic causal inference is a recent framework for learning the causal graph between two variables from observational data by finding the information-theoretically simplest struct…

cs.LG2025

Private Continuous-Time Synthetic Trajectory Generation via Mean-Field Langevin Dynamics

Anming Gu, Edward Chien, Kristjan Greenewald

We provide an algorithm to privately generate continuous-time data (e.g. marginals from stochastic differential equations), which has applications in highly sensitive domains invol…

cs.LG2025

Neural Estimation for Scaling Entropic Multimarginal Optimal Transport

Dor Tsur, Ziv Goldfeld, Kristjan Greenewald +1

Multimarginal optimal transport (MOT) is a powerful framework for modeling interactions between multiple distributions, yet its applicability is bottlenecked by a high computationa…

cs.LG2025

Revisiting Group Relative Policy Optimization: Insights into On-Policy and Off-Policy Training

Youssef Mroueh, Nicolas Dupuis, Brian Belgodere +6

We revisit Group Relative Policy Optimization (GRPO) in both on-policy and off-policy optimization regimes. Our motivation comes from recent work on off-policy Proximal Policy Opti…

cs.CY2025

Fairness Is More Than Algorithms: Racial Disparities in Time-to-Recidivism

Jessy Xinyi Han, Kristjan Greenewald, Devavrat Shah

Racial disparities in recidivism remain a persistent challenge within the criminal justice system, increasingly exacerbated by the adoption of algorithmic risk assessment tools. Pa…