Where Did Your Model Learn That? Label-free Influence for Self-supervised Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Harilal, Nidhin, Rege, Amit Kiran, Bafghi, Reza Akbarian, Raissi, Maziar, Monteleoni, Claire
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917876430012416
author Harilal, Nidhin
Rege, Amit Kiran
Bafghi, Reza Akbarian
Raissi, Maziar
Monteleoni, Claire
author_facet Harilal, Nidhin
Rege, Amit Kiran
Bafghi, Reza Akbarian
Raissi, Maziar
Monteleoni, Claire
contents Self-supervised learning (SSL) has revolutionized learning from large-scale unlabeled datasets, yet the intrinsic relationship between pretraining data and the learned representations remains poorly understood. Traditional supervised learning benefits from gradient-based data attribution tools like influence functions that measure the contribution of an individual data point to model predictions. However, existing definitions of influence rely on labels, making them unsuitable for SSL settings. We address this gap by introducing Influence-SSL, a novel and label-free approach for defining influence functions tailored to SSL. Our method harnesses the stability of learned representations against data augmentations to identify training examples that help explain model predictions. We provide both theoretical foundations and empirical evidence to show the utility of Influence-SSL in analyzing pre-trained SSL models. Our analysis reveals notable differences in how SSL models respond to influential data compared to supervised models. Finally, we validate the effectiveness of Influence-SSL through applications in duplicate detection, outlier identification and fairness analysis. Code is available at: \url{https://github.com/cryptonymous9/Influence-SSL}.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17170
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Where Did Your Model Learn That? Label-free Influence for Self-supervised Learning
Harilal, Nidhin
Rege, Amit Kiran
Bafghi, Reza Akbarian
Raissi, Maziar
Monteleoni, Claire
Machine Learning
Computer Vision and Pattern Recognition
Self-supervised learning (SSL) has revolutionized learning from large-scale unlabeled datasets, yet the intrinsic relationship between pretraining data and the learned representations remains poorly understood. Traditional supervised learning benefits from gradient-based data attribution tools like influence functions that measure the contribution of an individual data point to model predictions. However, existing definitions of influence rely on labels, making them unsuitable for SSL settings. We address this gap by introducing Influence-SSL, a novel and label-free approach for defining influence functions tailored to SSL. Our method harnesses the stability of learned representations against data augmentations to identify training examples that help explain model predictions. We provide both theoretical foundations and empirical evidence to show the utility of Influence-SSL in analyzing pre-trained SSL models. Our analysis reveals notable differences in how SSL models respond to influential data compared to supervised models. Finally, we validate the effectiveness of Influence-SSL through applications in duplicate detection, outlier identification and fairness analysis. Code is available at: \url{https://github.com/cryptonymous9/Influence-SSL}.
title Where Did Your Model Learn That? Label-free Influence for Self-supervised Learning
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.17170