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

Institutions

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
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

CNN-JEPA: Self-Supervised Pretraining Convolutional Neural Networks Using Joint Embedding Predictive Architecture

András Kalapos, Bálint Gyires-Tóth
2 citations
Neural Networks and Applications
0.69
article

CNN-JEPA: Self-Supervised Pretraining Convolutional Neural Networks Using Joint Embedding Predictive Architecture

András Kalapos, Bálint Gyires-Tóth
article en
2 citations

Abstract

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.

Budapest University of Technology and Economics (HU)
Openalex Percentile: Top 22%
Neural Networks and Applications
0.69
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

CNN-JEPA: Self-Supervised Pretraining Convolutional Neural Networks Using Joint Embedding Predictive Architecture — András Kalapos, Bálint Gyires-Tóth · (2024) | TGRS Research Map | TGRS