LayerLock: Non-collapsing Representation Learning with Progressive Freezing
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2025
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866915523731652608 |
|---|---|
| author | Erdogan, Goker Parthasarathy, Nikhil Ionescu, Catalin Hudson, Drew A. Lerchner, Alexander Zisserman, Andrew Sajjadi, Mehdi S. M. Carreira, Joao |
| author_facet | Erdogan, Goker Parthasarathy, Nikhil Ionescu, Catalin Hudson, Drew A. Lerchner, Alexander Zisserman, Andrew Sajjadi, Mehdi S. M. Carreira, Joao |
| contents | We introduce LayerLock, a simple yet effective approach for self-supervised visual representation learning, that gradually transitions from pixel to latent prediction through progressive layer freezing. First, we make the observation that during training of video masked-autoencoding (MAE) models, ViT layers converge in the order of their depth: shallower layers converge early, deeper layers converge late. We then show that this observation can be exploited to accelerate standard MAE by progressively freezing the model according to an explicit schedule, throughout training. Furthermore, this same schedule can be used in a simple and scalable approach to latent prediction that does not suffer from "representation collapse". We apply our proposed approach, LayerLock, to large models of up to 4B parameters with results surpassing those of non-latent masked prediction on the 4DS perception suite. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2509_10156 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | LayerLock: Non-collapsing Representation Learning with Progressive Freezing Erdogan, Goker Parthasarathy, Nikhil Ionescu, Catalin Hudson, Drew A. Lerchner, Alexander Zisserman, Andrew Sajjadi, Mehdi S. M. Carreira, Joao Computer Vision and Pattern Recognition We introduce LayerLock, a simple yet effective approach for self-supervised visual representation learning, that gradually transitions from pixel to latent prediction through progressive layer freezing. First, we make the observation that during training of video masked-autoencoding (MAE) models, ViT layers converge in the order of their depth: shallower layers converge early, deeper layers converge late. We then show that this observation can be exploited to accelerate standard MAE by progressively freezing the model according to an explicit schedule, throughout training. Furthermore, this same schedule can be used in a simple and scalable approach to latent prediction that does not suffer from "representation collapse". We apply our proposed approach, LayerLock, to large models of up to 4B parameters with results surpassing those of non-latent masked prediction on the 4DS perception suite. |
| title | LayerLock: Non-collapsing Representation Learning with Progressive Freezing |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2509.10156 |