atlas

Research map / Label-efficient and robust learning / Self-supervised and contrastive

Self-Supervised Representation Learning with Meta Comprehensive Regularization

AAAI 2024 · Label-efficient and robust learning › Self-supervised and contrastive · approach: Data-efficient and robust self-supervision · 6 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Adds a plug-and-play CompMod module with Meta Comprehensive Regularization (bi-level optimization plus maximum-entropy coding) to existing SSL methods so that learned representations capture more task-relevant, non-shared information.

Problem

Data augmentation in self-supervised learning enforces invariance across views, so only the shared information is captured and label-related information lost to augmentation is discarded, hurting downstream generalization.

Approach

Fuse the embeddings of two augmented views into a comprehensive embedding, regularize each view toward it via a maximum-entropy coding (lossy coding rate, Taylor-expanded) loss, and optimize the comprehensive projection head with a bi-level scheme; the approach is added on top of SimCLR, BYOL and Barlow Twins.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add CompMod and MCR to BYOLBYOL, ResNet-18 · CIFAR-10linear accuracy: 91.73 -> 93.96Helped
Add CompMod and MCR to SimCLRSimCLR, ResNet-18 · CIFAR-100linear accuracy: 66.83 -> 68.91Helped
Add CompMod and MCR to Barlow TwinsBarlow Twins, ResNet-18 · ImageNet-100top-1 accuracy: 72.88 -> 76.88Helped
Add CompMod and MCR to Barlow TwinsBarlow Twins, ResNet-18 · ImageNet, semi-supervised 1% labelstop-1 accuracy: 55.0 -> 56.1Helped
Add CompMod and MCR to BYOLBYOL, ImageNet-pretrained network · COCO instance segmentationAP: 33.1 -> 35.4Helped
Use all listed augmentations together rather than horizontal flip and rotate only (ablation)BYOL+, ResNet-18 · CIFAR-10linear accuracy: 92.64 -> 93.96Helped
Vary λ1 and λ2 over the tested values (ablation)SimCLR+, ResNet-18 · CIFAR-10classification accuracy: 91.11 to 93.91Mixed
Replace representation-space fusion with Mixup at α=0.1 (ablation)BYOL+ · CIFAR-10classification accuracy: 93.96 -> 91.03Hurt
Fuse features in embedding space instead of representation space (ablation)BYOL+ · CIFAR-10classification accuracy: 93.96 -> 92.58Hurt
Remove bi-level optimization (ablation)BYOL+ · CIFAR-10classification accuracy: 93.96 -> 92.05Hurt

Limitations

Experimental setup

Datasets: CIFAR-10 CIFAR-100 STL-10 Tiny ImageNet ImageNet-100 ImageNet COCO

Benchmarks: CIFAR-10 CIFAR-100 STL-10 Tiny ImageNet ImageNet-100 ImageNet COCO

Metrics: top-1 accuracy top-5 accuracy 5-nn accuracy AP AP50 AP75

Baselines: SimCLR BarlowTwins BYOL SimSiam W-MSE SwAV SSL-HSIC VICReg MoCo CMC

Builds on: SimCLR BYOL Barlow Twins maximum entropy coding MAML-style meta-learning

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy