FLoRA: Enhancing Vision-Language Models with Parameter-Efficient Federated Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Nguyen, Duy Phuong, Munoz, J. Pablo, Jannesari, Ali
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929325679312896
author Nguyen, Duy Phuong
Munoz, J. Pablo
Jannesari, Ali
author_facet Nguyen, Duy Phuong
Munoz, J. Pablo
Jannesari, Ali
contents In the rapidly evolving field of artificial intelligence, multimodal models, e.g., integrating vision and language into visual-language models (VLMs), have become pivotal for many applications, ranging from image captioning to multimodal search engines. Among these models, the Contrastive Language-Image Pre-training (CLIP) model has demonstrated remarkable performance in understanding and generating nuanced relationships between text and images. However, the conventional training of such models often requires centralized aggregation of vast datasets, posing significant privacy and data governance challenges. To address these concerns, this paper proposes a novel approach that leverages Federated Learning and parameter-efficient adapters, i.e., Low-Rank Adaptation (LoRA), to train VLMs. This methodology preserves data privacy by training models across decentralized data sources and ensures model adaptability and efficiency through LoRA's parameter-efficient fine-tuning. Our approach accelerates training time by up to 34.72 times and requires 2.47 times less memory usage than full fine-tuning.
format Preprint
id arxiv_https___arxiv_org_abs_2404_15182
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FLoRA: Enhancing Vision-Language Models with Parameter-Efficient Federated Learning
Nguyen, Duy Phuong
Munoz, J. Pablo
Jannesari, Ali
Machine Learning
Artificial Intelligence
In the rapidly evolving field of artificial intelligence, multimodal models, e.g., integrating vision and language into visual-language models (VLMs), have become pivotal for many applications, ranging from image captioning to multimodal search engines. Among these models, the Contrastive Language-Image Pre-training (CLIP) model has demonstrated remarkable performance in understanding and generating nuanced relationships between text and images. However, the conventional training of such models often requires centralized aggregation of vast datasets, posing significant privacy and data governance challenges. To address these concerns, this paper proposes a novel approach that leverages Federated Learning and parameter-efficient adapters, i.e., Low-Rank Adaptation (LoRA), to train VLMs. This methodology preserves data privacy by training models across decentralized data sources and ensures model adaptability and efficiency through LoRA's parameter-efficient fine-tuning. Our approach accelerates training time by up to 34.72 times and requires 2.47 times less memory usage than full fine-tuning.
title FLoRA: Enhancing Vision-Language Models with Parameter-Efficient Federated Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2404.15182