FedDTPT: Federated Discrete and Transferable Prompt Tuning for Black-Box Large Language Models

TL;DR

This paper introduces FedDTPT, a federated prompt tuning method for black-box large language models that enhances privacy, reduces communication costs, and produces transferable prompts through a novel token-level discrete optimization approach.

Contribution

It presents the first federated discrete prompt tuning framework for black-box LLMs, combining token-level discrete optimization with semantic filtering and clustering for improved performance.

Findings

Achieves higher accuracy than state-of-the-art methods.

Reduces communication overhead in federated settings.

Demonstrates robustness to non-iid data and transferability of prompts.

Abstract

In recent years, large language models (LLMs) have significantly advanced the field of natural language processing (NLP). By fine-tuning LLMs with data from specific scenarios, these foundation models can better adapt to various downstream tasks. However, the fine-tuning process poses privacy leakage risks, particularly in centralized data processing scenarios. To address user privacy concerns, federated learning (FL) has been introduced to mitigate the risks associated with centralized data collection from multiple sources. Nevertheless, the privacy of LLMs themselves is equally critical, as potential malicious attacks challenge their security, an issue that has received limited attention in current research. Consequently, establishing a trusted multi-party model fine-tuning environment is essential. Additionally, the local deployment of large LLMs incurs significant storage costs and high computational demands. To address these challenges, we propose for the first time a federated discrete and transferable prompt tuning, namely FedDTPT, for black-box large language models. In the client optimization phase, we adopt a token-level discrete prompt optimization method that leverages a feedback loop based on prediction accuracy to drive gradient-free prompt optimization through the MLM API. For server optimization, we employ an attention mechanism based on semantic similarity to filter all local prompt tokens, along with an embedding distance elbow detection and DBSCAN clustering strategy to enhance the filtering process. Experimental results demonstrate that, compared to state-of-the-art methods, our approach achieves higher accuracy, reduced communication overhead, and robustness to non-iid data in a black-box setting. Moreover, the optimized prompts are transferable.