English

Learning Distributions from Multiple Data Providers

Data Structures and Algorithms 2026-07-27 v1 Computer Science and Game Theory Machine Learning Machine Learning

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/\epsilon^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)\Theta(n/\epsilon^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 \Theta(n/\epsilon^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)\alpha \in (1,2), there exists a query family with optimal PAC rate Θ~(nα/ϵ2)\widetilde \Theta(n^\alpha/\epsilon^2).

Cite

@article{arxiv.2607.24732,
  title  = {Learning Distributions from Multiple Data Providers},
  author = {Jon Kleinberg and Amin Saberi and Xizhi Tan and Grigoris Velegkas},
  journal= {arXiv preprint arXiv:2607.24732},
  year   = {2026}
}