Showing 2024Show all
3 papers · 1 filter
cs.LG2024
Supervised Kernel Thinning
Albert Gong, Kyuseong Choi, Raaz Dwivedi
The kernel thinning algorithm of Dwivedi & Mackey (2024) provides a better-than-i.i.d. compression of a generic set of points. By generating high-fidelity coresets of size signific…
stat.ML2024
Learning Counterfactual Distributions via Kernel Nearest Neighbors
Kyuseong Choi, Jacob Feitelberg, Caleb Chin +2
Consider a setting with multiple units (e.g., individuals, cohorts, geographic locations) and outcomes (e.g., treatments, times, items), where the goal is to learn a multivariate d…
stat.ML2024
Distributional Matrix Completion via Nearest Neighbors in the Wasserstein Space
Jacob Feitelberg, Kyuseong Choi, Anish Agarwal +1
We study the problem of distributional matrix completion: Given a sparsely observed matrix of empirical distributions, we seek to impute the true distributions associated with both…