CNN-JEPA: Self-Supervised Pretraining Convolutional Neural Networks Using Joint Embedding Predictive Architecture
Self-supervised learning (SSL) has become an im-portant approach in pretraining large neural networks, enabling unprecedented scaling of model and dataset sizes. While recent advances like I-JEPA have shown promising results for Vision Transformers, adapting such methods to Convolutional Neural Networks (CNNs) presents unique challenges. In this paper, we introduce CNN -JEPA, a novel SSL method that successfully applies the joint embedding predictive architecture approach to CNN s. Our method incorporates a sparse CNN encoder to handle masked inputs, a fully convolutional predictor using depthwise separable convolutions, and an improved masking strategy. We demonstrate that CNN-JEPA outperforms I-JEPA with ViT architectures on ImageNet-l00, achieving a 73.3% linear top-1 accuracy using a standard ResN et-50 encoder. Compared to other CNN-based SSL methods, CNN-JEPA requires 17– 35 % less training time for the same number of epochs and approaches the linear and k-NN top-l accuracies of BYOL, SimCLR, and VICReg. Our approach offers a simpler, more efficient alternative to existing SSL methods for CNNs, requiring minimal augmentations and no separate projector network.
Authors
- András Kalapos (ORCID: https://orcid.org/0000-0002-9018-1372)
- Bálint Gyires-Tóth (ORCID: https://orcid.org/0000-0003-1059-9822)
Institutions
- Budapest University of Technology and Economics (HU)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1109/icmla61862.2024.00169
- Citations
- 2
- Primary Topic
- Neural Networks and Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.69