Title
Distributed Fine-tuning of Language Models on Private Data
Abstract
One of the big challenges in machine learning applications is that training data can be different from the real-world data faced by the algorithm. In language modeling, users’ language (e.g. in private messaging) could change in a year and be completely different from what we observe in publicly available data. At the same time, public data can be used for obtaining general knowledge (i.e. general model of English). We study approaches to distributed fine-tuning of a general model on user private data with the additional requirements of maintaining the quality on the general data and minimization of communication costs. We propose a novel technique that significantly improves prediction quality on users’ language compared to a general model and outperforms gradient compression methods in terms of communication efficiency. The proposed procedure is fast and leads to an almost 70% perplexity reduction and 8.7 percentage point improvement in keystroke saving rate on informal English texts. Finally, we propose an experimental framework for evaluating differential privacy of distributed training of language models and show that our approach has good privacy guarantees.
Year
Venue
Field
2018
international conference on learning representations
Training set,Differential privacy,Computer science,Fine-tuning,Keystroke logging,Minification,General knowledge,Artificial intelligence,Percentage point,Language model,Machine learning
DocType
Citations 
PageRank 
Conference
0
0.34
References 
Authors
0
5
Name
Order
Citations
PageRank
Vadim Popov102.70
Mikhail Kudinov201.69
Irina Piontkovskaya302.37
Petr Vytovtov401.01
Alex Nevidomsky501.35