Differentially Private Training of Mixture of Experts Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Tholoniat, Pierre, Inan, Huseyin A., Kulkarni, Janardhan, Sim, Robert
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916121813188608
author Tholoniat, Pierre
Inan, Huseyin A.
Kulkarni, Janardhan
Sim, Robert
author_facet Tholoniat, Pierre
Inan, Huseyin A.
Kulkarni, Janardhan
Sim, Robert
contents This position paper investigates the integration of Differential Privacy (DP) in the training of Mixture of Experts (MoE) models within the field of natural language processing. As Large Language Models (LLMs) scale to billions of parameters, leveraging expansive datasets, they exhibit enhanced linguistic capabilities and emergent abilities. However, this growth raises significant computational and privacy concerns. Our study addresses these issues by exploring the potential of MoE models, known for their computational efficiency, and the application of DP, a standard for privacy preservation. We present the first known attempt to train MoE models under the constraints of DP, addressing the unique challenges posed by their architecture and the complexities of DP integration. Our initial experimental studies demonstrate that MoE models can be effectively trained with DP, achieving performance that is competitive with their non-private counterparts. This initial study aims to provide valuable insights and ignite further research in the domain of privacy-preserving MoE models, softly laying the groundwork for prospective developments in this evolving field.
format Preprint
id arxiv_https___arxiv_org_abs_2402_07334
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Differentially Private Training of Mixture of Experts Models
Tholoniat, Pierre
Inan, Huseyin A.
Kulkarni, Janardhan
Sim, Robert
Cryptography and Security
Machine Learning
This position paper investigates the integration of Differential Privacy (DP) in the training of Mixture of Experts (MoE) models within the field of natural language processing. As Large Language Models (LLMs) scale to billions of parameters, leveraging expansive datasets, they exhibit enhanced linguistic capabilities and emergent abilities. However, this growth raises significant computational and privacy concerns. Our study addresses these issues by exploring the potential of MoE models, known for their computational efficiency, and the application of DP, a standard for privacy preservation. We present the first known attempt to train MoE models under the constraints of DP, addressing the unique challenges posed by their architecture and the complexities of DP integration. Our initial experimental studies demonstrate that MoE models can be effectively trained with DP, achieving performance that is competitive with their non-private counterparts. This initial study aims to provide valuable insights and ignite further research in the domain of privacy-preserving MoE models, softly laying the groundwork for prospective developments in this evolving field.
title Differentially Private Training of Mixture of Experts Models
topic Cryptography and Security
Machine Learning
url https://arxiv.org/abs/2402.07334