Towards the Fundamental Limits of Knowledge Transfer over Finite Domains
This work addresses the fundamental limits of knowledge transfer in machine learning, providing theoretical insights and optimal methods for scenarios with varying levels of teacher information, which is incremental but offers precise bounds for domain-specific applications.
The paper tackles the problem of characterizing the statistical efficiency of knowledge transfer from a teacher to a student classifier over finite domains, showing that privileged information at three levels accelerates transfer, with convergence rates improving from √(|S||A|/n) to |S|/n as more information is provided.
We characterize the statistical efficiency of knowledge transfer through $n$ samples from a teacher to a probabilistic student classifier with input space $\mathcal S$ over labels $\mathcal A$. We show that privileged information at three progressive levels accelerates the transfer. At the first level, only samples with hard labels are known, via which the maximum likelihood estimator attains the minimax rate $\sqrt{{|{\mathcal S}||{\mathcal A}|}/{n}}$. The second level has the teacher probabilities of sampled labels available in addition, which turns out to boost the convergence rate lower bound to ${{|{\mathcal S}||{\mathcal A}|}/{n}}$. However, under this second data acquisition protocol, minimizing a naive adaptation of the cross-entropy loss results in an asymptotically biased student. We overcome this limitation and achieve the fundamental limit by using a novel empirical variant of the squared error logit loss. The third level further equips the student with the soft labels (complete logits) on ${\mathcal A}$ given every sampled input, thereby provably enables the student to enjoy a rate ${|{\mathcal S}|}/{n}$ free of $|{\mathcal A}|$. We find any Kullback-Leibler divergence minimizer to be optimal in the last case. Numerical simulations distinguish the four learners and corroborate our theory.