arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.07514cs.CV

CNN-JEPA:使用联合嵌入预测架构自监督预训练卷积神经网络

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

Budapest University of Technology and Economics(布达佩斯技术与经济大学)

András Kalapos, Bálint Gyires-Tóth

AI总结:

本文提出CNN-JEPA,将联合嵌入预测架构成功应用于CNN,通过稀疏编码器、深度可分离卷积预测器和改进掩码策略,在ImageNet-100上以ResNet-50达到73.3%线性top-1准确率,且训练时间比同类方法减少17-35%。

摘要:

自监督学习(SSL)已成为预训练大型神经网络的重要方法,使模型和数据集规模得以空前扩展。虽然I-JEPA等近期进展在视觉Transformer上展现了良好结果,但将此类方法适配到卷积神经网络(CNN)面临独特挑战。本文提出CNN-JEPA,一种成功将联合嵌入预测架构方法应用于CNN的新型SSL方法。该方法采用稀疏CNN编码器处理掩码输入、使用深度可分离卷积的全卷积预测器以及改进的掩码策略。我们证明CNN-JEPA在ImageNet-100上优于采用ViT架构的I-JEPA,使用标准ResNet-50编码器达到73.3%的线性top-1准确率。与其他基于CNN的SSL方法相比,CNN-JEPA在相同训练轮数下所需训练时间减少17-35%,并接近BYOL、SimCLR和VICReg的线性及k-NN top-1准确率。我们的方法为CNN提供了一种更简单、更高效的现有SSL方法替代方案,仅需极少的数据增强且无需单独投影网络。

英文摘要:

Self-supervised learning (SSL) has become an important 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 CNNs. 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-100, achieving a 73.3% linear top-1 accuracy using a standard ResNet-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-1 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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑