cs.DSJul 27, 2026

Learning Distributions from Multiple Data Providers

Authors: Jon KleinbergAmin SaberiXizhi TanGrigoris Velegkas

Organizations: Cornell University · Stanford University · Google Research

Abstract

Motivated by learning from heterogeneous and overlapping data providers, we study a stylized model of distribution learning from restricted conditional samples. The goal is to learn an unknown distribution pp on a finite domain [n][n]. The learner is given a fixed family of queryable sets S2[n]\mathscr{S} \subseteq 2^{[n]}, and each query to SSS \in \mathscr{S} returns an independent sample from the conditional distribution p(S)p(\cdot \mid S). Learnability is governed by the co-occurrence graph associated with S\mathscr{S}: two domain elements are adjacent if they appear together in some queryable set. Pointwise consistency is achievable when this graph is connected on the target support. PAC learning requires more: it is possible when the co-occurrence graph is complete. The optimal sample complexity of PAC learning ranges from nearly linear to quadratic. Every query family with complete co-occurrence graph admits sample complexity O~(n2/ε2)\widetilde O(n^2/ε^2), and this bound is tight in the worst case. On the other hand, if [n][n] is queryable then ordinary sampling improves the bound to Θ(n/ε2)Θ(n/ε^2), and this cannot be improved further even if every set is queryable. More generally, we identify hierarchical comparabilityas a sufficient structural condition on S\mathscr S under which the optimal complexity is nearly linear, Θ~(n/ε2)\widetilde Θ(n/ε^2), with pairwise query families as a canonical example. Finally, the full range of polynomial rates between linear and quadratic is attainable: for every α(1,2)α\in (1,2), there exists a query family with optimal PAC rate Θ~(nα/ε2)\widetilde Θ(n^α/ε^2).

Explore similar work

CardsList