Collaborative Learning with Different Labeling Functions
Yuyang Deng, Mingda Qiao
TL;DR
This work studies learning $n$ classifiers across $n$ distributions under different labeling functions, aiming to minimize the total labeled data while achieving $\epsilon$-accuracy on each distribution. It introduces $(k,\epsilon)$-realizability and an $(G,k)$-augmentation of the hypothesis class to enable ERM-based learning with a VC-dimension bound, yielding a near-optimal sample complexity of $O(kd\log(n/k) + n\log n)$ (for fixed constants). The paper proves NP-hardness of ERM over the augmented class for $k\ge3$ and provides two computationally efficient special cases: identical marginals and $2$-refutable hypothesis classes via approximate coloring, including a bipartite ($k=2$) scenario with favorable complexity. These results delineate when collaborative learning with heterogeneous labeling is statistically feasible and when computational barriers necessitate structure-based algorithms, with practical implications for federated, multi-task, and distributed learning contexts.
Abstract
We study a variant of Collaborative PAC Learning, in which we aim to learn an accurate classifier for each of the $n$ data distributions, while minimizing the number of samples drawn from them in total. Unlike in the usual collaborative learning setup, it is not assumed that there exists a single classifier that is simultaneously accurate for all distributions. We show that, when the data distributions satisfy a weaker realizability assumption, which appeared in [Crammer and Mansour, 2012] in the context of multi-task learning, sample-efficient learning is still feasible. We give a learning algorithm based on Empirical Risk Minimization (ERM) on a natural augmentation of the hypothesis class, and the analysis relies on an upper bound on the VC dimension of this augmented class. In terms of the computational efficiency, we show that ERM on the augmented hypothesis class is NP-hard, which gives evidence against the existence of computationally efficient learners in general. On the positive side, for two special cases, we give learners that are both sample- and computationally-efficient.
