TensorX
返回文献探索

Paper · arXiv 2410.01680

PHI-S: Distribution Balancing for Label-Free Multi-Teacher Distillation

Mike Ranzinger, Jon Barker, Greg Heinrich, Pavlo Molchanov, Bryan Catanzaro, Andrew Tao

34 upvotesOctober 2, 2024arXiv 预印本
AI 摘要

Heterogeneous multi-teacher knowledge distillation improves visual foundation models by optimizing activation statistics and alignment, with PHI Standardization shown to achieve the best student model quality.

heterogeneous multi-teacher knowledge distillationactivation statisticsloss functionstatistical normalization techniquesHadamard matricesisotropic standardizationPHI Standardization

Abstract

Various visual foundation models have distinct strengths and weaknesses, both of which can be improved through heterogeneous multi-teacher knowledge distillation without labels, termed "agglomerative models." We build upon this body of work by studying the effect of the teachers' activation statistics, particularly the impact of the loss function on the resulting student model quality. We explore a standard toolkit of statistical normalization techniques to better align the different distributions and assess their effects. Further, we examine the impact on downstream teacher-matching metrics, which motivates the use of Hadamard matrices. With these matrices, we demonstrate useful properties, showing how they can be used for isotropic standardization, where each dimension of a multivariate distribution is standardized using the same scale. We call this technique "PHI Standardization" (PHI-S) and empirically demonstrate that it produces the best student model across the suite of methods studied.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
PHI-S: Distribution Balancing for Label-Free Multi-Teacher Distillation | TensorX