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

SASSL:通过神经风格迁移增强自监督学习

SASSL: Enhancing Self-Supervised Learning via Neural Style Transfer

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Google Research(谷歌研究院);Queen’s University, Canada(加拿大女王大学);Google DeepMind(谷歌DeepMind)

Renan A. Rojas-Gomez, Karan Singhal, Ali Etemad, Alex Bijamov, Warren R. Morningstar, Philip Andrew Mansfield

AI总结:

针对自监督学习现有数据增强破坏图像结构与语义的问题,提出基于神经风格迁移的SASSL增强方法,解耦图像语义与风格并仅变换风格,在ImageNet等基准上提升分类及迁移性能且不影响预训练吞吐量。

摘要:

现有自监督学习中的数据增强方法虽然形式多样,但无法保留自然图像的固有结构。这会导致增强后的样本发生畸变,语义信息受损,最终影响下游任务性能。为克服这一局限,我们提出SASSL(面向自监督学习的风格增强),这是一种基于神经风格迁移的新型数据增强技术。SASSL将图像中的语义属性与风格属性解耦,仅对图像的风格施加变换同时保留内容,生成能更好保留语义信息的多样化样本。与MoCo、SimCLR和BYOL等成熟自监督方法相比,SASSL将ImageNet上的图像分类top-1准确率提升了最高2个百分点,同时在多个数据集上实现了更优的迁移学习性能。由于SASSL可作为数据增强流水线的一部分异步执行,获得这些性能提升的同时不会改变预训练吞吐量。

英文摘要:

Existing data augmentation in self-supervised learning, while diverse, fails to preserve the inherent structure of natural images. This results in distorted augmented samples with compromised semantic information, ultimately impacting downstream performance. To overcome this limitation, we propose SASSL: Style Augmentations for Self Supervised Learning, a novel data augmentation technique based on Neural Style Transfer. SASSL decouples semantic and stylistic attributes in images and applies transformations exclusively to their style while preserving content, generating diverse samples that better retain semantic information. SASSL boosts top-1 image classification accuracy on ImageNet by up to 2 percentage points compared to established self-supervised methods like MoCo, SimCLR, and BYOL, while achieving superior transfer learning performance across various datasets. Because SASSL can be performed asynchronously as part of the data augmentation pipeline, these performance impacts can be obtained with no change in pretraining throughput.

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

↑