SynBench: Task-Agnostic Benchmarking of Pretrained Representations using Synthetic Data
This provides a method for researchers and practitioners to benchmark pretrained models more efficiently, though it is incremental as it builds on existing representation learning paradigms.
The paper tackles the problem of evaluating pretrained representations without relying on downstream tasks by introducing SynBench, a task-agnostic benchmarking framework using synthetic data from a Gaussian mixture model, which quantifies robustness-accuracy trade-offs and shows experimental results matching linear probing performance on downstream tasks.
Recent success in fine-tuning large models, that are pretrained on broad data at scale, on downstream tasks has led to a significant paradigm shift in deep learning, from task-centric model design to task-agnostic representation learning and task-specific fine-tuning. As the representations of pretrained models are used as a foundation for different downstream tasks, this paper proposes a new task-agnostic framework, \textit{SynBench}, to measure the quality of pretrained representations using synthetic data. We set up a reference by a theoretically-derived robustness-accuracy tradeoff of the class conditional Gaussian mixture. Given a pretrained model, the representations of data synthesized from the Gaussian mixture are used to compare with our reference to infer the quality. By comparing the ratio of area-under-curve between the raw data and their representations, SynBench offers a quantifiable score for robustness-accuracy performance benchmarking. Our framework applies to a wide range of pretrained models taking continuous data inputs and is independent of the downstream tasks and datasets. Evaluated with several pretrained vision transformer models, the experimental results show that our SynBench score well matches the actual linear probing performance of the pre-trained model when fine-tuned on downstream tasks. Moreover, our framework can be used to inform the design of robust linear probing on pretrained representations to mitigate the robustness-accuracy tradeoff in downstream tasks.