Towards Building the Federated GPT: Federated Instruction Tuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Jianyi, Vahidian, Saeed, Kuo, Martin, Li, Chunyuan, Zhang, Ruiyi, Yu, Tong, Zhou, Yufan, Wang, Guoyin, Chen, Yiran
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916109151633408
author Zhang, Jianyi
Vahidian, Saeed
Kuo, Martin
Li, Chunyuan
Zhang, Ruiyi
Yu, Tong
Zhou, Yufan
Wang, Guoyin
Chen, Yiran
author_facet Zhang, Jianyi
Vahidian, Saeed
Kuo, Martin
Li, Chunyuan
Zhang, Ruiyi
Yu, Tong
Zhou, Yufan
Wang, Guoyin
Chen, Yiran
contents While "instruction-tuned" generative large language models (LLMs) have demonstrated an impressive ability to generalize to new tasks, the training phases heavily rely on large amounts of diverse and high-quality instruction data (such as ChatGPT and GPT-4). Unfortunately, acquiring high-quality data, especially when it comes to human-written data, can pose significant challenges both in terms of cost and accessibility. Moreover, concerns related to privacy can further limit access to such data, making the process of obtaining it a complex and nuanced undertaking. Consequently, this hinders the generality of the tuned models and may restrict their effectiveness in certain contexts. To tackle this issue, our study introduces a new approach called Federated Instruction Tuning (FedIT), which leverages federated learning (FL) as the learning framework for the instruction tuning of LLMs. This marks the first exploration of FL-based instruction tuning for LLMs. This is especially important since text data is predominantly generated by end users. Therefore, it is imperative to design and adapt FL approaches to effectively leverage these users' diverse instructions stored on local devices, while preserving privacy and ensuring data security. In the current paper, by conducting widely used GPT-4 auto-evaluation, we demonstrate that by exploiting the heterogeneous and diverse sets of instructions on the client's end with the proposed framework FedIT, we improved the performance of LLMs compared to centralized training with only limited local instructions. Further, in this paper, we developed a Github repository named Shepherd. This repository offers a foundational framework for exploring federated fine-tuning of LLMs using heterogeneous instructions across diverse categories.
format Preprint
id arxiv_https___arxiv_org_abs_2305_05644
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Towards Building the Federated GPT: Federated Instruction Tuning
Zhang, Jianyi
Vahidian, Saeed
Kuo, Martin
Li, Chunyuan
Zhang, Ruiyi
Yu, Tong
Zhou, Yufan
Wang, Guoyin
Chen, Yiran
Computation and Language
Distributed, Parallel, and Cluster Computing
Systems and Control
While "instruction-tuned" generative large language models (LLMs) have demonstrated an impressive ability to generalize to new tasks, the training phases heavily rely on large amounts of diverse and high-quality instruction data (such as ChatGPT and GPT-4). Unfortunately, acquiring high-quality data, especially when it comes to human-written data, can pose significant challenges both in terms of cost and accessibility. Moreover, concerns related to privacy can further limit access to such data, making the process of obtaining it a complex and nuanced undertaking. Consequently, this hinders the generality of the tuned models and may restrict their effectiveness in certain contexts. To tackle this issue, our study introduces a new approach called Federated Instruction Tuning (FedIT), which leverages federated learning (FL) as the learning framework for the instruction tuning of LLMs. This marks the first exploration of FL-based instruction tuning for LLMs. This is especially important since text data is predominantly generated by end users. Therefore, it is imperative to design and adapt FL approaches to effectively leverage these users' diverse instructions stored on local devices, while preserving privacy and ensuring data security. In the current paper, by conducting widely used GPT-4 auto-evaluation, we demonstrate that by exploiting the heterogeneous and diverse sets of instructions on the client's end with the proposed framework FedIT, we improved the performance of LLMs compared to centralized training with only limited local instructions. Further, in this paper, we developed a Github repository named Shepherd. This repository offers a foundational framework for exploring federated fine-tuning of LLMs using heterogeneous instructions across diverse categories.
title Towards Building the Federated GPT: Federated Instruction Tuning
topic Computation and Language
Distributed, Parallel, and Cluster Computing
Systems and Control
url https://arxiv.org/abs/2305.05644