MIP: CLIP-based Image Reconstruction from PEFT Gradients

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhou, Peiheng, Hu, Ming, Xie, Xiaofei, Huang, Yihao, Chen, Kangjie, Chen, Mingsong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911794372542464
author Zhou, Peiheng
Hu, Ming
Xie, Xiaofei
Huang, Yihao
Chen, Kangjie
Chen, Mingsong
author_facet Zhou, Peiheng
Hu, Ming
Xie, Xiaofei
Huang, Yihao
Chen, Kangjie
Chen, Mingsong
contents Contrastive Language-Image Pre-training (CLIP) model, as an effective pre-trained multimodal neural network, has been widely used in distributed machine learning tasks, especially Federated Learning (FL). Typically, CLIP-based FL adopts Parameter-Efficient Fine-Tuning (PEFT) for model training, which only fine-tunes adapter parameters or soft prompts rather than the full parameters. Although PEFT is different from the traditional training mode, in this paper, we theoretically analyze that the gradients of adapters or soft prompts can still be used to perform image reconstruction attacks. Based on our theoretical analysis, we propose Multm-In-Parvo (MIP), a proprietary reconstruction attack method targeting CLIP-based distributed machine learning architecture. Specifically, MIP can reconstruct CLIP training images according to the gradients of soft prompts or an adapter. In addition, MIP includes a label prediction strategy to accelerate convergence and an inverse gradient estimation mechanism to avoid the vanishing gradient problem on the text encoder. Experimental results show that MIP can effectively reconstruct training images according to the gradients of soft prompts or adapters of CLIP models.
format Preprint
id arxiv_https___arxiv_org_abs_2403_07901
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MIP: CLIP-based Image Reconstruction from PEFT Gradients
Zhou, Peiheng
Hu, Ming
Xie, Xiaofei
Huang, Yihao
Chen, Kangjie
Chen, Mingsong
Computer Vision and Pattern Recognition
Machine Learning
Contrastive Language-Image Pre-training (CLIP) model, as an effective pre-trained multimodal neural network, has been widely used in distributed machine learning tasks, especially Federated Learning (FL). Typically, CLIP-based FL adopts Parameter-Efficient Fine-Tuning (PEFT) for model training, which only fine-tunes adapter parameters or soft prompts rather than the full parameters. Although PEFT is different from the traditional training mode, in this paper, we theoretically analyze that the gradients of adapters or soft prompts can still be used to perform image reconstruction attacks. Based on our theoretical analysis, we propose Multm-In-Parvo (MIP), a proprietary reconstruction attack method targeting CLIP-based distributed machine learning architecture. Specifically, MIP can reconstruct CLIP training images according to the gradients of soft prompts or an adapter. In addition, MIP includes a label prediction strategy to accelerate convergence and an inverse gradient estimation mechanism to avoid the vanishing gradient problem on the text encoder. Experimental results show that MIP can effectively reconstruct training images according to the gradients of soft prompts or adapters of CLIP models.
title MIP: CLIP-based Image Reconstruction from PEFT Gradients
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2403.07901