Original: D IFFERENTIALLY P RIVATE F INE - TUNING OF L ANGUAGE M ODELS
Revised: D IFFERENTIALLY P RIVATE F INE - TUNING OF L ANGUAGE M ODELS

Original: A BSTRACT
Revised: A BSTRACT

Original: We give simpler, sparser, and faster algorithms for differentially private finetuning of large-scale pre-trained language models, which achieve the state-ofthe-art privacy versus utility tradeoffs on many standard NLP tasks.
Revised: We give simpler, sparser, and faster algorithms for differentially private finetuning of large-scale pre-trained language models, which achieve the state-ofthe-art privacy versus utility tradeoffs on many standard NLP tasks.

Original: We propose a meta-framework for this problem, inspired by the recent success of highly parameter-efficient methods for fine-tuning.
Revised: We propose a meta-framework for this problem, inspired by the recent success of highly parameter-efficient methods for fine-tuning.

Original: Our experiments show that differentially private adaptations of these approaches outperform previous private algorithms in three important dimensions: utility, privacy, and the computational and memory cost of private training.
Revised: Our experiments show that differentially private adaptations of these approaches outperform previous private algorithms in three important dimensions: utility, privacy, and the computational and memory cost of private training.

Original: On many commonly studied datasets, the utility of private models approaches that of non-private models.
Revised: On many commonly studied datasets, the utility of private models approaches that of non-private models.

Original: For example, on the
Revised: For example, on the

Original: MNLI dataset we achieve an accuracy of 87 .
Revised: MNLI dataset we achieve an accuracy of 87 .

Original: 8% using RoBERTa-Large and 83 .
Revised: 8% using RoBERTa-Large and 83 .

Original: 5% using RoBERTa-Base with a privacy budget of ε = 6 . 7 .
Revised: 5% using RoBERTa-Base with a privacy budget of ε = 6 . 7 .

Original: In comparison, absent privacy constraints, RoBERTa-Large achieves an accuracy of 90 . 2% .
Revised: In comparison, absent privacy constraints, RoBERTa-Large achieves an accuracy of 90 . 2% .

Original: Our findings are similar for natural language generation when privately fine-tuning GPT-2.
Revised: Our findings are similar for natural language generation when privately fine-tuning GPT-2.

Original: Our experiments also show that larger models are better suited for private fine-tuning: while they are well known to achieve superior accuracy non-privately, we find that they also better maintain their accuracy when privacy is introduced.
Revised: Our experiments also show that larger models are better suited for private fine-tuning: while they are well known to achieve superior accuracy non-privately, we find that they also better maintain their accuracy when privacy is introduced.

Original:  I NTRODUCTION
Revised:  I NTRODUCTION

Original: Deep learning models are well known to leak sensitive information about the dataset when trained using conventional methods (Shokri et al., 2017; Carlini et al., 2019; 2021).
Revised: Deep learning models are well known to leak sensitive information about the dataset when trained using conventional methods (Shokri et al., 2017; Carlini et al., 2019; 2021).

Original: To combat this issue, models can instead be trained to guarantee differential privacy (DP) (Dwork et al., 2006b), a strong notion of data privacy which limits the influence of any individual training point on the final model.
Revised: To combat this issue, models can instead be trained to guarantee differential privacy (DP) (Dwork et al., 2006b), a strong notion of data privacy which limits the influence of any individual training point on the final model.

Original: While DP is one of the few approaches capable of providing machine learning models with rigorous privacy guarantees, it generally comes at a cost in terms of test accuracy.
Revised: While DP is one of the few approaches capable of providing machine learning models with rigorous privacy guarantees, it generally comes at a cost in terms of test accuracy.

Original: One oft-cited explanation is that the constraint of DP necessitates much more training data (Tram`er & Boneh, 2021; Feldman, 2020; Brown et al., 2021).
Revised: One oft-cited explanation is that the constraint of DP necessitates much more training data (Tram`er & Boneh, 2021; Feldman, 2020; Brown et al., 2021).

Original: Unfortunately, more training data may be hard to acquire, particularly in settings where privacy is a concern.
Revised: Unfortunately, more training data may be hard to acquire, particularly in settings where privacy is a concern.

Original: Parallel to these developments, Transformer-based (Vaswani et al., 2017) large language models (LLMs), including the BERT (Devlin et al., 2019; Liu et al., 2019) and GPT (Radford et al., 2018; 2019; Brown et al., 2020) families, have enabled significant progress in natural language processing, achieving state-of-the-art accuracy in almost every task considered.
Revised: Parallel to these developments, Transformer-based (Vaswani et al., 2017) large language models (LLMs), including the BERT (Devlin et al., 2019; Liu et al., 2019) and GPT (Radford et al., 2018; 2019; Brown et al., 2020) families, have enabled significant progress in natural language processing, achieving state-of-the-art accuracy in almost every task considered.

Original: These models are first pre-trained on an extremely large and diverse public dataset.
Revised: These models are first pre-trained on an extremely large and diverse public dataset.

Original: The weights are then fine-tuned for
Revised: The weights are then fine-tuned for

Original: [Table]
Revised: [Table]

Original: each task of interest using a much smaller task-specific dataset.
Revised: each task of interest using a much smaller task-specific dataset.

Original: For example, a single pre-trained
Revised: For example, a single pre-trained

Original: GPT-family model may be fine-tuned for various downstream tasks, such as email reply suggestion, sentence completion in text editors, language translation and more.
Revised: GPT-family model may be fine-tuned for various downstream tasks, such as email reply suggestion, sentence completion in text editors, language translation and more.

Original: This two-stage paradigm can naturally be adapted to solve tasks in private learning, automatically addressing the aforementioned data shortage issue via the massive scale of the public pre-training dataset.
Revised: This two-stage paradigm can naturally be adapted to solve tasks in private learning, automatically addressing the aforementioned data shortage issue via the massive scale of the public pre-training dataset.

Original: One may pre-train the model on public data as usual, 

Revised: One may pre-train the model on public data as usual,

Original: but then fine-tune the model privately .
Revised: but then fine-tune the model privately .

Original: Despite the success of these models task-specific fine-tuning introduces a number of technical challenges.
Revised: Despite the success of these models task-specific fine-tuning introduces a number of technical challenges.

Original: In the non-private setting, the immense size of LLMs makes it impractical to fine-tune the full model and store a separate copy of the parameters for hundreds of downstream tasks.
Revised: In the non-private setting, the immense size of LLMs makes it impractical to fine-tune the full model and store a separate copy of the parameters for hundreds of downstream tasks.

Original: Things only get worse with privacy, which leads to overheads in terms of running time, memory usage, and most importantly, accuracy.
Revised: Things only get worse with privacy, which leads to overheads in terms of running time, memory usage, and most importantly, accuracy.

Original: The magnitude of noise introduced to a model due to DP grows as the model size increases (Bassily et al., 2014; Abadi et al., 2016; Bun et al., 2014), which can overwhelm any signal for larger models.
Revised: The magnitude of noise introduced to a model due to DP grows as the model size increases (Bassily et al., 2014; Abadi et al., 2016; Bun et al., 2014), which can overwhelm any signal for larger models.

Original: A recent line of work in the non-private literature has proposed parameter-efficient methods to alleviate the issues of storage and computational cost for fine-tuning (Houlsby et al., 2019; Li & Liang, 2021;
Revised: A recent line of work in the non-private literature has proposed parameter-efficient methods to alleviate the issues of storage and computational cost for fine-tuning (Houlsby et al., 2019; Li & Liang, 2021;

Original: Aghajanyan et al., 2020; Hu et al., 2021; Mahabadi et al., 2021).
Revised: Aghajanyan et al., 2020; Hu et al., 2021; Mahabadi et al., 2021).

Original: The main focus of our work is to explore parameter-efficiency in the context of private learning.
Revised: The main focus of our work is to explore parameter-efficiency in the context of private learning.

Original: 1. O UR C ONTRIBUTIONS
Revised: 1. O UR C ONTRIBUTIONS

Original: Our primary contribution is to show that advanced parameter-efficient methods can lead to simpler and significantly improved algorithms for private fine-tuning.
Revised: Our primary contribution is to show that advanced parameter-efficient methods can lead to simpler and significantly improved algorithms for private fine-tuning.

Original: Our framework is illustrated in
Revised: Our framework is illustrated in

Original: Figure 1.
Revised: Figure 1.

Original: Our findings and contributions are summarized as follows:
Revised: Our findings and contributions are summarized as follows:

Original: State-of-the-art utility and privacy.
Revised: State-of-the-art utility and privacy.

Original: Empirical evaluation of our algorithms reveals that they achieve state-of-the-art accuracy versus privacy tradeoffs, improving upon the previous best (Yu et al., 2021b).
Revised: Empirical evaluation of our algorithms reveals that they achieve state-of-the-art accuracy versus privacy tradeoffs, improving upon the previous best (Yu et al., 2021b).

Original: More importantly, for many fine-tuning tasks, the utility of models trained with DP approaches that of non-private models.
Revised: More importantly, for many fine-tuning tasks, the utility of models trained with DP approaches that of non-private models.

Original: For example, privately fine-tuning RoBERTa-Large on the
Revised: For example, privately fine-tuning RoBERTa-Large on the

Original: MNLI data set (Williams et al., 2018), we achieve an accuracy of 87 .
Revised: MNLI data set (Williams et al., 2018), we achieve an accuracy of 87 .

Original: 8% with a privacy budget of ( ε = 6 . 7 , δ = 1e-6).
Revised: 8% with a privacy budget of ( ε = 6 . 7 , δ = 1e-6).

Original: Without privacy guarantees, RoBERTa-Large achieves an accuracy of 90 .
Revised: Without privacy guarantees, RoBERTa-Large achieves an accuracy of 90 .

Original: 2% (GPT-3 is known to achieve 91 .
Revised: 2% (GPT-3 is known to achieve 91 .

Original: 7% (Hu et al., 2021)); see Table 1 for a summary.
Revised: 7% (Hu et al., 2021)); see Table 1 for a summary.

Original: We also explore private natural language generation tasks, fine-tuning GPT-2 models on the E2E dataset (Novikova et al., 2017).
Revised: We also explore private natural language generation tasks, fine-tuning GPT-2 models on the E2E dataset (Novikova et al., 2017).

Original: Again, the utility approaches non-private levels: we achieve a ROUGE-L score of 0.6755 with GPT-2-Large and ( ε = 5 . 4 , δ = 1e-5), compared to 0.72 without privacy.
Revised: Again, the utility approaches non-private levels: we achieve a ROUGE-L score of 0.6755 with GPT-2-Large and ( ε = 5 . 4 , δ = 1e-5), compared to 0.72 without privacy.

Original: Larger models are better.
Revised: Larger models are better.

Original: Prior work has consistently shown that larger language models achieve better accuracy for downstream tasks.
Revised: Prior work has consistently shown that larger language models achieve better accuracy for downstream tasks.

Original: Our results give evidence that this phenomenon extends to the private setting.
Revised: Our results give evidence that this phenomenon extends to the private setting.

Original: For example, on the MNLI dataset, RoBERTa-Base achieves an accuracy of 83 .
Revised: For example, on the MNLI dataset, RoBERTa-Base achieves an accuracy of 83 .

Original: 5% whereas RoBERTa-Large achieves an accuracy of 87 .
Revised: 5% whereas RoBERTa-Large achieves an accuracy of 87 .

Original: 8% , both under a privacy budget of ( ε = 6 . 7 , δ = 1e-6).
Revised: 8% , both under a privacy budget of ( ε = 6 . 7 , δ = 1e-6).

Original: Similarly, privately fine-tuning with E2E, GPT-2-Small, GPT-2-Medium, and GPT-2-Large achieve ROUGE-L scores of 0.6219, 0.6645 and 0.6755 respectively, all under a privacy budget of ( ε = 5 . 4 , δ = 1e-5).
Revised: Similarly, privately fine-tuning with E2E, GPT-2-Small, GPT-2-Medium, and GPT-2-Large achieve ROUGE-L scores of 0.6219, 0.6645 and 0.6755 respectively, all under a privacy budget of ( ε = 5 . 4 , δ = 1e-5).

Original: While established in the non-private setting, we find this phenomenon quite surprising under DP.
Revised: While established in the non-private setting, we find this phenomenon quite surprising under DP.

Original: There is often a tension when choosing private model architectures: larger models may have higher capacity, but necessitate the introduction of more noise.
Revised: There is often a tension when choosing private model architectures: larger models may have higher capacity, but necessitate the introduction of more noise.

Original: Consequently, smaller and simpler private models achieve the better accuracy in several settings (Papernot et al., 2019; Tram ` er & Boneh, 2021).
Revised: Consequently, smaller and simpler private models achieve the better accuracy in several settings (Papernot et al., 2019; Tram ` er & Boneh, 2021).

Original: In contrast, our experiments show that fine-tuning the biggest models achieves the best accuracy.
Revised: In contrast, our experiments show that fine-tuning the biggest models achieves the best accuracy.

Original: Simpler, sparser, and faster.
Revised: Simpler, sparser, and faster.

Original: Beyond accuracy concerns, DP requirements also lead to significant overheads in terms of computation and memory usage.
Revised: Beyond accuracy concerns, DP requirements also lead to significant overheads in terms of computation and memory usage.

Original: The large number of parameters contributes to the high cost of training LLMs, and things get worse under privacy, which has been documented to increase training time by up to two orders of magnitude (Carlini et al., 2019; Subramani et al., 2021).
Revised: The large number of parameters contributes to the high cost of training LLMs, and things get worse under privacy, which has been documented to increase training time by up to two orders of magnitude (Carlini et al., 2019; Subramani et al., 2021).

Original: The parameter-efficient approaches we employ partially offset this issue: as we only update a small fraction of the total number of parameters, training becomes considerably more computationally and memory efficient.
Revised: The parameter-efficient approaches we employ partially offset this issue: as we only update a small fraction of the total number of parameters, training becomes considerably more computationally and memory efficient.

Original: Furthermore, as in the non-private setting, this framework leads to a modular design, where a single large pre-trained model can be augmented with lightweight modifications for each individual downstream task.
Revised: Furthermore, as in the non-private setting, this framework leads to a modular design, where a single large pre-trained model can be augmented with lightweight modifications for each individual downstream task.

Original: To the best of our knowledge, we are the first to fine-tune GPT-2-Large using differential privacy, the largest model trained thus far using DP.
Revised: To the best of our knowledge, we are the first to fine-tune GPT-2-Large using differential privacy, the largest model trained thus far using DP.

Original: Given our state-of-the-art results for a variety of standard NLP tasks using advanced fine-tuning techniques, we believe that our paper will serve as a benchmark for further work in this direction.
Revised: Given our state-of-the-art results for a variety of standard NLP tasks using advanced fine-tuning techniques, we believe that our paper will serve as a benchmark for further work in this direction.

Original: For example, the best average accuracy achieved by the prior work of Yu et al.
Revised: For example, the best average accuracy achieved by the prior work of Yu et al.

Original: (2021b) on four standard NLP tasks in Table 1 is 83 .
Revised: (2021b) on four standard NLP tasks in Table 1 is 83 .

Original: 9% using ε = 8 (and the same δ as in Table 1), whereas we can achieve an average accuracy of 90 .
Revised: 9% using ε = 8 (and the same δ as in Table 1), whereas we can achieve an average accuracy of 90 .

Original: 3% using ε = 6 .
Revised: 3% using ε = 6 .

Original: 7 by a combination of better algorithms, larger models, and new privacy accounting techniques.
Revised: 7 by a combination of better algorithms, larger models, and new privacy accounting techniques.

Original: Finally, though recently considered elsewhere (see Section 5), we put further focus on the framing of public pre-training and private fine-tuning as an important conceptual direction in DP deep learning.
Revised: Finally, though recently considered elsewhere (see Section 5), we put further focus on the framing of public pre-training and private fine-tuning as an important conceptual direction in DP deep learning.

Original:  P RELIMINARIES AND P RIOR A LGORITHM B ASELINES
Revised:  P RELIMINARIES AND P RIOR A LGORITHM B ASELINES

Original: Recall the formal definition of differential privacy.
Revised: Recall the formal definition of differential privacy.

Original: Definition 2.1 (Differential Privacy (DP) (Dwork et al., 2006b;a)) .
Revised: Definition 2.1 (Differential Privacy (DP) (Dwork et al., 2006b;a)) .

Original: A randomized algorithm
Revised: A randomized algorithm

Original: A is ( ε , δ )-differentially private if for any two neighboring datasets D and D ′ , which differ in exactly the data pertaining to a single user, and for all sets S of possible outputs:
Revised: A is ( ε , δ )-differentially private if for any two neighboring datasets D and D ′ , which differ in exactly the data pertaining to a single user, and for all sets S of possible outputs:

Original: Pr[ A ( D ) ∈ S
Revised: Pr[ A ( D ) ∈ S

Original: [Equation]
Revised: [Equation]

Original: We review prior techniques for private fine-tuning.
Revised: We review prior techniques for private fine-tuning.

Original: 2. F ULL F INE - TUNING VIA DPSGD
Revised: 2. F ULL F INE - TUNING VIA DPSGD

Original: To train a machine learning model with privacy, the most popular algorithm is the celebrated DP stochastic gradient descent (DPSGD)
Revised: To train a machine learning model with privacy, the most popular algorithm is the celebrated DP stochastic gradient descent (DPSGD)

Original: (Song et al., 2013; Bassily et al., 2014; Abadi et al., 2016).
Revised: (Song et al., 2013; Bassily et al., 2014; Abadi et al., 2016).

Original: This optimization method serves as a drop-in replacement for SGD, augmenting it with the addition of per-example gradient clipping and Gaussian noise addition steps.
Revised: This optimization method serves as a drop-in replacement for SGD, augmenting it with the addition of per-example gradient clipping and Gaussian noise addition steps.

Original: These two steps serve to limit and mask the contribution of a single example.
Revised: These two steps serve to limit and mask the contribution of a single example.

Original: Two key points to note are that a) per-example gradient clipping incurs significant computational and memory overheads in most implementations, and b) noise introduced due to privacy grows as the square-root as the number of model parameters.
Revised: Two key points to note are that a) per-example gradient clipping incurs significant computational and memory overheads in most implementations, and b) noise introduced due to privacy grows as the square-root as the number of model parameters.

Original: With this tool in place, the most basic fine-tuning strategy is to train all parameters using DPSGD.
Revised: With this tool in place, the most basic fine-tuning strategy is to train all parameters using DPSGD.

Original: 2. R EPARAMETRIZED G RADIENT P ERTURBATION
Revised: 2. R EPARAMETRIZED G RADIENT P ERTURBATION

Original: To mitigate the limitations of DPSGD, a recent work of Yu et al.
Revised: To mitigate the limitations of DPSGD, a recent work of Yu et al.

Original: (2021b) introduced an elegant method called reparametrized gradient perturbation (RGP).
Revised: (2021b) introduced an elegant method called reparametrized gradient perturbation (RGP).

Original: RGP exploits the implicit low-rank structure in the gradient updates of SGD to substantially improve upon DPSGD.
Revised: RGP exploits the implicit low-rank structure in the gradient updates of SGD to substantially improve upon DPSGD.

Original: Specifically, they reparametrize each layer’s weight matrix W into LR + ˜ W , where L and R are low-rank gradientcarrier matrices and ˜ W is the residual weight.
Revised: Specifically, they reparametrize each layer’s weight matrix W into LR + ˜ W , where L and R are low-rank gradientcarrier matrices and ˜ W is the residual weight.

Original: The authors show that one can obtain a lowdimensional projection of W ’s gradient by taking gradients only of the low-rank matrices L and R (and not the high-rank ˜ W ).
Revised: The authors show that one can obtain a lowdimensional projection of W ’s gradient by taking gradients only of the low-rank matrices L and R (and not the high-rank ˜ W ).

Original: Privacy is introduced by clipping and noising these low-dimensional gradients of L and R .
Revised: Privacy is introduced by clipping and noising these low-dimensional gradients of L and R .

Original: While this low-dimensional projection loses some of the signal in W ’s gradient, it turns out to contain enough to still achieve high accuracy.
Revised: While this low-dimensional projection loses some of the signal in W ’s gradient, it turns out to contain enough to still achieve high accuracy.

Original: At the same time, the low-dimensional gradients alleviate the aforementioned issues related to privatization, significantly reducing the memory consumption and noise introduced.
Revised: At the same time, the low-dimensional gradients alleviate the aforementioned issues related to privatization, significantly reducing the memory consumption and noise introduced.

Original:  O UR A PPROACH 3. A M ETA - FRAMEWORK
Revised:  O UR A PPROACH 3. A M ETA - FRAMEWORK

Original: We introduce our approach as a meta-framework for private deep learning, which abstracts the key principles of recent fine-tuning methods.
Revised: We introduce our approach as a meta-framework for private deep learning, which abstracts the key principles of recent fine-tuning methods.

Original: Suppose f ( W PT ; x ) is a pre-trained model where W PT are the pre-trained weights and x is any input.
Revised: Suppose f ( W PT ; x ) is a pre-trained model where W PT are the pre-trained weights and x is any input.

Original: We create a new fine-tuning model
Revised: We create a new fine-tuning model

Original: [Equation]
Revised: [Equation]

Original: which incorporates additional trainable parameters θ , where dim( θ ) ≪ dim( W PT )
Revised: which incorporates additional trainable parameters θ , where dim( θ ) ≪ dim( W PT )

Original: That is, the number of new parameters in θ is a small fraction of the original number of parameters in the pretrained weights W PT .
Revised: That is, the number of new parameters in θ is a small fraction of the original number of parameters in the pretrained weights W PT .

Original: Fine-tuning is done by running DPSGD on the additional parameters θ , while freezing the weights of pre-trained model W PT .
Revised: Fine-tuning is done by running DPSGD on the additional parameters θ , while freezing the weights of pre-trained model W PT .

Original: The new parameters are initialized to θ 0 such that
Revised: The new parameters are initialized to θ 0 such that

Original: [Equation]
Revised: [Equation]

Original: The initialization condition (2) is very important, as it ensures that fine-tuning starts at the pre-trained model and improves it by modifying the parameters θ .
Revised: The initialization condition (2) is very important, as it ensures that fine-tuning starts at the pre-trained model and improves it by modifying the parameters θ .

Original: Most fine-tuning methods are additive and have the following special form:
Revised: Most fine-tuning methods are additive and have the following special form:

Original: [Equation]
Revised: [Equation]

Original: i.e., they modify the pre-trained weights by adding a correction term π ( θ ) parametrized by θ .
Revised: i.e., they modify the pre-trained weights by adding a correction term π ( θ ) parametrized by θ .

Original: Recent work in the non-private literature has described concrete instantiations of this framework (Houlsby et al., 2019; Mahabadi et al., 2021;
Revised: Recent work in the non-private literature has described concrete instantiations of this framework (Houlsby et al., 2019; Mahabadi et al., 2021;

Original: Hu et al., 2021), which (crucially) are effective when dim( θ ) ≪ dim( W PT ) .
Revised: Hu et al., 2021), which (crucially) are effective when dim( θ ) ≪ dim( W PT ) .

Original: In the non-private setting, such reparametrizations are useful for reducing the computation and memory required for fine-tuning, and enable lightweight and plug-in modifications to the base model for different downstream tasks.
Revised: In the non-private setting, such reparametrizations are useful for reducing the computation and memory required for fine-tuning, and enable lightweight and plug-in modifications to the base model for different downstream tasks.

Original: At the same time, they maintain (or sometimes surpass) the accuracy achieved by full fine-tuning.
Revised: At the same time, they maintain (or sometimes surpass) the accuracy achieved by full fine-tuning.

Original: However, this approach seems to be more important for achieving high accuracy in the context of private fine-tuning.
Revised: However, this approach seems to be more important for achieving high accuracy in the context of private fine-tuning.

Original: For instance, full fine-tuning of RoBERTa-Base with DPSGD achieves an accuracy of 53 . 1% for the MNLI dataset (Yu et al., 2021b), whereas all the algorithms that fine-tune based on Equation (1) all achieve accuracy better than 80% .
Revised: For instance, full fine-tuning of RoBERTa-Base with DPSGD achieves an accuracy of 53 . 1% for the MNLI dataset (Yu et al., 2021b), whereas all the algorithms that fine-tune based on Equation (1) all achieve accuracy better than 80% .

Original: We give some intuition as to why parameter-efficient methods seem to be more effective for private fine-tuning.
Revised: We give some intuition as to why parameter-efficient methods seem to be more effective for private fine-tuning.

Original: For simplicity, we assume that the fine-tuning method is additive as in (3), such that the fine-tuned weights W FT = W PT + π ( θ ) .
Revised: For simplicity, we assume that the fine-tuning method is additive as in (3), such that the fine-tuned weights W FT = W PT + π ( θ ) .

Original: We can imagine that W FT lies on a manifold passing through W PT of very small dimension (equal to the dimension of θ ) compared to the dimension of
Revised: We can imagine that W FT lies on a manifold passing through W PT of very small dimension (equal to the dimension of θ ) compared to the dimension of

Original: W PT .
Revised: W PT .

Original: Even if the parameters θ are very noisy due to the noise added during DPSGD, we will always stay in this manifold.
Revised: Even if the parameters θ are very noisy due to the noise added during DPSGD, we will always stay in this manifold.

Original: In particular, we are not disturbing the pre-trained weights in most directions (those orthogonal to the manifold near W PT ).
Revised: In particular, we are not disturbing the pre-trained weights in most directions (those orthogonal to the manifold near W PT ).

Original: If we run DPSGD on all the weights instead, then we add noise in all directions, thus potentially unlearning the knowledge learned during pre-training, especially in low data regimes.
Revised: If we run DPSGD on all the weights instead, then we add noise in all directions, thus potentially unlearning the knowledge learned during pre-training, especially in low data regimes.

Original: Besides substantial gains in the accuracy, the above method of reparametrization has several other advantages:
Revised: Besides substantial gains in the accuracy, the above method of reparametrization has several other advantages:

Original: • A single pre-trained model such as BERT or GPT is generally applied to hundreds of downstream tasks via fine-tuning.
Revised: • A single pre-trained model such as BERT or GPT is generally applied to hundreds of downstream tasks via fine-tuning.

Original: Private fine-tuning using previous methods requires updating all parameters and storing a different copy of the fine-tuned model per task.
Revised: Private fine-tuning using previous methods requires updating all parameters and storing a different copy of the fine-tuned model per task.

Original: This creates substantial overheads for storing and deploying, and can be very expensive in practice.
Revised: This creates substantial overheads for storing and deploying, and can be very expensive in practice.

Original: On the other hand, the reparametrization (1) means that we only need to store a single pretrained model that can be shared across many downstream tasks.
Revised: On the other hand, the reparametrization (1) means that we only need to store a single pretrained model that can be shared across many downstream tasks.

Original: Each downstream task requires only a small number of new parameters that can be plugged in.
Revised: Each downstream task requires only a small number of new parameters that can be plugged in.

Original: • Differentially private training requires computing and storing per-example gradients, which increases the memory footprint.
Revised: • Differentially private training requires computing and storing per-example gradients, which increases the memory footprint.

Original: In our approach, however, learning is done in a much lower dimension, hence saving on the memory cost as compared to prior works.
Revised: In our approach, however, learning is done in a much lower dimension, hence saving on the memory cost as compared to prior works.

Original: Finally, we expect that (1) also gives a more communication-efficient method of fine-tuning in distributed settings such as federated learning, due to the significantly smaller number of parameters learned during fine-tuning.
Revised: Finally, we expect that (1) also gives a more communication-efficient method of fine-tuning in distributed settings such as federated learning, due to the significantly smaller number of parameters learned during fine-tuning.

Original: 3. I NSTANTIATING THE M ETA - FRAMEWORK
Revised: 3. I NSTANTIATING THE M ETA - FRAMEWORK

Original: In this section, we discuss a few ways to instantiate our meta-framework.
Revised: In this section, we discuss a few ways to instantiate our meta-framework.

Original: This list is non-exhaustive, but covers the methods we employ in our experiments.
Revised: This list is non-exhaustive, but covers the methods we employ in our experiments.

Original: 3.2. F INE - TUNING VIA L OW -R ANK A DAPTATION
Revised: 3.2. F INE - TUNING VIA L OW -R ANK A DAPTATION

Original: Low-Rank Adaptation (LoRA) (Hu et al., 2021) is an additive fine-tuning scheme as defined in (3).
Revised: Low-Rank Adaptation (LoRA) (Hu et al., 2021) is an additive fine-tuning scheme as defined in (3).

Original: For each dense weight matrix W i PT of size a × b in the pre-trained network, we add a low-rank correction term L i R i , i.e.,
Revised: For each dense weight matrix W i PT of size a × b in the pre-trained network, we add a low-rank correction term L i R i , i.e.,

Original: [Equation]
Revised: [Equation]

Original: 4) where L i ∈ R a × r , R
Revised: 4) where L i ∈ R a × r , R

Original: i ∈
Revised: i ∈

Original: R r × b are new trainable parameters.
Revised: R r × b are new trainable parameters.

Original: Hu et al.
Revised: Hu et al.

Original: (2021) apply this reparameterization only to the Transformer attention weights ( W q , W v ), and freeze all other weights (e.g., W k and W o and those in the feed-forward layers).
Revised: (2021) apply this reparameterization only to the Transformer attention weights ( W q , W v ), and freeze all other weights (e.g., W k and W o and those in the feed-forward layers).

Original: The rank r is typically chosen to be small, e.g., r = 4 , 16 , 64 .
Revised: The rank r is typically chosen to be small, e.g., r = 4 , 16 , 64 .

Original: Since most parameters in Transformer architectures are dense weight matrices, choosing a small r results in a nearly square-root reduction in the number of parameters.
Revised: Since most parameters in Transformer architectures are dense weight matrices, choosing a small r results in a nearly square-root reduction in the number of parameters.

Original: 3.2. F INE - TUNING VIA A DAPTERS
Revised: 3.2. F INE - TUNING VIA A DAPTERS

Original: Houlsby et al.
Revised: Houlsby et al.

Original: (2019) propose adapter-based fine-tuning, in which we modify the architecture of the pre-trained model by adding new “adapter” layers after each attention and feed-forward layer.
Revised: (2019) propose adapter-based fine-tuning, in which we modify the architecture of the pre-trained model by adding new “adapter” layers after each attention and feed-forward layer.

Original: Adapter layers are bottleneck layers with residual connections.
Revised: Adapter layers are bottleneck layers with residual connections.

Original: Specifically, given an input x , an adapter layer A performs
Revised: Specifically, given an input x , an adapter layer A performs

Original: [Equation]
Revised: [Equation]

Original: where U is an up-projection affine linear map, D is a down-projection affine linear map, and τ is a non-linear activation function such as the Gaussian error Linear Unit (GeLU)
Revised: where U is an up-projection affine linear map, D is a down-projection affine linear map, and τ is a non-linear activation function such as the Gaussian error Linear Unit (GeLU)

Original: (Hendrycks &
Revised: (Hendrycks &

Original: Gimpel, 2016).
Revised: Gimpel, 2016).

Original: If x has dimension d , then U ∈ R d × r , D ∈ R r × d for some r ≪ d .
Revised: If x has dimension d , then U ∈ R d × r , D ∈ R r × d for some r ≪ d .

Original: Thus, the number of introduced parameters is significantly less than the number of parameters in the pre-trained model.
Revised: Thus, the number of introduced parameters is significantly less than the number of parameters in the pre-trained model.

Original: When fine-tuning, the parameters of the original model are frozen, and only parameters of the adapter layers, as well as layer normalizations, are
Revised: When fine-tuning, the parameters of the original model are frozen, and only parameters of the adapter layers, as well as layer normalizations, are

Original: [Table]
Revised: [Table]

Original: 3.2. F INE - TUNING VIA C OMPACTER
Revised: F INE - TUNING VIA C OMPACTER

Original: The recent work of Mahabadi et al.
Revised: The recent work of Mahabadi et al.

Original: (2021) introduces Compacters (Compact adapters), a method which further improves the parameter efficiency of adapters.
Revised: (2021) introduces Compacters (Compact adapters), a method which further improves the parameter efficiency of adapters.

Original: This is done by replacing the dense matrices in the up-projection U and down-projection D by tensor products of smaller matrices, thus reducing the number of trainable parameters.
Revised: This is done by replacing the dense matrices in the up-projection U and down-projection D by tensor products of smaller matrices, thus reducing the number of trainable parameters.

Original: Specifically, they replace the dense matrix M ℓ in the adapter layer ℓ by a low-rank parameterized hypercomplex multiplication (LPHM) layer, i.e., each dense matrix M ℓ ∈ R a × b is expressed as
Revised: Specifically, they replace the dense matrix M ℓ in the adapter layer ℓ by a low-rank parameterized hypercomplex multiplication (LPHM) layer, i.e., each dense matrix M ℓ ∈ R a × b is expressed as

Original: [Equation]
Revised: [Equation]

Original: where A i ∈ R n × n , S ℓi ∈ R a/n × k , T ℓi ∈ R k × b/n and ⊗ is the matrix Kronecker product.
Revised: where A i ∈ R n × n , S ℓi ∈ R a/n × k , T ℓi ∈ R k × b/n and ⊗ is the matrix Kronecker product.

Original: Note the matrices A i are not indexed by the layer ℓ because these matrices are shared among all the adapter layers.
Revised: Note the matrices A i are not indexed by the layer ℓ because these matrices are shared among all the adapter layers.

Original: Since each adapter layers has two dense matrices (one for up-projection and one for down-projection), if there are L adapter layers, this reduces the number of parameters from L (2 ab ) to L (2( a + b ) k ) + n 3 .
Revised: Since each adapter layers has two dense matrices (one for up-projection and one for down-projection), if there are L adapter layers, this reduces the number of parameters from L (2 ab ) to L (2( a + b ) k ) + n 3 .

Original: In practice, a and b are chosen to be either the model dimension d or the intermediate representation dimension r in the adapters, n is typically chosen to be a small constant such as n = 2 , 4 , 8 , 12 and k is chosen to be 1 .
Revised: In practice, a and b are chosen to be either the model dimension d or the intermediate representation dimension r in the adapters, n is typically chosen to be a small constant such as n = 2 , 4 , 8 , 12 and k is chosen to be 1 .

Original: [Equation]
Revised: [Equation]

Original: FFICIENT T UNING W ORK ?
Revised: FFICIENT T UNING W ORK ?

Original: Theoretical explanation of success of parameter-efficient fine-tuning methods is active area of research in deep learning.
Revised: Theoretical explanation of success of parameter-efficient fine-tuning methods is active area of research in deep learning.

Original: Indeed, since trends have consistently shown that model accuracy increases with size, how can one achieve competitive accuracy while fine-tuning less than 1% of the parameters?
Revised: Indeed, since trends have consistently shown that model accuracy increases with size, how can one achieve competitive accuracy while fine-tuning less than 1% of the parameters?

Original: One popular hypothesis is intrinsic dimensionality (Li et al., 2018), which posits that the minimum number of parameters needed to train a machine learning model may be much less than the total number of model parameters.
Revised: One popular hypothesis is intrinsic dimensionality (Li et al., 2018), which posits that the minimum number of parameters needed to train a machine learning model may be much less than the total number of model parameters.

Original: Aghajanyan et al.
Revised: Aghajanyan et al.

Original: (2020) explore this hypothesis in the context of fine-tuning LLMs, showing that one can achieve most of their accuracy by training only a very small number of parameters (chosen via a random projection).
Revised: (2020) explore this hypothesis in the context of fine-tuning LLMs, showing that one can achieve most of their accuracy by training only a very small number of parameters (chosen via a random projection).

Original: Perhaps surprisingly, they find that as the model size increases, intrinsic dimension decreases , in the limit exhibiting zero-shot learning.
Revised: Perhaps surprisingly, they find that as the model size increases, intrinsic dimension decreases , in the limit exhibiting zero-shot learning.

Original: While we did not explore this hypothesis in the context of DP due to computational restrictions, we believe it may be an interesting lens through which one can understand the effectiveness of private parameter-efficient fine-tuning.
Revised: While we did not explore this hypothesis in the context of DP due to computational restrictions, we believe it may be an interesting lens through which one can understand the effectiveness of private parameter-efficient fine-tuning.

Original: 3. C OMPARISION WITH B ASELINE A LGORITHMS
Revised: 3. C OMPARISION WITH B ASELINE A LGORITHMS

Original: We highlight some key algorithmic differences between our proposed methods and the baselines of full fine-tuning and RGP.
Revised: We highlight some key algorithmic differences between our proposed methods and the baselines of full fine-tuning and RGP.

Original: • DPSGD and RGP both require updating all parameters of the pre-trained model, whereas our proposed methods update only a tiny fraction (between 0 .
Revised: • DPSGD and RGP both require updating all parameters of the pre-trained model, whereas our proposed methods update only a tiny fraction (between 0 .

Original: 05% and 1% ).
Revised: 05% and 1% ).

Original: The rightmost columns of Tables 3 and 4 list the number of parameters trained by these algorithms.
Revised: The rightmost columns of Tables 3 and 4 list the number of parameters trained by these algorithms.

Original: • RGP performs a low-rank decomposition of weight matrices which is similar to LoRA, though there are subtle differences.
Revised: • RGP performs a low-rank decomposition of weight matrices which is similar to LoRA, though there are subtle differences.

Original: Recall that in RGP, at the beginning of each iteration t , the historical weight matrix W t − 1 is decomposed to find a low-rank product LR .
Revised: Recall that in RGP, at the beginning of each iteration t , the historical weight matrix W t − 1 is decomposed to find a low-rank product LR .

Original: The gradients on L and R are then projected back to the full parameter space to perform the descent step.
Revised: The gradients on L and R are then projected back to the full parameter space to perform the descent step.

Original: Hence, RGP keeps modifying the pre-trained weights during learning.
Revised: Hence, RGP keeps modifying the pre-trained weights during learning.

Original: LoRA can be viewed as a simplification of RGP.
Revised: LoRA can be viewed as a simplification of RGP.

Original: LoRA reparametrizes W FT :
Revised: LoRA reparametrizes W FT :

Original: = W PT + LR , where the pre-trained weight matrix W PT is frozen during training.
Revised: = W PT + LR , where the pre-trained weight matrix W PT is frozen during training.

Original: Hence, compared to
Revised: Hence, compared to

Original: [Table]
Revised: [Table]

Original: RGP, LoRA eliminates the decomposition and the projection to the full parameter space at each iteration, simplifying the implementation andreducing the running time and memory cost.
Revised: RGP, LoRA eliminates the decomposition and the projection to the full parameter space at each iteration , simplifying the implementation and

Revised: reducing the running time and memory cost

Original: This is summarized in Table 2.
Revised: This is summarized in Table 2.

Original: We observe that DP LoRA reduces the memory cost by about 33% and the training speed by 8% . 
Revised: We observe that DP LoRA reduces the memory cost by about 33% and the training speed by 8% . As

Original: As we will see, this simplification also results in improved utility.
Revised: we will see , this simplification also results in improved utility.

Original: • Neither full fine-tuning nor RGP fall into our meta-framework described by (1).
Revised: • Neither full fine-tuning nor RGP fall into our meta-framework described by (1).

Original: Thus, if a pre-trained model is to be applied to several downstream tasks, one must store a separate set of weights for each task, incurring a significant memory cost and losing the plug-in
Revised: Thus, if a pre-trained model is to be applied to several downstream tasks, one must store a separate set of weights for each task, incurring a significant memory cost and losing the plug-in

Original: functionality.
Revised: functionality.

Original: In contrast, our methods are much more lightweight.
Revised: In contrast, our methods are much more lightweight.

Original:  E XPERIMENTS
Revised:  E XPERIMENTS

Original: We experimentally evaluate our methods for DP fine-tuning to demonstrate their utility, privacy, and parameter-efficiency.
Revised: We experimentally evaluate our methods for DP fine-tuning to demonstrate their utility, privacy, and parameter-efficiency.

Original: We investigate both language understanding and text generation tasks to establish that our techniques are applicable to a variety of tasks and model architectures.
Revised: We investigate both language understanding and text generation tasks to establish that our techniques are applicable to a variety of tasks and model architectures.

Original: Our code is publicly available at https://github.com/AnonymousAKES/
Revised: Our code is publicly available at https://github.com/AnonymousAKES/

Original: Differentially-Private-Fine-tuning-of-Language-Models .
Revised: Differentially-Private-Fine-tuning-of-Language-Models .

Original: 4. F INE -T UNING FOR L ANGUAGE U NDERSTANDING T ASKS
Revised: 4. F INE -T UNING FOR L ANGUAGE U NDERSTANDING T ASKS

Original: We first compare our methods with state-of-the-art fine-tuning algorithms using models from the
Revised: We first compare our methods with state-of-the-art fine-tuning algorithms using models from the

Original: BERT family, which was used in the prior work (Yu et al., 2021b).
Revised: BERT family, which was used in the prior work (Yu et al., 2021b).

Original: Specifically, we use RoBERTa models (Liu et al., 2019), which are pre-trained on public data collected from the web.
Revised: Specifically, we use RoBERTa models (Liu et al., 2019), which are pre-trained on public data collected from the web.

Original: RoBERTaBase has 125M parameters and RoBERTa-Large has 355M parameters.
Revised: RoBERTaBase has 125M parameters and RoBERTa-Large has 355M parameters.

Original: We choose four downstream tasks: MNLI, QQP, QNLI, and SST-2 from GLUE (Wang et al., 2018), following Yu et al.
Revised: We choose four downstream tasks: MNLI, QQP, QNLI, and SST-2 from GLUE (Wang et al., 2018), following Yu et al.

Original: (2021b).
Revised: (2021b).

Original: Implementation Details: For fine-tuning with adapters, we may choose the intermediate representation dimension r , shared across all adapter layers.
Revised: Implementation Details: For fine-tuning with adapters, we may choose the intermediate representation dimension r , shared across all adapter layers.

Original: For fine-tuning with Compacter, we can choose both r and the Kronecker product kernel dimension n in (6).
Revised: For fine-tuning with Compacter, we can choose both r and the Kronecker product kernel dimension n in (6).

Original: For LoRA fine-tuning, we add bottleneck branches for both the attention layers and the feedforward layers, which differs slightly from the addition of bottleneck branches for only the W q and W v matrices of the attention layers as done by Hu et al.
Revised: For LoRA fine-tuning, we add bottleneck branches for both the attention layers and the feedforward layers, which differs slightly from the addition of bottleneck branches for only the W q and W v matrices of the attention layers as done by Hu et al.

Original: Given the same bottleneck representation dimension r in (4), our new implementation uses twice as many trainable parameters as the original paper, and achieves some improvements for learning with DP.
Revised: Given the same bottleneck representation dimension r in (4), our new implementation uses twice as many trainable parameters as the original paper, and achieves some improvements for learning with DP.

Original: We perform privacy accounting using the approach of Gopi et al.
Revised: We perform privacy accounting using the approach of Gopi et al.

Original: (2021), which currently gives the tightest bounds.
Revised: (2021), which currently gives the tightest bounds.

Original: Hyperparameter choice: Given the large number of hyperparameter choices, e.g., the intermediate representation dimension, learning rate, weight decay, privacy parameter δ , and model size, an exhaustive grid search over all hyperparameters is expensive.
Revised: Hyperparameter choice: Given the large number of hyperparameter choices, e.g., the intermediate representation dimension, learning rate, weight decay, privacy parameter δ , and model size, an exhaustive grid search over all hyperparameters is expensive.

Original: Our hyperparameter choices are informed by prior work and are as follows.
Revised: Our hyperparameter choices are informed by prior work and are as follows.

Original: For privacy parameters, we use δ = 1e-5 for SSTand QNLI and δ = 1e-6 for QQP and MNLI due to their dataset sizes, and use noise multipliers 0 . 92 , 0 . 83 , 0 . 66 and 0 .
Revised: For privacy parameters, we use δ = 1e-5 for SSTand QNLI and δ = 1e-6 for QQP and MNLI due to their dataset sizes, and use noise multipliers 0 . 92 , 0 . 83 , 0 . 66 and 0 .

Original: 65 for SST-2, QNLI, QQP, and MNLI, respectively, which is the same as Yu et al.
Revised: 65 for SST-2, QNLI, QQP, and MNLI, respectively, which is the same as Yu et al.

Original: (2021b).
Revised: (2021b).

Original: In Appendix B, we run experiments under different privacy parameters.
Revised: In Appendix B, we run experiments under different privacy parameters.

Original: The proposed framework performs well under a wide range choices of ε and δ .
Revised: The proposed framework performs well under a wide range choices of ε and δ .

Original: The clipping threshold is 10 for
Revised: The clipping threshold is 10 for

Original: [Table]
Revised: [Table]

Original: all methods.
Revised: all methods.

Original: The batch size is 2000.
Revised: The batch size is 2000.

Original: In Appendix D, we show the performance of the proposed algorithm is stable across a wide range of choices of clipping thresholds and batch sizes.
Revised: In Appendix D, we show the performance of the proposed algorithm is stable across a wide range of choices of clipping thresholds and batch sizes.

Original: For adapters and Compacter, we follow the original papers and choose r from { 16 , 48 , 96 } and n from { 4 , 8 , 12 } .
Revised: For adapters and Compacter, we follow the original papers and choose r from { 16 , 48 , 96 } and n from { 4 , 8 , 12 } .

Original: For LoRA, we choose the best-performing rank r from the set { 4 , 16 , 48 , 64 } .
Revised: For LoRA, we choose the best-performing rank r from the set { 4 , 16 , 48 , 64 } .

Original: The best performing hyperparameters are noted in Tables 3 and 4.
Revised: The best performing hyperparameters are noted in Tables 3 and 4.

Original: We train for 20 epochs using AdamW (Loshchilov &
Revised: We train for 20 epochs using AdamW (Loshchilov &

Original: Hutter, 2019) with weight decay 1e-2 and search over four learning rates { 5e-4, 1e-3, 2e-3, 5e-3 } .
Revised: Hutter, 2019) with weight decay 1e-2 and search over four learning rates { 5e-4, 1e-3, 2e-3, 5e-3 } .

Original: Results: We report the prediction accuracy in Tables 3 and 4.
Revised: Results: We report the prediction accuracy in Tables 3 and 4.

Original: Our experiments using RoBERTaBase serve as a direct comparison to Yu et al.
Revised: Our experiments using RoBERTaBase serve as a direct comparison to Yu et al.

Original: (2021b) who only trained the base model, whereas
Revised: (2021b) who only trained the base model, whereas

Original: RoBERTa-Large experiments demonstrate the significance of using larger models.
Revised: RoBERTa-Large experiments demonstrate the significance of using larger models.

Original: Our key findings are: (1) On all datasets, our methods achieve the best accuracy while training a tiny fraction of parameters; larger models give significant improvements.
Revised: Our key findings are: (1) On all datasets, our methods achieve the best accuracy while training a tiny fraction of parameters; larger models give significant improvements.

Original: (2) Noticeable improvements in ε versus Yu et al.
Revised: (2) Noticeable improvements in ε versus Yu et al.

Original: (2021b) are primarily due to new privacy accountants based on Fourier-based numerical composition (Koskela et al., 2020; 2021; Gopi et al., 2021); we use the accountant in Gopi et al.
Revised: (2021b) are primarily due to new privacy accountants based on Fourier-based numerical composition (Koskela et al., 2020; 2021; Gopi et al., 2021); we use the accountant in Gopi et al.

Original: (2021) since it is the most efficient.
Revised: (2021) since it is the most efficient.

Original: (3) Private adapters provide the best average performance for
Revised: (3) Private adapters provide the best average performance for

Original: RoBERTa-Base, whereas LoRA outperforms all other methods for RoBERTa-Large.
Revised: RoBERTa-Base, whereas LoRA outperforms all other methods for RoBERTa-Large.

Original: Remark: While our experiments indicate that full fine-tuning does not achieve competitive performance, there could be a choice of hyperparameters that improves upon the reported numbers, e.g., “mega” batch sizes (in the millions) in Anil et al.
Revised: Remark: While our experiments indicate that full fine-tuning does not achieve competitive performance, there could be a choice of hyperparameters that improves upon the reported numbers, e.g., “mega” batch sizes (in the millions) in Anil et al.

Original: We note that our main message is that one does not need to fine-tune all parameters to achieve the best accuracy.
Revised: We note that our main message is that one does not need to fine-tune all parameters to achieve the best accuracy.

Original: Nevertheless, it is interesting to wonder if full fine-tuning with DPSGD can match the accuracy of parameter-efficient methods.
Revised: Nevertheless, it is interesting to wonder if full fine-tuning with DPSGD can match the accuracy of parameter-efficient methods.

Original: A positive answer would imply that private and non-private fine-tuning conceptually mirror each other.
Revised: A positive answer would imply that private and non-private fine-tuning conceptually mirror each other.

Original: 4. F INE - TUNING FOR N ATURAL L ANGUAGE G ENERATION (NLG)
Revised: 4. F INE - TUNING FOR N ATURAL L ANGUAGE G ENERATION (NLG)

Original: Next, we study private fine-tuning for text generation problems using the GPT-2 series of models on the End-2-End (E2E)
Revised: Next, we study private fine-tuning for text generation problems using the GPT-2 series of models on the End-2-End (E2E)

Original: NLG challenge (Novikova et al., 2017), one of the primary benchmarks used in recent works on non-private fine-tuning (Hu et al., 2021; Li & Liang, 2021).
Revised: NLG challenge (Novikova et al., 2017), one of the primary benchmarks used in recent works on non-private fine-tuning (Hu et al., 2021; Li & Liang, 2021).

Original: We use GPT-2-Small (117M parameters), GPT-2-Medium (345M parameters), and GPT-2-Large (774M parameters).
Revised: We use GPT-2-Small (117M parameters), GPT-2-Medium (345M parameters), and GPT-2-Large (774M parameters).

Original: To the best of our knowledge, we are the first to privately fine-tune for E2E or fine-tune GPT-2-Large.
Revised: To the best of our knowledge, we are the first to privately fine-tune for E2E or fine-tune GPT-2-Large.

Original: The purpose of this section is not to evaluate various fine-tuning algorithms, but to show that private fine-tuning is competitive with non-private fine-tuning for text generation problems.
Revised: The purpose of this section is not to evaluate various fine-tuning algorithms, but to show that private fine-tuning is competitive with non-private fine-tuning for text generation problems.

Original: Due to the high cost of training, we report experimental results only for fine-tuning with LoRA.
Revised: Due to the high cost of training, we report experimental results only for fine-tuning with LoRA.

Revised: In Appendix F, we present additional experiments on NLG that include private fine-tuning of the

Revised: GPT-2-XL model with 1.5 billion parameters.

Revised: Other noticeable points in the additional experiments include 1) we show improved performance using better hyperparameters; 2) we test different privacy parameters; 3) we consider a new dataset DART (Nan et al., 2021).

Original: E2E NLG challenge: The E2E dataset was introduced by Novikova et al.
Revised: E2E NLG challenge: The E2E dataset in Novikova et al.

Original: (2017), and contains template-like information in the restaurant domain to be mapped to natural language with end-toend training.
Revised: (2017) contains template-like information in the restaurant domain to be mapped to natural language with end-to-end training.

Original: The dataset consists of 42K training samples, 4.6K validation samples, and 4.6K test samples.
Revised: The dataset consists of 42K training samples, 4.6K validation samples, and 4.6K test samples.

Original: We use standard metrics such as BLUE, ROUGE-L, etc., used in (Hu et al., 2021) for measuring the quality of predictions.
Revised: We use standard metrics such as BLUE, ROUGE-L, etc., used in (Hu et al., 2021) for evaluation.

Original: Hyperparameter choice: For LoRA, we choose the bottleneck rank r = 4 in (4) and fine-tune W q and W v matrices of the attention layers as in the original paper.
Revised: Hyperparameter choice: For LoRA, we choose the bottleneck rank r = 4 in (4) and fine-tune W q and W v matrices of the attention layers as in the original paper.

Original: We optimize using AdamW with  learning rate 2e-4, weight decay 1e-2 and train our models for 5 epochs using batch size 64.
Revised: We optimize using AdamW with [Table] learning rate 2e-4, weight decay 1e-2 and train our models for 5 epochs using batch size 64.

Original: We take the gradient clipping parameter to be 1.0 and  the noise multiplier to be 0.5 for the accountant in Gopi et al.
Revised: We take the gradient clipping parameter to be 1.0 and set the noise multiplier as 0.5.

Original: (2021), achieving ε = 5 . 4 , δ = 1e-5. [Table] Results: The results of our experiments are summarized in the Table 5, which reiterate the main themes of this paper: private fine-tuning with a parameter-efficient approach performs close to their non-private counterparts and show consistent improvement in the utility as the model size increases.
Revised: Results: The results of our experiments are summarized in the Table 5, which reiterate the main themes of this paper: private fine-tuning with a parameter-efficient approach performs close to their non-private counterparts and show consistent improvement in the utility as the model size increases.

Original:  R ELATED W ORK
Revised:  R ELATED W ORK

Original: More on DP learning: Some work studies private language models on more traditional architectures such as LSTMs (Hochreiter & Schmidhuber, 1997), either training with DPSGD (McMahan et al., 2018; Carlini et al., 2019) or related heuristics (Ramaswamy et al., 2020).
Revised: More on DP learning: Some work studies private language models on more traditional architectures such as LSTMs (Hochreiter & Schmidhuber, 1997), either training with DPSGD (McMahan et al., 2018; Carlini et al., 2019) or related heuristics (Ramaswamy et al., 2020).

Original: Though pretraining on public data is suggested (McMahan et al., 2018), public data appears to only be used in one of these works for honest hyperparameter selection (Ramaswamy et al., 2020).
Revised: Though pretraining on public data is suggested (McMahan et al., 2018), public data appears to only be used in one of these works for honest hyperparameter selection (Ramaswamy et al., 2020).

Original: A few more recent works consider training LLMs with DP.
Revised: A few more recent works consider training LLMs with DP.

Original: Anil et al.
Revised: Anil et al.

Original: (2021) privately train BERT-Large from scratch, compared to our work which focuses on private fine-tuning.
Revised: (2021) privately train BERT-Large from scratch, compared to our work which focuses on private fine-tuning.

Original: (Hoory et al., 2021;
Revised: (Hoory et al., 2021;

Original: Basu et al., 2021) perform private full fine-tuning of BERT models.
Revised: Basu et al., 2021) perform private full fine-tuning of BERT models.

Original: Hoory et al.
Revised: Hoory et al.

Original: (2021) achieve accuracy which is comparable to the non-private model, but additionally supplement the public pre-training data with additional domain-relevant material, while we use off-the-shelf pre-trained models.
Revised: (2021) achieve accuracy which is comparable to the non-private model, but additionally supplement the public pre-training data with additional domain-relevant material, while we use off-the-shelf pre-trained models.

Original: Basu et al.
Revised: Basu et al.

Original: (2021) observe significant drops in utility, compared to our parameter-efficient methods which do not.
Revised: (2021) observe significant drops in utility, compared to our parameter-efficient methods which do not.

Original: While Kerrigan et al.
Revised: While Kerrigan et al.

Original: (2020) consider public pre-training and private fine-tuning, their experiments are on much smaller architectures (i.e., feedforward networks with three hidden layers).
Revised: (2020) consider public pre-training and private fine-tuning, their experiments are on much smaller architectures (i.e., feedforward networks with three hidden layers).

Original: A simultaneous work of Ginart et al.
Revised: A simultaneous work of Ginart et al.

Original: (2022) investigates private prediction (rather than learning) for next-token prediction.
Revised: (2022) investigates private prediction (rather than learning) for next-token prediction.

Original: A subsequent work by Senge et al.
Revised: A subsequent work by Senge et al.

Original: (2021) also investigates the effect of private fine-tuning on various NLP tasks.
Revised: (2021) also investigates the effect of private fine-tuning on various NLP tasks.

Original: In a concurrent work, Li et al.
Revised: In a concurrent work, Li et al.

Original: (2022) also investigate DP fine-tuning of LLMs.
Revised: (2022) also investigate DP fine-tuning of LLMs.

Original: In several cases, their results demonstrate qualitatively similar findings as ours.
Revised: In several cases, their results demonstrate qualitatively similar findings as ours.

Original: While our experiments focus primarily on parameter-efficient fine-tuning methods, interestingly, they show that private full fine-tuning can also achieve comparable utility if the experimental setup is configured properly, e.g., using suitable hyperparameters.
Revised: While our experiments focus primarily on parameter-efficient fine-tuning methods, interestingly, they show that private full fine-tuning can also achieve comparable utility if the experimental setup is configured properly, e.g., using suitable hyperparameters.

Original: In Appendix C, we run experiments under the setup in Li et al.
Revised: In Appendix C, we run experiments under the setup in Li et al.

Original: We show their setup can also improve the performance of our methods.
Revised: We show their setup can also improve the performance of our methods.

Original: More on Fine-tuning: There exist other parameter-efficient tuning methods which we did not evaluate in our work.
Revised: More on Fine-tuning: There exist other parameter-efficient tuning methods which we did not evaluate in our work.

Original: Some of these include random subspace projection (exploiting intrinsic dimensionality (Li et al., 2018;
Revised: Some of these include random subspace projection (exploiting intrinsic dimensionality (Li et al., 2018;

Original: Aghajanyan et al., 2020)), prefix and prompt tuning (Li & Liang, 2021; Lester et al., 2021), tuning only biases (Cai et al., 2020; Ben Zaken et al., 2021), and other architecture variants including Adapters (Pfeiffer et al., 2021; R¨uckl´e et al., 2020).
Revised: Aghajanyan et al., 2020)), prefix and prompt tuning (Li & Liang, 2021; Lester et al., 2021), tuning only biases (Cai et al., 2020; Ben Zaken et al., 2021), and other architecture variants including Adapters (Pfeiffer et al., 2021; R¨uckl´e et al., 2020).

Original: An interesting direction for future work is to see whether parameter-efficient tuning approaches specifically designed for the private setting can achieve higher utility.
Revised: An interesting direction for future work is to see whether parameter-efficient tuning approaches specifically designed for the private setting can achieve higher utility.

Original: We also mention zero-shot learning, in which no task-specific dataset is required and thus perfect privacy is achieved.
Revised: We also mention zero-shot learning, in which no task-specific dataset is required and thus perfect privacy is achieved.

Original: Currently, zero-shot approaches achieve low utility compared to fine-tuning, though it is possible that future models may narrow this gap.
Revised: Currently, zero-shot approaches achieve low utility compared to fine-tuning, though it is possible that future models may narrow this gap.

Original:  C ONCLUSION
Revised:  C ONCLUSION

Original: So far, DP deep learning has focused on training models from scratch.
Revised: So far, DP deep learning has focused on training models from scratch.

Original: The spectacular success of transfer learning in real-world applications, however, shows that private fine-tuning is an equally pertinent problem to study and deserves more attention.
Revised: The spectacular success of transfer learning in real-world applications, however, shows that private fine-tuning is an equally pertinent problem to study and deserves more attention.

Original: We show that by combining recent advances in
Revised: We show that by combining recent advances in

Original: NLP, parameter-efficiency, privacy accounting, and using larger models, one can privately fine-tune models whose utility approaches that of non-private models.
Revised: NLP, parameter-efficiency, privacy accounting, and using larger models, one can privately fine-tune models whose utility approaches that of non-private models.

Original: We hope our work inspires more study on the core problem of private fine-tuning, which we believe to be a central direction for research in private machine learning, leading to more interaction between the LLM and DP communities.
Revised: We hope our work inspires more study on the core problem of private fine-tuning, which we believe to be a central direction for research in private machine learning, leading to more interaction between the LLM and DP communities.

Original: A CKNOWLEDGMENTS
Revised: A CKNOWLEDGMENTS

Original: The authors would like to thank Rabeeh Karimi Mahabadi for sharing hyperparameters based on experiments in Mahabadi et al.
Revised: The authors would like to thank Rabeeh Karimi Mahabadi for sharing hyperparameters based on experiments in Mahabadi et al.

Original: Janardhan Kulkarni would like to thank Edward Hu for sharing many ideas on fine-tuning.
Revised: Janardhan Kulkarni would like to thank Edward Hu for sharing many ideas on fine-tuning.

Original: Gautam Kamath is supported by an NSERC Discovery Grant
Revised: Gautam Kamath is supported by an NSERC Discovery Grant

Original: A A DDITIONAL R ELATED W ORK
Revised: A A DDITIONAL R ELATED W ORK

Original: Our investigation fits more broadly into a line of work employing public data for private data analysis.
Revised: Our investigation fits more broadly into a line of work employing public data for private data analysis.

Original: Some works on image classification consider pre-training on a large public dataset and finetuning on a smaller private dataset (Abadi et al., 2016;
Revised: Some works on image classification consider pre-training on a large public dataset and finetuning on a smaller private dataset (Abadi et al., 2016;

Original: Papernot et al., 2019; Tram ` er & Boneh, 2021;
Revised: Papernot et al., 2019; Tram ` er & Boneh, 2021;

Original: Luo et al., 2021).
Revised: Luo et al., 2021).

Original: In particular, Luo et al.
Revised: In particular, Luo et al.

Original: (2021) investigate the role of parameter efficiency in private fine-tuning ResNet models, and propose strategies to choose which parameters to finetune.
Revised: (2021) investigate the role of parameter efficiency in private fine-tuning ResNet models, and propose strategies to choose which parameters to finetune.

Original: One line of work uses unlabeled public data to train a student model (Papernot et al., 2017; 2018; Bassily et al., 2018), including one work simultaneous to our own for natural language generation Tian et al.
Revised: One line of work uses unlabeled public data to train a student model (Papernot et al., 2017; 2018; Bassily et al., 2018), including one work simultaneous to our own for natural language generation Tian et al.

Original: Another recent idea uses a small amount of public data to identify a lower-dimensional subspace of the gradients in which to perform private descent (Zhou et al., 2021;
Revised: Another recent idea uses a small amount of public data to identify a lower-dimensional subspace of the gradients in which to perform private descent (Zhou et al., 2021;

Original: Yu et al., 2021a; Kairouz et al., 2021).
Revised: Yu et al., 2021a; Kairouz et al., 2021).

Original: A simultaneous work of Amid et al.
Revised: A simultaneous work of Amid et al.

Original: (2021) uses public data in the mirror map for a private mirror descent algorithm.
Revised: (2021) uses public data in the mirror map for a private mirror descent algorithm.

Original: Finally, other works (both theoretical and experimental) investigate the role of public data in private query release, synthetic data generation, and prediction (Ji & Elkan, 2013; Beimel et al., 2016; Alon et al., 2019; Nandi & Bassily, 2020;
Revised: Finally, other works (both theoretical and experimental) investigate the role of public data in private query release, synthetic data generation, and prediction (Ji & Elkan, 2013; Beimel et al., 2016; Alon et al., 2019; Nandi & Bassily, 2020;

Original: Bassily et al., 2020a;b; Liu et al., 2021).
Revised: Bassily et al., 2020a;b; Liu et al., 2021).

Original: B E XPERIMENTS WITH D IFFERENT P RIVACY P ARAMETERS
Revised: B E XPERIMENTS WITH D IFFERENT P RIVACY P ARAMETERS

Original: Now we test our framework under different privacy constraints.
Revised: Now we test our framework under different privacy constraints.

Original: Specifically, we run LoRA on the language understanding tasks with various choices of privacy parameters ε and δ .
Revised: Specifically, we run LoRA on the language understanding tasks with various choices of privacy parameters ε and δ .

Original: We consider both
Revised: We consider both

Original: RoBERTa-Base and RoBERTa-Large.
Revised: RoBERTa-Base and RoBERTa-Large.

Original: For the RoBERTa-Large model, we set ε = 1 and 3 with δ being the same as those in Section 4.
Revised: For the RoBERTa-Large model, we set ε = 1 and 3 with δ being the same as those in Section 4.

Original: We use the PRV accountant (Gopi et al., 2021).
Revised: We use the PRV accountant (Gopi et al., 2021).

Original: After getting the noise multipliers, we also reduce the value of δ and report the corresponding value of ε .
Revised: After getting the noise multipliers, we also reduce the value of δ and report the corresponding value of ε .

Original: The hyperparameters are the same as those in Section 4.
Revised: The hyperparameters are the same as those in Section 4.

Original: We run experiments on all four tasks, i.e., MNLI ( n ∼ 392k), QQP ( n ∼ 364k),
Revised: We run experiments on all four tasks, i.e., MNLI ( n ∼ 392k), QQP ( n ∼ 364k),

Original: QNLI ( n ∼ 104k), and SST-2 ( n ∼ 67k).
Revised: QNLI ( n ∼ 104k), and SST-2 ( n ∼ 67k).

Original: We report the results in Table 6.
Revised: We report the results in Table 6.

Original: The performance of our framework is decent even with very tight privacy budgets.
Revised: The performance of our framework is decent even with very tight privacy budgets.

Original: For instance, with ε < 2 and δ = 1 / 1000 n , the accuracy gap between the non-private baseline is only 3.8 for MNLI and 2.1 for
Revised: For instance, with ε < 2 and δ = 1 / 1000 n , the accuracy gap between the non-private baseline is only 3.8 for MNLI and 2.1 for

Original: SST-2.
Revised: SST-2.

Original: For the RoBERTa-Base model, we try various choices of ε .
Revised: For the RoBERTa-Base model, we try various choices of ε .

Original: The values of ε are chosen from [0 . 1 , 0 . 5 , 1 , 3 , 5 , 8 , 12] .
Revised: The values of ε are chosen from [0 . 1 , 0 . 5 , 1 , 3 , 5 , 8 , 12] .

Original: All other settings are the same as those in Section 4.
Revised: All other settings are the same as those in Section 4.

Original: We run experiments on the MNLI and SST-2 datasets.
Revised: We run experiments on the MNLI and SST-2 datasets.

Original: The results are presented in Figure 2.
Revised: The results are presented in Figure 2.

Original: Our framework performs well for a wide range of ε .
Revised: Our framework performs well for a wide range of ε .

Original: We note that our algorithm achieves meaningful accuracy even for very tight privacy parameters ε = 0 .
Revised: We note that our algorithm achieves meaningful accuracy even for very tight privacy parameters ε = 0 .

Original: 5 and 1 .
Revised: 5 and 1 .

Original: Such values of ε are rarely explored when training deep models with differential privacy.
Revised: Such values of ε are rarely explored when training deep models with differential privacy.

Original: A cc u r a c y
Revised: A cc u r a c y

Original: SST0.
Revised: SST0.

Original: A cc u r a c y
Revised: A cc u r a c y

Original: Non-private DP LoRA
Revised: Non-private DP LoRA

Original: A cc u r a c y
Revised: A cc u r a c y

Original: MNLI 0.
Revised: MNLI 0.

Original: A cc u r a c y
Revised: A cc u r a c y

Original: [Figure]
Revised: [Figure]

Original: [Table]
Revised: [Table]

Original: C F INE -T UNING FOR L ANGUAGE U NDERSTANDING T ASKS WITH L ARGE B ATCH S IZE AND F ULL -P RECISION
Revised: C F INE -T UNING FOR L ANGUAGE U NDERSTANDING T ASKS WITH L ARGE B ATCH S IZE AND F ULL -P RECISION

Original: Li et al.
Revised: Li et al.

Original: (2022) show the performance of fine-tuning the full model can be significantly improved with proper configuration.
Revised: (2022) show the performance of fine-tuning the full model can be significantly improved with proper configuration.

Original: In this section, we re-evaluate the tasks in Table 3 and 4 under the configuration in Li et al.
Revised: In this section, we re-evaluate the tasks in Table 3 and 4 under the configuration in Li et al.

Original: and show such a configuration also improves the performance of our methods.
Revised: and show such a configuration also improves the performance of our methods.

Original: The configuration in Li et al.
Revised: The configuration in Li et al.

Original: (2022) has two major differences compared to that in Section 4.1.
Revised: (2022) has two major differences compared to that in Section 4.1.

Original: The first difference is Li et al.
Revised: The first difference is Li et al.

Original: (2022) run experiments with full-precision while the experiments in Section 4.1 use half-precision.
Revised: (2022) run experiments with full-precision while the experiments in Section 4.1 use half-precision.

Original: Using half-precision is a common approach to speed up NLP experiments (Ott et al., 2018).
Revised: Using half-precision is a common approach to speed up NLP experiments (Ott et al., 2018).

Original: However, half-precision may incur underflow issue which impacts the model performance (Micikevicius et al., 2017).
Revised: However, half-precision may incur underflow issue which impacts the model performance (Micikevicius et al., 2017).

Original: The second difference is they use larger batch size for larger datasets.
Revised: The second difference is they use larger batch size for larger datasets.

Original: For example, the batch size for MNLI is roughly six times larger than the batch size for SST-2 in Li et al.
Revised: For example, the batch size for MNLI is roughly six times larger than the batch size for SST-2 in Li et al.

Original: In Section 4.1, we use the same batch size for all datasets.
Revised: In Section 4.1, we use the same batch size for all datasets.

Original: We follow the above setup and re-evaluate DP-LoRA and DP-Adapter.
Revised: We follow the above setup and re-evaluate DP-LoRA and DP-Adapter.

Original: The results are in Tableand 8.
Revised: The results are in Tableand 8.

Original: The results of full fine-tuning with differential privacy are directly adopted from Li et al.
Revised: The results of full fine-tuning with differential privacy are directly adopted from Li et al.

Original: [Table]
Revised: [Table]

Original: Batchsize 0.
Revised: Batchsize 0.

Original: C li
Revised: C li

Original: pp i n g T
Revised: pp i n g T

Original: h
Revised: h

Original: r e s h
Revised: r e s h

Original: o l d 90.
Revised: o l d 90.

Original: Learning Rate 0.
Revised: Learning Rate 0.

Original: Batchsize 0.
Revised: Batchsize 0.

Original: C li
Revised: C li

Original: pp i n g T
Revised: pp i n g T

Original: h
Revised: h

Original: r e s h
Revised: r e s h

Original: o l d 91.
Revised: o l d 91.

Original: Learning Rate 0.
Revised: Learning Rate 0.

Original: Method
Revised: Method

Original: [Table]
Revised: [Table]

Original: MET
Revised: MET

Original: ROUGE
Revised: ROUGE

Original: [Table] [Table]
Revised: [Table] [Table]

Original: [Table]
Revised: [Table]

Original: The configuration in Li et al.
Revised: The configuration in Li et al.

Original: (2022) further improves the strong results in Table 3 and 4.
Revised: (2022) further improves the strong results in Table 3 and 4.

Original: For example, we achieve 89.0% accuracy on the MNLI dataset, which is only 1.2% lower than the accuracy without DP constraint.
Revised: For example, we achieve 89.0% accuracy on the MNLI dataset, which is only 1.2% lower than the accuracy without DP constraint.

Original: Moreover, the benefit of the proposed framework over full finetuning is still clear.
Revised: Moreover, the benefit of the proposed framework over full finetuning is still clear.

Original: The average accuracy of the proposed algorithms is ∼ 3% higher than that of full fine-tuning.
Revised: The average accuracy of the proposed algorithms is ∼ 3% higher than that of full fine-tuning.

Original: D O N THE I NFLUENCE OF H YPERPARAMETERS
Revised: D O N THE I NFLUENCE OF H YPERPARAMETERS

Original: Here we demonstrate that our algorithms perform well for a wide range of hyperparameters.
Revised: Here we demonstrate that our algorithms perform well for a wide range of hyperparameters.

Original: We study two hyperparameters that are directly related to the variance of noise: clipping threshold and batchsize.
Revised: We study two hyperparameters that are directly related to the variance of noise: clipping threshold and batchsize.

Original: The clipping threshold is chosen from [0 . 1 , 1 . 0 , 3 . 0 , 5 . 0 , 10 . 0] and the batchsize is chosen from [200 , 500 , 1000 , 2000 , 4000] .
Revised: The clipping threshold is chosen from [0 . 1 , 1 . 0 , 3 . 0 , 5 . 0 , 10 . 0] and the batchsize is chosen from [200 , 500 , 1000 , 2000 , 4000] .

Original: We note that we keep the number of updates the same as that in
Revised: We note that we keep the number of updates the same as that in

Original: Section 4 when the batchsize is changed.
Revised: Section 4 when the batchsize is changed.

Original: We fine-tune the RoBERTa-Base model with differentially private LoRA ( r = 16 ) on the SST-2 dataset.
Revised: We fine-tune the RoBERTa-Base model with differentially private LoRA ( r = 16 ) on the SST-2 dataset.

Original: The results are presented in Figure 3.
Revised: The results are presented in Figure 3.

Original: DP LoRA performs well for all the hyperparameters considered.
Revised: DP LoRA performs well for all the hyperparameters considered.

Original: The gap between the best accuracy and the worst accuracy is only 2%.
Revised: The gap between the best accuracy and the worst accuracy is only 2%.

Original: E F ULL F INE -T UNING WITH GPT- 
Revised: E F ULL F INE -T UNING WITH GPT-

Original: All results in Table 5 (in the main body), both private and non-private, perform fine-tuning using
Revised: All results in Table 5 (in the main body), both private and non-private, perform fine-tuning using

Original: LoRA.
Revised: LoRA.

Original: In Table 9, we additionally report utility of non-private full fine-tuning.
Revised: In Table 9, we additionally report utility of non-private full fine-tuning.

Original: These numbers are taken from Table 1 of Li & Liang (2021).
Revised: These numbers are taken from Table 1 of Li & Liang (2021).

Original: In general, these numbers are slightly lower than those obtained by performing non-private fine-tuning with LoRA.
Revised: In general, these numbers are slightly lower than those obtained by performing non-private fine-tuning with LoRA.

Original: F A DDITIONAL EXPERIMENTS ON N ATURAL L ANGUAGE G ENERATION
Revised: F A DDITIONAL EXPERIMENTS ON N ATURAL L ANGUAGE G ENERATION

Original: In this section, we perform additional experiments on private fine-tuning for text generation problems using the GPT-2 series of models.
Revised: In this section, we perform additional experiments on private fine-tuning for text generation problems using the GPT-2 series of models.

Original: This includes the private fine-tuning of the GPT-2-XL model with 1.5B parameters.
Revised: This includes the private fine-tuning of the GPT-2-XL model with 1.5B parameters.

Original: There are three main points to note compared to our results in the main body: 1) We show an improved performance on E2E NLG challenge using better hyperparameters; 2)
Revised: There are three main points to note compared to our results in the main body: 1) We show an improved performance on E2E NLG challenge using better hyperparameters; 2)

Original: We
Revised: We

Original: [Table] [Table]
Revised: [Table] [Table]

Original: conduct experiments on E2E dataset with different privacy parameters to show that large language models perform strong even with smaller privacy budgets; 3) Finally, we conduct new experiments on DART dataset.
Revised: conduct experiments on E2E dataset with different privacy parameters to show that large language models perform strong even with smaller privacy budgets; 3) Finally, we conduct new experiments on DART dataset.

Original: F. I MPROVING THE PERFORMANCE ON E2E NLG CHALLENGE
Revised: F. I MPROVING THE PERFORMANCE ON E2E NLG CHALLENGE

Original: We improve the results of Table 5 with the following set of new hyperparameters
Revised: We improve the results of Table 5 with the following set of new hyperparameters

Original: Hyperparameter choice: For LoRA, we choose the bottleneck rank r = 4 in (4) and fine-tune W q and W v matrices of the attention layers as in the original paper.
Revised: Hyperparameter choice: For LoRA, we choose the bottleneck rank r = 4 in (4) and fine-tune W q and W v matrices of the attention layers as in the original paper.

Original: We optimize using AdamW with learning rate 4e-4, weight decay 1e-2 and train our models for 20 epochs.
Revised: We optimize using AdamW with learning rate 4e-4, weight decay 1e-2 and train our models for 20 epochs.

Original: We use batch size 128.
Revised: We use batch size 128.

Original: We take the gradient clipping parameter to be 1.0 and the noise multiplier to be 0.6 for the accountant in Gopi et al.
Revised: We take the gradient clipping parameter to be 1.0 and the noise multiplier to be 0.6 for the accountant in Gopi et al.

Original: (2021), achieving ε = 6 . 0 , δ = 1e-5.
Revised: (2021), achieving ε = 6 . 0 , δ = 1e-5.

Original: Results: The results of our experiments are summarized in the Table 10.
Revised: Results: The results of our experiments are summarized in the Table 10.

Original: F. E XPERIMENTS WITH D IFFERENT P RIVACY P ARAMETERS
Revised: F. E XPERIMENTS WITH D IFFERENT P RIVACY P ARAMETERS

Original: On E2E dataset, we test our framework with smaller privacy budgets ( ε < 5 and δ ≪ 1 /n ) where n is the number of samples in the training data.
Revised: On E2E dataset, we test our framework with smaller privacy budgets ( ε < 5 and δ ≪ 1 /n ) where n is the number of samples in the training data.

Original: Hyperparameter choice: The hyperparameter choices are similar as in Section F.1.
Revised: Hyperparameter choice: The hyperparameter choices are similar as in Section F.1.

Original: The only difference is that we increase the noise multiplier to be 0.71 for the accountant in Gopi et al.
Revised: The only difference is that we increase the noise multiplier to be 0.71 for the accountant in Gopi et al.

Original: (2021), achieving the following ( ε, δ ) pairs: ( ε = 3 . 0 , δ = 1e-5), ( ε = 3 .
Revised: (2021), achieving the following ( ε, δ ) pairs: ( ε = 3 . 0 , δ = 1e-5), ( ε = 3 .

Original: 4 , δ = 1 / 10 n ), ( ε = 3 .
Revised: 4 , δ = 1 / 10 n ), ( ε = 3 .

Original: 9 , δ = 1 / 100 n ) and ( ε = 4 .
Revised: 9 , δ = 1 / 100 n ) and ( ε = 4 .

Original: 5 , δ = 1 / 1000 n ).
Revised: 5 , δ = 1 / 1000 n ).

Original: Results: The results of our experiments are summarized in the Table 11.
Revised: Results: The results of our experiments are summarized in the Table 11.

Original: There are a couple of interesting observations comparing Table 11 with Table 10.
Revised: There are a couple of interesting observations comparing Table 11 with Table 10.

Original: First, we observe that although privacy budget is tight in Table 11, the results are quite similar to Table 10, which shows that our methods also perform very well under stronger privacy guarantees.
Revised: First, we observe that although privacy budget is tight in Table 11, the results are quite similar to Table 10, which shows that our methods also perform very well under stronger privacy guarantees.

Original: A more interesting observation is that under smaller epsilon regimes, the performance for private fine-tuning of
Revised: A more interesting observation is that under smaller epsilon regimes, the performance for private fine-tuning of

Original: GPT-2-XL model improves.
Revised: GPT-2-XL model improves.

Original: Observe that the performance improvement is more prominent going from GPT-2-Small to GPT-2-XL in this setting, which may indicate that larger models can be even more effective in private learning when the privacy budgets are tight.
Revised: Observe that the performance improvement is more prominent going from GPT-2-Small to GPT-2-XL in this setting, which may indicate that larger models can be even more effective in private learning when the privacy budgets are tight.

Original: [Table]
Revised: [Table]

Original: P ERFORMING EXPERIMENTS ON DART DATASET
Revised: P ERFORMING EXPERIMENTS ON DART DATASET

Original: We study the DART dataset as a text generation problem for private fine-tuning of GPT-2 series of models.
Revised: We study the DART dataset as a text generation problem for private fine-tuning of GPT-2 series of models.

Original: DART: DART was introduced as an open-domain data-to-text dataset by Nan et al.
Revised: DART: DART was introduced as an open-domain data-to-text dataset by Nan et al.

Original: The dataset consists of 62K training samples, 6.9K validation samples, and 12K test samples.
Revised: The dataset consists of 62K training samples, 6.9K validation samples, and 12K test samples.

Original: In comparison to E2E, the dataset is larger and the task is more challenging.
Revised: In comparison to E2E, the dataset is larger and the task is more challenging.

Original: Hyperparameter choice: The hyperparameter choices are similar as in the previous setting.
Revised: Hyperparameter choice: The hyperparameter choices are similar as in the previous setting.

Original: The only difference is that we use batch size 256 for the experiments on DART.
Revised: The only difference is that we use batch size 256 for the experiments on DART.

Original: This achieves ε = 6 . 8 , δ = 1e-5 on DART using the accountant in Gopi et al.
Revised: This achieves ε = 6 . 8 , δ = 1e-5 on DART using the accountant in Gopi et al.

Original: Results
Revised: Results