定位自监督视觉编码器中记忆数据的层级与单元,揭示其分布规律。
Localizing Memorization in SSL Vision Encoders
- 提出层级和单元级记忆度量,无需标签即可前向计算
- 发现大量单元高度记忆单个样本,且在深层和全连接层更显著
- 结果可用于模型微调与剪枝,适合关注模型可解释性的研究者
近期研究发现,尽管自监督学习(SSL)编码器在数百万图像上训练,仍会记忆个别数据点。然而,关于记忆行为在编码器内部的具体位置仍不清楚。为此,我们提出两种独立于下游任务、无需标签信息的度量方法,可在前向传播中实现对每层(layermem)和每单元(unitmem)的记忆定位。通过对卷积与基于Transformer的多种编码器架构,在对比与非对比式SSL框架下于多样数据集上的分析,我们发现:(1)尽管记忆随网络深度增加,但高记忆单元遍布整个编码器;(2)大量单元表现出对单个样本的显著记忆,这与有监督模型形成反差;(3)异常或离群样本引发远高于常规样本的层与单元记忆;(4)在视觉Transformer中,多数记忆集中在全连接层。此外,记忆定位具有提升微调效果与指导剪枝策略的潜力。
原文摘要 · Abstract (English)
Recent work on studying memorization in self-supervised learning (SSL) suggests that even though SSL encoders are trained on millions of images, they still memorize individual data points. While effort has been put into characterizing the memorized data and linking encoder memorization to downstream utility, little is known about where the memorization happens inside SSL encoders. To close this gap, we propose two metrics for localizing memorization in SSL encoders on a per-layer (layermem) and per-unit basis (unitmem). Our localization methods are independent of the downstream task, do not require any label information, and can be performed in a forward pass. By localizing memorization in various encoder architectures (convolutional and transformer-based) trained on diverse datasets with contrastive and non-contrastive SSL frameworks, we find that (1) while SSL memorization increases with layer depth, highly memorizing units are distributed across the entire encoder, (2) a significant fraction of units in SSL encoders experiences surprisingly high memorization of individual data points, which is in contrast to models trained under supervision, (3) atypical (or outlier) data points cause much higher layer and unit memorization than standard data points, and (4) in vision transformers, most memorization happens in the fully-connected layers. Finally, we show that localizing memorization in SSL has the potential to improve fine-tuning and to inform pruning strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。