Towards a Unified Representation Evaluation Framework Beyond Downstream Tasks
This addresses the need for better representation evaluation in self-supervised learning and foundation models, though it is incremental as it builds on existing attempts without a new paradigm.
The paper tackles the problem that downstream probing overlooks key representation attributes like equivariance and disentanglement, and introduces a unified evaluation framework to quantify these, finding that models with similar downstream performance can differ substantially in these attributes.
Downstream probing has been the dominant method for evaluating model representations, an important process given the increasing prominence of self-supervised learning and foundation models. However, downstream probing primarily assesses the availability of task-relevant information in the model's latent space, overlooking attributes such as equivariance, invariance, and disentanglement, which contribute to the interpretability, adaptability, and utility of representations in real-world applications. While some attempts have been made to measure these qualities in representations, no unified evaluation framework with modular, generalizable, and interpretable metrics exists. In this paper, we argue for the importance of representation evaluation beyond downstream probing. We introduce a standardized protocol to quantify informativeness, equivariance, invariance, and disentanglement of factors of variation in model representations. We use it to evaluate representations from a variety of models in the image and speech domains using different architectures and pretraining approaches on identified controllable factors of variation. We find that representations from models with similar downstream performance can behave substantially differently with regard to these attributes. This hints that the respective mechanisms underlying their downstream performance are functionally different, prompting new research directions to understand and improve representations.