English

Self-Distillation of Hidden Layers for Self-Supervised Representation Learning

Computer Vision and Pattern Recognition 2026-03-17 v1 Machine Learning

Abstract

The landscape of self-supervised learning (SSL) is currently dominated by generative approaches (e.g., MAE) that reconstruct raw low-level data, and predictive approaches (e.g., I-JEPA) that predict high-level abstract embeddings. While generative methods provide strong grounding, they are computationally inefficient for high-redundancy modalities like imagery, and their training objective does not prioritize learning high-level, conceptual features. Conversely, predictive methods often suffer from training instability due to their reliance on the non-stationary targets of final-layer self-distillation. We introduce Bootleg, a method that bridges this divide by tasking the model with predicting latent representations from multiple hidden layers of a teacher network. This hierarchical objective forces the model to capture features at varying levels of abstraction simultaneously. We demonstrate that Bootleg significantly outperforms comparable baselines (+10% over I-JEPA) on classification of ImageNet-1K and iNaturalist-21, and semantic segmentation of ADE20K and Cityscapes.

Keywords

Cite

@article{arxiv.2603.15553,
  title  = {Self-Distillation of Hidden Layers for Self-Supervised Representation Learning},
  author = {Scott C. Lowe and Anthony Fuller and Sageev Oore and Evan Shelhamer and Graham W. Taylor},
  journal= {arXiv preprint arXiv:2603.15553},
  year   = {2026}
}
R2 v1 2026-07-01T11:22:41.882Z