Original: Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guarantees
Revised: Distributed Methods with Compressed Communication for Solving Variational Inequalities, with Theoretical Guarantees

Original: Anonymous Author(s)
Revised: Anonymous Author(s)

Original: AfﬁliationAddress email
Revised: AfﬁliationAddress email

Original: Abstract
Revised: Abstract

Original: Variational inequalities in general and saddle point problems in particular are increasingly relevant in machine learning applications, including adversarial learning, GANs, transport and robust optimization.
Revised: Variational inequalities in general and saddle point problems in particular are increasingly relevant in machine learning applications, including adversarial learning, GANs, transport and robust optimization.

Original: With increasing data and problemsizes necessary to train high performing models across various applications, weneed to rely on parallel and distributed computing.
Revised: With increasing data and problemsizes necessary to train high performing models across various applications, weneed to rely on parallel and distributed computing.

Original: However, in distributed training, communication among the compute nodes is a key bottleneck during training, and this problem is exacerbated for high dimensional and over-parameterizedmodels.
Revised: However, in distributed training, communication among the compute nodes is a key bottleneck during training, and this problem is exacerbated for high dimensional and over-parameterizedmodels.

Original: Due to these considerations, it is important to equip existing methodswith strategies that would allow to reduce the volume of transmitted informationduring training while obtaining a model of comparable quality.
Revised: Due to these considerations, it is important to equip existing methodswith strategies that would allow to reduce the volume of transmitted informationduring training while obtaining a model of comparable quality.

Original: In this paper,we present the ﬁrst theoretically grounded distributed methods for solving variational inequalities and saddle point problems using compressed communication:
Revised: In this paper,we present the ﬁrst theoretically grounded distributed methods for solving variational inequalities and saddle point problems using compressed communication:

Original: MASHA1 and MASHA2 .
Revised: MASHA1 and MASHA2 .

Original: Our theory and methods allow for the use of both unbiased (such as Rand k ; MASHA1 ) and contractive (such as Top k ; MASHA2 ) compressors.
Revised: Our theory and methods allow for the use of both unbiased (such as Rand k ; MASHA1 ) and contractive (such as Top k ; MASHA2 ) compressors.

Original: New algorithms support bidirectional compressions, and also can bemodiﬁed for stochastic setting with batches and for federated learning with partialparticipation of clients.
Revised: New algorithms support bidirectional compressions, and also can bemodiﬁed for stochastic setting with batches and for federated learning with partialparticipation of clients.

Original: We empirically validated our conclusions using two experimental setups: a standard bilinear min-max problem and large-scale distributedadversarial training of transformers.
Revised: We empirically validated our conclusions using two experimental setups: a standard bilinear min-max problem, and large-scale distributedadversarial training of transformers.

Original:  Introduction1.1 The expressive power of variational inequalities
Revised:  Introduction1.1 The expressive power of variational inequalities

Original: Due to their abstract mathematical nature and the associated ﬂexibility they offer in modeling variouspractical problems of interest, variational inequalities (VI) have been an active area of research inapplied mathematics for more than half a century [65, 31, 22].
Revised: Due to their abstract mathematical nature and the associated ﬂexibility they offer in modeling variouspractical problems of interests, variational inequalities (VI) have been an active area of research inapplied mathematics for more than half a century [65, 31, 22].

Original: It is well known that VIs can be usedto formulate and study optimization problems, saddle point problems (SPPs) , games and ﬁxed pointproblems in an elegant unifying mathematical framework [9]
Revised: It is well known that VIs can be usedto formulate and study optimization problems, saddle point problems (SPPs) , games and ﬁxed pointproblems, for example, in an elegant unifying mathematical framework [9].

Original: Recently, a series of works by various authors [15, 26, 58, 13, 49] built a bridge between VIs/SPPsand GANs [28].
Revised: Recently, a series of works by various authors [15, 26, 58, 13, 49] built a bridge between VIs/SPPsand GANs [28].

Original: This allows transferring established insights and well-known techniques from thevast literature on VIs/SPPs, such as averaging and extrapolation, to the study of GANs.
Revised: This allows to successfully transfer established insights and well-known techniquesfrom the vast literature on VIs/SPPs, such as averaging and extrapolation, to the study of GANs.

Original: Besidestheir usefulness in studying GANs and alternative adversarial learning models [57], VIs/SPPs haverecently attracted considerable attention of the machine learning community due to their abilityto model other situations where the minimization of a single loss function does not sufﬁce, such asauction theory [80], supervised learning with non-separable losses [39] or non-separable regularizers[7] and reinforcement learning [69, 66, 38].
Revised: Besides their usefulness in studying GANs and alternative adversarial learning models [57], VIs/SPPshave recently attracted considerable attention of the machine learning community due to their abilityto model other situations where the minimization of a single loss function does not sufﬁce, such asauction theory [80], supervised learning with non-separable loss [39] or non-separable regularizer[7] and reinforcement learning [69, 66, 38].

Original: to not In summary, variational inequalities have recently become a potent tool enabling new advances intheoretical and practical ML.
Revised: In summary, VIs have recently become a potent tool enabling new advances in practical machinelearning situations reaching beyond supervised learning where optimization problems and techniques, which can be seen as special instances of VIs and methods for solving them, reign supreme.

Original: 1.2 Training of supervised models via distributed optimization
Revised: 1.2 Training of supervised models via distributed optimization

Original: On the other hand, for classical and much better understood supervised machine learning or minimization problems, researchers and practitioners face other challenges, which, until recently, havebeen outside of research on VIs.
Revised: On the other hand, for classical and much better understood supervised machine learning /minimization problems, researchers and practitioners face other challenges, which, until recently, have been outside of VI’s research.

Original: Indeed,  training  supervised machine learning models in general,and deep neural networks in particular, is still extremely challenging.
Revised: Indeed, the training of modern supervised machinelearning models in general, and deep neural networks in particular, is still extremely challenging.

Original: Due to their desire to improvethe generalization of deployed models, machine learning engineers need to rely on training datasetsof ever increasing sizes and on elaborate over-parametrized models [5].
Revised: Due to their desire to improve the generalization of deployed models, machine learning engineersneed to rely on training datasets of ever increasing sizes and on elaborate over-parametrized models[5].

Original: Supporting workloads ofsuch unprecedented magnitudes would be impossible without combining the latest advances in hardware acceleration, distributed systems and distributed algorithm design [83].
Revised: Supporting workloads of such unprecedented magnitudes would be impossible without combining the latest advances in hardware acceleration, distributed systems and distributed algorithmdesign [83].

Original: When training such  models in a distributed fashion,the communication cost is often a bottleneckof the training system, and for this reason, a lot of effort was recently targeted at the design ofcommunication-efﬁcient distributed optimization methods [45, 76, 25, 29].
Revised: When training such modern supervised models in a distributed fashion, communication cost is often the bottleneck of the training system, and for this reason, a lot of effort was recently targetedat the design of communication efﬁcient distributed optimization methods [45, 76, 25, 29].

Original: A particularly successful technique for improving the communication efﬁciency of distributed ﬁrst-order optimizationmethods is communication compression.
Revised: A particularly successful technique for improving the communication efﬁciency of distributed ﬁrst orderoptimization methods is communication compression.

Original: The idea behind this technique is rooted in the observationthat in practical implementations, it is often advantageous to communicate messages compressedvia lossy compression techniques instead of communicating the full messages [75, 2].
Revised: The idea behind this technique is rooted inthe observation that in practical implementations it is often advantageous to communicate messagescompressed via (often randomized) lossy compression techniques instead of communicating the fullmessages [75, 2].

Original: If the numberof parallel workers is large enough, the noise introduced by compression is negligible, and trainingwith compressed communication  often leads to a comparable test error while reducing the amountof communicated data, which results in faster training both in theory and practice [59, 29].
Revised: If the number of parallel workers is large enough, the noise introduced by compression is reduced, and training with compressed communication will often lead to comparabletest error while reducing the amount of communicated bits, which results in faster training, both intheory and practice [59, 29].

Original: 1.3 Two classes of compression operators
Revised: 1.3 Two classes of compression operators

Original: This paper focuses on compression methods for distributed VIs and SPPs.
Revised: The paper focuses on compression methods for distributed VIs and SPPs.

Original: First, let us give the maindeﬁnitions used in the work.
Revised: Let us give the maindeﬁnitions.

Original: We say that a (possibly) stochastic mapping Q : R d !
Revised: We say that a (possibly) stochastic mapping Q : R d !

Original: R d is anunbiased compression operator if there exists a constant q  1 such that
Revised: R d is an unbiased compressionoperator if there exists a constant q  1 such that

Original: [Equation]
Revised: [Equation]

Original: Furthermore, we say that a stochastic mapping C : R d !
Revised: Further, we say that a stochastic mapping C : R d !

Original: R d is a contractive compression operatorif there exists a constant   1 such that
Revised: R d is a contractive compression operator ifthere exists a constant   1 such that

Original: [Equation]
Revised: [Equation]

Original: If b is the number of bits needed to represent a single ﬂoating point number (e.g., b = 32 or b = 64 ),then the number of bits needed to represent a generic vector z 2 R d is k z k bits := bd .
Revised: If b is the number of bits needed to represent a single ﬂoat (e.g., b = 32 or b = 64 ), then the numberof bits needed to represent a generic vector z 2 R d is k z k bits := bd .

Original: To describehow much a compression operator reduces its input vector on average, we deﬁne the notion ofexpected density, denoted via   1 := 1 bd E k Q
Revised: To describe how much acompression operator reduces its input vector on average, we deﬁne the notion of expected density,denoted via   1 := 1 bd E k Q

Original: ( z ) k bits , where k Q ( z ) k bits is the number of bits neededto represent the quantized vector Q ( z ) .
Revised: ( z ) k bits , where k Q ( z ) k bits is the number of bits needed to represent thequantized vector Q ( z ) .

Original: Note that   1 .
Revised: Note that   1 .

Original: Forexample, for the Rand k operator [3, 10]we have q =  =
Revised: For the Rand k operator [3, 10] we have q =  =

Original: d/k .
Revised: d/k .

Original: 1.4 Towards communication-efﬁcient distributed methods for VIs and SPPs
Revised: 1.4 Towards communication-efﬁcient distributed methods for VIs and SPPs

Original: Classical VI/SPP algorithms such as the Extra Gradient methodwere originally proposed by[46] and later studied by many authors [62, 41] in different setups, including  a distributed one[77, 52, 61, 73].
Revised: Classical VI/SPP algorithms such as the Extra Gradient method originally proposed by [46] andlater studied by many authors [62, 41], including in a distributed environment [77, 52, 61, 73].

Original: Among them, a number of works stand out trying to solve the communication bottleneck challenge using various approaches such as local steps or data-similarity [88, 34, 16, 11, 12].
Revised: Among them, a number of works stand out trying to solve the communication bottleneck challengeusing various approaches such as local steps, data-similarity etc.[88, 34, 16, 11, 12].

Original: Despite the fact that  compression is one of the most popular communication-efﬁcient approaches fordistributed minimization problems, no work has yet paid attention to the compression techniquesneither for distributed SPPs nor for VIs , with the exception of the work [88], which relies on rounding to the nearest integer multiple of a certain quantity.
Revised: But despitethe fact that the use of compression is one of the most popular communication-efﬁcient approachesfor distributed minimization problems, no work has yet paid attention to the compression techniqueneither for distributed SPPs nor for VIs , with the exception of the work [88], which relies on rounding to the nearest integer multiple of a certain quantity.

Original: This compression mechanism does not offertheoretical beneﬁts and does not even lead to convergence to the solution, since the errors introducedthrough rounding persist and prevent the method from solving the problem.
Revised: This compression mechanism does not offertheoretical beneﬁts and does not even lead to convergence to the solution since the errors introducedthrough rounding persist and prevent the method from solving the problem.

Original:  Summary of Contributions
Revised:  Summary of Contributions

Original: In this paper, we investigate whether it is possible to design communication-efﬁcient algorithms forsolving distributed VIs/SPPs by borrowing generic communication compression techniques (1) and (2) from the optimization literature [75, 2, 59, 29, 72] and embedding them into established, efﬁcientmethods for solving VIs/SPPs [46, 62, 41, 1].
Revised: In this paper, we investigate whether it is possible to design communication-efﬁcient algorithmsfor solving distributed VI/SPP by borrowing generic communication compression techniques (1)and (2) from the optimization literature [75, 2, 59, 29, 72] and embedding them into established,efﬁcient methods for solving VIs/SPPs [46, 62, 41, 1].

Original: Whether or not this is possible is an open problem.
Revised: Whether or not this is possible is an openproblem.

Original: Insummary,we design the ﬁrst algorithms with compression for solving general distributed VI/SPP (see
Revised: In summary,we design the ﬁrst algorithms with compression for solving general distributed VI/SPP (see

Original: Section 3, Equation 3) in the deterministic (see (4) ), stochastic (see (45) ) and federated (see(55) ) regimes, supporting both unbiased ( MASHA1 = Algorithms 1, 5, 7) and contractive( MASHA2 =
Revised: Section 3, Equation 3) in the deterministic (see (4) ), stochastic (see (45) ) and federated (see(55) ) regimes, supporting both unbiased ( MASHA1 = Algorithms 1, 5, 7) and contractive( MASHA2 =

Original: Algorithms 2, 6, 8) compressors.
Revised: Algorithms 2, 6, 8) compressors.

Original: Convergence of all our methods is analyzedin strongly-monotone (strongly convex - strongly concave), monotone (convex - concave)non-monotone/ Minty (non-convex-non-concave) cases.
Revised: Convergence of all our methods are analyzedin strongly-monotone (strongly convex - strongly concave), monotone (convex - concave)non-monotone/minty (non-convex-non-concave) cases.

Original: 2.1 Two types of compressors
Revised: 2.1 Two types of compressors

Original: We develop two approaches for distributed VIs/SPPs depending on whether we use unbiased (1) orcontractive (2) compressors, since each type ofa compressor demands a different algorithmic designand a different analysis.
Revised: We develop two approaches for distributed VIs/SPPs depending on whether we use unbiased (1) orcontractive (2) compressors, since each type of compressor demands a different algorithmic designand a different analysis.

Original: In particular, contractive compressors are notoriously hard to analyze evenfor optimization problems [44, 72].
Revised: In particular, contractive compressors are notoriously hard to analyze evenfor optimization problems [44, 72].

Original: Our method based on unbiased compressors is called MASHA1(Algorithm 1), and our method based on contraction compressors is called MASHA2 (Algorithm 2).
Revised: Our method based on unbiased compressors is called MASHA1(Algorithm 1), and our method based on contraction compressors is called MASHA2 (Algorithm 2).

Original: 2.2 Theoretical complexity results
Revised: 2.2 Theoretical complexity results

Original: We establish a number of theoretical complexity results for our methods, which we summarize in Table 1 (Appendix A).
Revised: We establish a number of theoretical complexity results for our methods, which we summarize in Table 1 (Appendix A).

Original: We consider the strongly monotone (strongly convex - strongly concave), monotone (convex - concave) regimes as well as the more general non-monotone/Minty (non-convex-nonconcave) regime.
Revised: We consider the strongly monotone (strongly convex - strongly concave), monotone (convex - concave) regimes as well as the more general non-monotone/minty (non-convex-nonconcave) regime.

Original: In the strongly monotone case, we obtain linear convergence results ( O (log 1 / ✏ ) )in terms of the distance to solution; in the monotoneone, we obtain fast sublinear convergenceresults ( O (1 / ✏ ) ) in terms of the gap function; and in the non-monotone case, we have sublinear convergence results ( O (1 / ✏ 2 ) ) in terms of the Euclidean norm of the operator.
Revised: In the strongly monotone case we obtain linear convergence results ( O (log 1 / ✏ ) )in terms of the distance to solution, in the monotone we obtain fast sublinear convergence results( O (1 / ✏ ) ) in terms of the gap function, and in the non-monotone case we have sublinear convergenceresults ( O (1 / ✏ 2 ) ) in terms of the Euclidean norm of the operator.

Original: To get an estimate for thenumber of information transmitted, one needs to multiply the estimates from Table 1 by 1 /  .
Revised: To get an estimate for the number of information transmitted, one need to multiply the estimates from Table 1 by 1 /  .

Original: Thenwe obtain that in terms of  transmitted information (and also time for communications), MASHA1is better by a factor p 1 /  + 1 /M
Revised: Then weget that from the point of view of the transmitted information (and also time for communications), MASHA1 is better by a factor p 1 /  + 1 /M

Original: ( M is the number of workers) in comparison with the classical Extra Gradient.
Revised: ( M – number of workers) in comparison with the classical Extra Gradient.

Original: It means that we get an acceleration of min { p  ; p M } times.
Revised: It means that we get an acceleration of min { p  ; p M } times.

Original: For example,
Revised: For example,

Original: ADIANA from [48](the theoretical SOTA method with unbiased compressions for strongly convexminimization) has the same acceleration.
Revised: ADIANA from [48](the theoretical SOTA method with unbiased compressions for strongly convexminimization) has the same accelaration.

Original: The same situation is with MASHA2 .
Revised: The same situation is with MASHA2 .

Original: The method has thesame compression dependent multiplier as ECLK from [71] (the theoretical SOTA with contractivecompression for minimization).
Revised: The method has thesame compression dependent multiplier as ECLK from [71] (the theoretical SOTA with contractivecompression for minimization).

Original: Based on these facts, we hypothesize that MASHA1 and MASHA2have unimprovable estimates (see Appendix C).
Revised: Based on these facts, we hypothesize that MASHA1 and MASHA2have unimprovable estimates (see Appendix C).

Original: 2.3 Stochastic case and variance reduction
Revised: 2.3 Stochastic case and variance reduction

Original: MASHA1 and MASHA2 are designed to handle the deterministic setting.
Revised: MASHA1 and MASHA2 are designed to handle the deterministic setting.

Original: However, in practice, thecomputation of the full operators/gradients isoften expensive, and thus we need to consider stochastic realizations.
Revised: But often, in practice,the computation of the full operators/gradients is expensive, then we need to deal with stochasticrealizations.

Original: In particular, a popular case is when each operator/gradient has a ﬁnite-sum structureon its own, e.g., ﬁnite-sum of batches.
Revised: In particular, a popular case is when each operator/gradient has a ﬁnite-sum structureon its own, e.g. , ﬁnite-sum of batches.

Original: For this issue, we consider two modiﬁcations: VR-MASHA1(Algorithm 5) and VR-MASHA2 (Algorithm 6).
Revised: For this issue, we consider two modiﬁcations: VR-MASHA1(Algorithm 5) and VR-MASHA2 (Algorithm 6).

Original: Both are enhanced with bespoke variance-reductiontechniques for better theoretical and practical performance.
Revised: Both are enhanced with bespoke variance-reductiontechniques for better theoretical and practical performance.

Original: These results can be interesting in thenon-distributed case.
Revised: These results can be interesting in thenon-distributed case.

Original: As far as we know, we are the ﬁrst who consider variance reduction for nonmonotone VIs.
Revised: As far as we know, we are the ﬁrst who consider variance reduction for nonmonotone VIs.

Original: We found only one paper on non-convex-concave saddle point problems [87] underthe PL condition.
Revised: We found only one paper on non-convex-concave saddle point problems [87] underthe PL condition.

Original: See Appendix G for details.
Revised: See Appendix G for details.

Original: 2.4 Federated learning and partial participation
Revised: 2.4 Federated learning and partial participation

Original: Federated learning [45, 42] is an important and popular branch of distributed methods; therefore,a good bonus for the algorithm is that it can be easily adapted for it.
Revised: Federated learning [45, 42] is an important and popular branch of distributed methods. Therefore,a good bonus for the algorithm is that it can be easily adapted for it.

Original: In a federated setup, wherethe computing devices are mobile phones, tablets, personal computers etc, the importance of thecommunication bottleneck is even higher.
Revised: In a federated setup wherethe computing devices are mobile phones, tablets, personal computers etc, the importance of thecommunication bottleneck is even higher.

Original: Also, participating devices can have weak and slow connections, or they can even disconnectfrom the network for a while.
Revised: In such circumstances, devices can have weak and slowconnections, or they can even disconnect for a while.

Original: At such moments, it is notnecessary to interrupt the learning process, and only available devices can be used.
Revised: At such moments, it is not necessary to interrupt the learning process, and only available devices can be used.

Original: To handle thesecases, we introduce two modiﬁcations, PP-MASHA1 (Algorithm 7) and PP-MASHA2 (Algorithm 8),that support the mode of partial participation of devices in the learning process.
Revised: Therefore, we introduce twomodiﬁcations: PP-MASHA1 (Algorithm 7) and PP-MASHA2 (Algorithm 8), that support the mode ofpartial participation of devices in the learning process.

Original: For minimization problems, a combination of quantization and partial participation occurs in [33, 68, 29].
Revised: For minimization problems, a combinationof quantization and partial participation occurs in [33, 68, 29].

Original: The resultsare contained in Appendix H.
Revised: The results are contained in Appendix H.

Original: 2.5 Bidirectional compression
Revised: 2.5 Bidirectional compression

Original: Most methods, especiallythose with contractive compressors, only use compression when transferring information from devices to the server.
Revised: Most methods, especially with contractive compressors, only use compression when transferringinformation from devices to the server.

Original: Meanwhile, quite often in practical situations, thetransfer of information from the server to the device is also expensive [32, 81, 68].
Revised: Meanwhile, quite often in practical situations, the transferof information from the server to the device is also expensive [32, 81, 68].

Original: In such situations it also makes sense to compress the information when sending it from the server to theagents.
Revised: In such situations it alsomakes sense to compress the information when sending it from the server to the agents.

Original: We can highlight several works on bidirectional unbiased [68] and contractive compressors[90, 81, 55, 23] for distributed minimization problems.
Revised: We canhighlight some works on bidirectional unbiased [68] and contractive compressors [90, 81, 55, 23]for distributed minimization problems.

Original: Still, most of these methods have their smallshortcomings in theoretical analysis such asconsidering only the deterministic setting, or assumingthe homogeneity of local functions.
Revised: But most of these methods have their small shortcomingsin theoretical analysis such as deterministic setting only, homogeneity of local functions, etc.

Original: All our methods MASHA1 , MASHA2 and their modiﬁcationssupport bidirectional compression.
Revised: Allour methods MASHA1 , MASHA2 and their modiﬁcations support bidirectional compression.

Original: See Appendix E and F for details.
Revised: See Appendix E and F for details.

Original: 2.6 Experiments
Revised: 2.6 Experiments

Original: Toy experiments on bilinear problems show that methods with compression for minimization problems may not work (diverge) for SPPs.
Revised: Toy experiments on bilinear problems show that methods with compression for minimization problems may not work (diverge) for SPPs.

Original: Also, we verify that MASHA1 and MASHA2 are much betterthan the classical Extra Gradient with added unbiased compression.
Revised: Also we verify that MASHA1 and MASHA2 are much betterthan the classical Extra Gradient with added unbiased compression.

Original: Experiments on adversarialtraining of large-scale transformermodels (ALBERT) show the practical importance of compression in distributed methods for large SPPs.
Revised: Experiments on adversarialtraining of large-scale transformer (ALBERT) show the practical importance of compression in distributed methods for large SPPs.

Original:  Problem Formulation and Assumptions
Revised:  Problem Formulation and Assumptions

Original: 3.1 Problem formulation
Revised: 3.1 Problem formulation

Original: We studya distributed variational inequality (VI) problem of the following form:
Revised: We study distributed variational inequality (VI) problem Find z ⇤

Original: [Equation]
Revised: [Equation]

Revised: R d such that

Original: [Equation]
Revised: [Equation]

Original: where F : R d !
Revised: where F : R d !

Original: R d is an operator with certain favorable properties (e.g., Lipschitzness andmonotonicity).
Revised: R d is an operator with certain favorable properties (e.g., Lipschitzness andmonotonicity).

Original: We assume that the training data describing F is distributed across M workers/nodes/clients
Revised: We assume that the training data describing F is distributed across M workers/nodes/clients

Original: [Equation]
Revised: [Equation]

Original: where F m : R d ! R d for all m 2 { 1 , 2 , . .
Revised: where F m : R d ! R d for all m 2 { 1 , 2 , . .

Original: , M } .
Revised: , M } .

Original: In Appendix B, we give several examples of VIsto show the breadth of this formalism.
Revised: In Appendix B, we give main examples of VIs toshow the breadth of this formalism.

Original: Here we mention only that for the saddle point problem
Revised: Here we mention only that for the saddle point problem

Original: [Equation]
Revised: [Equation]

Original: we take z :=
Revised: we take z :=

Original: [ x, y ] and F m ( z ) :
Revised: [ x, y ] and F m ( z ) :

Original: =
Revised: =

Original: [ r x g ( x, y ) , r y g ( x, y )] .
Revised: [ r x g ( x, y ) , r y g ( x, y )] .

Original: 3.2 Assumptions
Revised: 3.2 Assumptions

Original: Next, we listthe two key assumptions, bothof which are standard in the literature on VIs.
Revised: Next, we list two key assumptions - both are standard in the literature on VIs.

Original: Assumption 3.1 (Lipschitzness)
Revised: Assumption 3.1 (Lipschitzness)

Original: The operator F is L -Lipschitz continuous, i.e. for all z 1 , z 2 2 R dwe have
Revised: The operator F is L -Lipschitz continuous, i.e. for all z 1 , z 2 2 R dwe have

Original: k F ( z 1 )  F ( z 2 )
Revised: k F ( z 1 )  F ( z 2 )

Original: k  L k z 1  z 2 k .
Revised: k  L k z 1  z 2 k .

Original: Each operator F m is L m -Lipschitz continuous, i.e. for all z 1 , z 2 2 R d
Revised: Each operator F m is L m -Lipschitz continuous, i.e. for all z 1 , z 2 2 R d

Original: it holds k F m ( z 1 ) 
Revised: it holds k F m ( z 1 )

Original: F m ( z 2 )
Revised: F m ( z 2 )

Original: k
Revised: k

Original:  L m k z 1  z 2 k .
Revised:  L m k z 1  z 2 k .

Original: Let us deﬁne new constant ˜ L as follows:
Revised: Let us deﬁne new constant ˜ L as follows ˜

Original: ˜ L 2 = 1 M MP
Revised: L 2 = 1 M M

Original: [Equation]
Revised: [Equation]

Original: For saddle point problems, these properties are equivalent to smoothness.
Revised: For saddle point problems, these properties are equivalent to smoothness.

Original: Assumption 3.2 (Monotonicity) We need three cases of monotonicity:(SM
Revised: Assumption 3.2 (Monotonicity) We need three cases of monotonicity(SM)

Original: Strong monotonicity.
Revised: Strong monotonicity.

Original: The operator F is µ -strongly monotone , i.e. for all z 1 , z 2 2 R d we haveh F ( z 1 )  F ( z 2 ) , z 1  z 2 i  µ k z 1  z 2 k 2 .
Revised: The operator F is µ -strongly monotone, i.e. for all z 1 , z 2 2 R d we haveh F ( z 1 )  F ( z 2 ) , z 1  z 2 i  µ k z 1  z 2 k 2 .(M)

Original: (M) Monotonicity.
Revised: Monotonicity.

Original: The operator F is monotone, i.e. for all z 1 , z
Revised: The operator F is monotone, i.e. for all z 1 , z

Original: 2 2 R d we have h F ( z 1 )  170 F ( z 2 ) , z 1  z 2 i  0 .(NM)
Revised: 2 2 R d we have h F ( z 1 )  172 F ( z 2 ) , z 1  z 2 i  0 .(NM)

Original: Non-monotonicity.
Revised: Non-monotonicity.

Original: The operator F is non-monotone (Minty), if and only if there exists z ⇤ 

Revised: The operator F is non-monotone (minty), if and only if there exists z ⇤

Original: R d such that for all z 2 R d we have h F ( z ) , z  z
Revised: R d such that for all z 2 R d we have h F ( z ) , z  z

Original: ⇤ i  0 .
Revised: ⇤ i  0 .

Original: The last assumption is called the Minty or variational stability condition.
Revised: The last assumption is called the minty or variational stability condition.

Original: It does not representgeneral non-monotonicity, but is already associated in the community with non-monotonicity [14,37, 58, 53, 43, 36, 19], particularly with the setup that is somewhat appropriate for GANs [51, 52,21, 8].
Revised: It is not a general nonmonotonicity, but is already associated in the community with non-monotonicity [14, 37, 58, 53, 43,36, 19], particularly with the setup, which is somewhat appropriate for GANS [51, 52, 21, 8].

Original:  MASHA
Revised:  MASHA

Original: In this section, we present new algorithms and their convergence.
Revised: In this Section we present new algorithms and their convergence.

Original: Section 4.1 is devoted to thealgorithm ( MASHA1 ) with unbiased compression.
Revised: Section 4.1 is devoted to the algorithm ( MASHA1 ) with unbiased compression.

Original: Section 4.2 – tothe algorithm ( MASHA2 ) withcontractive compression.
Revised: Section 4.2 – to algorithm ( MASHA2 ) with contractivecompression.

Original: Appendix G describes modiﬁcations for the stochastic case, and Appendix H covers thecase of federated learning. Appendix C is devoted to the hypothesis about the optimalityof MASHA1 and MASHA2 .
Revised: Appendix gives modiﬁcations for the stochastic case – Section G, and for the federated learning – Section H. Appendix C is devoted to the hypothesis about optimality of MASHA1and MASHA2 .

Original: 4.1 MASHA1 : Handling Unbiased Compressors
Revised: 4.1 MASHA1 : Handling Unbiased Compressors

Original: Before presenting our algorithm, let us discuss which approaches can be used to construct it.
Revised: Before presenting our algorithm, let us discuss which approaches can be used to construct it.

Original: Asdiscussed in Sections 1 and 2, compression methods play an important role in distributed minimization problems.
Revised: Asdiscussed in Sections 1 and 2, compression methods play an important role in distributed minimization problems.

Original: All these methods are modiﬁcations of the classical gradient descent (GD): forinstance, the authors of [2] compress stochastic gradients.
Revised: All these methods are modiﬁcations of the classical GD. For instance, the authorsof [2] compress stochastic gradients.

Original: Therefore, it is a natural idea to use GDtype methods for VIs as well.
Revised: Therefore, it is a natural idea to use GD-type methods for VIsas well.

Original: However, it is a well-known fact that GD-type methods can give badconvergence estimates (see Section B.1 from [67]) or do not converge at all (see Section 7.2 and 8.2from [27]) even on the simplest SPPs and VIs.
Revised: But it is a well-known fact that GD-type methods can give bad convergence estimates (see Section B.1 from [67]) or do not converge at all (see Section 7.2 and 8.2 from [27]) even on thesimplest SPPs and VIs.

Original: From a practical point of view, this approach canalso fail (see QSGD and EF in Section 5.1).
Revised: From a practical point of view, this approach can also fail (see QSGD and EF in Section 5.1).

Original: In the non-distributed case, this problem has long beensolved and the Extra Gradient method [46, 62, 41] is used instead of the gradient descent:
Revised: In the non-distributed case, this problem has long been solved and the Extra Gradient method [46, 62, 41] is used instead of GD:

Original: [Equation]
Revised: [Equation]

Original: This method is optimal for both VIs and SPPs and has an estimate of convergence
Revised: This method is optimal for both VIs and SPPs and has an estimate of convergence

Original: ˜ O
Revised: ˜ O

Original: ( L / µ ) in thestrongly monotone case.
Revised: ( L / µ ) in thestrongly monotone case.

Original: Therefore, the second idea for the compressed method is to add compression operators to the method (6), e.g. use Q k ( F ( z k )) and
Revised: Therefore, the second idea for the compressed method is to add compression operators to the method (6), e.g. use Q k ( F ( z k )) and

Original: Q k +1 / 2 ( F ( z k +1 / 2 )) instead of F ( z k )and F ( z k +1 / 2 ) .
Revised: Q k +1 / 2 ( F ( z k +1 / 2 )) instead of F ( z k )and F ( z k +1 / 2 ) .

Original: In Section I we analyse this method, but it gives an estimate ˜ O
Revised: In Section I we analyse this method, but it gives an estimate ˜ O

Original: [Equation]
Revised: [Equation]

Original: which is considerably worse in terms of L / µ than the original Extra Gradient method.
Revised: which is considerably worse in terms of L / µ than the original Extra Gradient method.

Original: The keyproblem is that in the analysis one has to deal with k Q k ( F ( z k +1 / 2 ))
Revised: The keyproblem is that in the analysis one has to deal with k Q k ( F ( z k +1 / 2 ))

Original:  Q k +1 / 2 ( F ( z k ))
Revised:  Q k +1 / 2 ( F ( z k ))

Original: k 2 .
Revised: k 2 .

Original: Withoutcompression operators, such difference is easily evaluated using Assumption 3.1, but when the compression operators are different (in fact the same, but have different randomness) we cannot make agood estimate for this term.
Revised: Without compression operators, such difference is easily evaluated using Assumption 3.1. But when thecompression operators are different (in fact the same, but have different randomness) we cannotmake a good estimate for this term.

Original: One possible solution is to use the same randomness in both steps of themethod (6), namely to substitute Q k ( F ( z k )) and Q k ( F ( z k +1 / 2 )) .
Revised: The idea arises to use the same randomness in both steps of themethod (6), namely to substitute Q k ( F ( z k )) and Q k ( F ( z k +1 / 2 )) .

Original: However, then z k +1 / 2 dependson the randomness Q k , and hence Q k ( F ( z k +1 / 2 )) is biased, which further complicates the analysis.
Revised: But then z k +1 / 2 depends on therandomness Q k , and hence Q k ( F ( z k +1 / 2 )) is biased, which further complicates the analysis.

Original: For exactly the same reasons, the various optimistic/single call modiﬁcations [70, 26, 35, 60] of the Extra Gradient method did not work for us either.
Revised: Forexactly the same reasons, the various optimistic/single call modiﬁcations [70, 26, 35, 60] of the Extra Gradient method did not work for us either.

Original: We have also tested the method (6) with compression techniques in practice (see CEG in Section 5.1), and it turns out to be worse than the method wewill present below.
Revised: We have also test the method (6) with compressionsin practice (see CEG in Section 5.1), and it turns out to be worse than the method we will presentbelow.

Original: In the end, the use of variance reduction and negative momentum techniques [1]is key in creating our algorithm.
Revised: In the end, the use of variance reduction and negative momentum techniques [1] is key increating our algorithm.

Original: These tricks are not in themselves relevant to distributed problems,but, in our case, they help in creating MASHA1 and MASHA2 .
Revised: These tricks are not in themselves relevant to distributed problems, but, inour case, they help in creating MASHA1 and MASHA2 .

Original: At the beginning of each MASHA1 iteration, all devices know the value of F ( w k ) , hence they cancalculate the value of z k +1 / 2 locally without communications. 
Revised: At the beginning of each MASHA1 iteration, all devices know the value of F ( w k ) , hence they cancalculate the value of z k +1 / 2 locally without communications. .

Original: Furthermore, each device sends thecompressed version of the difference F m ( z k +1 / 2 )
Revised: Further, each device sends thecompressed version of the difference F m ( z k +1 / 2 )

Original:  F m ( w k ) to the server.
Revised:  F m ( w k ) to the server.

Original: The compression ofthese transfers is done by their localoperators { Q dev m } .
Revised: The compression onthese transfers is done by their local { Q dev m } operators.

Original: The server aggregates the information fromdevices, averages it, compresses bythe Q serv operator and makes a broadcast to all devices.
Revised: The server aggregates the information fromdevices, averages it, compresses by Q serv operator and makes a broadcast to all devices.

Original: As aresult, an unbiased estimate of F ( z k +1 / 2 )  F ( w k ) appears at each node. Also, the nodes receiveone bit of information b k . This bit is generated randomly on the server and is equal to 1 withprobability 1  ⌧ , where 1  ⌧ is small. Note that b k can be generated locally, it is enough to use the
Revised: As a result,

Original: Algorithm 
Revised: Algorithm 1 MASHA

Original: Parameters: Stepsize  > 0 , parameter ⌧ 2 (0; 1) , number of iterations K .
Revised: Parameters: Stepsize  > 0 , parameter ⌧ 2 (0; 1) , number of iterations K .

Original: Initialization: Choose z
Revised: Initialization: Choose z

Original: =
Revised: =

Original: w 0 2 Z .
Revised: w 0 2 Z .

Original: Devices send F m ( w 0 ) to server and get F ( w 0 ) for k = 0 , 1 , 2 , . .
Revised: Devices send F m ( w 0 ) to server and get F ( w 0 ) for k = 0 , 1 , 2 , . .

Original: , K  1 do for each device m in parallel do 
Revised: , K  1 do for each device m in parallel do z

Original: [Equation]
Revised: [Equation]

Original: Send g km =
Revised: Sends g km =

Original: Q dev m 
Revised: Q dev m ( F m

Original: [Equation]
Revised: [Equation]

Original:  F m ( w
Revised: z [Equation] +

Original: [Equation]
Revised: [Equation]

Original:  to server end for for server do
Revised: )  F m ( w k )) to server end for for server do

Original: Send to devices g k = Q serv h
Revised: Sends to devices g k = Q serv h

Revised: [Equation]

Revised: P

Original: [Equation]
Revised: [Equation]

Original:  i Send to devices one bit b k : 1 with probability 1  ⌧ , 0 with  probability ⌧ end for for each device m in parallel do
Revised: km i Sends to devices one bit b k : 1 with probability 1  ⌧ , 0 with with probability ⌧ end for for each device m in parallel do

Original: [Equation]
Revised: [Equation]

Original: If b k = 1 then w k +1 =
Revised: If b k = 1 then w k +1 =

Original: z k , send F m ( w k +1 ) to server and get F ( w k +1 ) else w k +1 ={++
+}
Revised: z k , sends F m ( w k +1 ) to server and gets F ( w k +1 ) else w k {++
+}

Revised: =

Original: w k end for end for same random generator and set the same seed on all devices.
Revised: w k end for end for an unbiased estimate of F ( z k +1 / 2 )  F ( w k ) appears at each node. Also, the nodes receive one bitof information b k . This bit is generated randomly on the server and is equal to 1 with probability1  ⌧ (where 1  ⌧ is small). Note that b k can be generated locally, it is enough to use the samerandom generator and set the same seed on all devices.

Original: Next, the devices  make a ﬁnal updateon z k +1 locally.
Revised: Next, the devices locally make a ﬁnal updateon z k +1 .

Original: The ﬁnal step is an update of w k +1 : if b k = 1 , then w k +1 =
Revised: The ﬁnal step is an update of w k +1 : if b k = 1 , then w k +1 =

Original: z k+1 or otherwisew k +

Revised: z k or otherwise w k +

Original: =
Revised: =

Original: w k .
Revised: w k .

Original: In the case when w k +

Revised: In the case when w k +

Original: =
Revised: =

Original: z k+1 , we need to exchange the uncompressed values of F m ( w k +1 ) to ensure that at the beginning of the next iteration, the value of F ( w k +1 ) is known toall agents.
Revised: z k , we need to exchange the uncompressed values of F m ( w k +1 ) in orderto ensure that at the beginning of the next iteration the value of F ( w k +1 ) is known to all agents.

Original: We use a possibly different compressors on the devices and  on the server.
Revised: Weuse a possibly difference compressor on each device and also on the server.

Original: To distinguishbetween them, we denote the following notation: Q dev m , q dev m ,  dev m and Q serv , q serv ,  serv .
Revised: To distinguish betweenthem, we denote the following notation: Q dev m , q dev m ,  dev m and Q serv , q serv ,  serv .

Original: Theorem 4.1 Let Assumption 3.1 and one case of Assumption 3.2 be satisﬁed.
Revised: Theorem 4.1 Let Assumption 3.1 and one case of Assumption 3.2 are satisﬁed.

Original: Then for some stepsize  , the following estimates on the number of iterationsof MASHA1 to achieve " -solution hold:
Revised: Then for some step the following estimates on MASHA1 number of iterations to achieve " -solution holds

Original: • inthe strongly monotone case (in terms of E [ k z K  z ⇤ k 2 ] ⇠ " ): O ([ 11  ⌧ + 
Revised: • in strongly monotone case (in terms of E [ k z K  z ⇤ k 2 ] ⇠ " ): O ([ 11  ⌧ + C q µ p 1  ⌧ ] log 1 " );• in monotone case (in terms of E max z 2 C [ h F ( u ) , ( 1 K [Equation] K

Original: [Equation]
Revised: [Equation]

Original: ] log 1 " );
Revised: 1 k [Equation]

Original: • in the monotone case ( E max z  2C [ h F ( u ) , ( 1 K
Revised: 0 z k +1 / 2 )  u i ] ⇠ " ): O ( [Equation] q

Original: [Equation]
Revised: [Equation]

Revised: ⇤ k 2 " p 1  ⌧ );

Original: • inthe non-monotone case (in terms of E [ 1 K
Revised: • in non-monotone case (in terms of E [ 1 K

Original: [Equation]
Revised: [Equation]

Original: where 
Revised: where C 2 q = q serv M

Original: [Equation]
Revised: [Equation]

Original: The full description of the algorithm, as well as the full statement of the theorem withits proof, canbe found in Appendix E.
Revised: A full description of the algorithm, as well as a full statement of the theorem with proof, can befound in Appendix E.

Original: The bounds in Theorem 4.1 are related to ⌧ .
Revised: The bounds in Theorem 4.1 are related to ⌧ .

Original: Let us ﬁnd an optimal way to choose it.
Revised: Let us ﬁnd an optimal way to choose it.

Original: Note that (inaverage) once per 1 / (1  ⌧ ) iterations (when b k = 1 ), we send uncompressed information.
Revised: Note that (inaverage) once per 1 / (1  ⌧ ) iterations (when b k = 1 ), we send uncompressed information.

Original: Based onthis observation, we can ﬁnd the best option for ⌧ .
Revised: Based onthis observation, we can ﬁnd the best option for ⌧ .

Original: Let us analyze the case of compression only onthe devices’ side ( q serv = 1 ).
Revised: Let us analyze the case of compressions only onthe devices’ side ( q serv = 1 ).

Original: For simplicity, we put Q dev m =
Revised: For simplicity, we put Q dev m =

Original: Q with q dev m =
Revised: Q with q dev m =

Original: q and  dev m
Revised: q and  dev m

Original: =
Revised: =

Original:  , also
Revised:  , also

Original: L m =
Revised: L m =

Original: ˜ L = L .
Revised: ˜ L = L .

Original: Since compression is done only on devices, we assume that the server’s broadcast ischeap and we only care about devices.
Revised: Since compression is done only on devices, we assume that the server’s broadcast ischeap and we only care about devices.

Original: Then at each iteration the device sends O ( 1 /  + 1  ⌧ ) bits– each time information compressed by  times and with probability 1  ⌧ we send the full package.
Revised: Then at each iteration the device sends O ( 1 /  + 1  ⌧ ) bits – each time information compressed by  times and with probability 1  ⌧ we send the full package.

Original: From this, we immediately get the optimal choice for ⌧ :
Revised: From where we immediately get the optimal choice for ⌧ :

Original: Corollary 4.2 Let Assumption 3.1 and one case of Assumption 3.2 be satisﬁed.
Revised: Corollary 4.2 Let Assumption 3.1 and one case of Assumption 3.2 are satisﬁed.

Original: Then, for some stepsize  and 1  ⌧ = 1 / , the following estimates on the number of iterationsof MASHA1 to achieve" -solution hold
Revised: Then for some step and 1  ⌧ = 1 /  the following estimates on MASHA1 number of iterations to achieve " -solutionholds

Original: • inthe strongly monotone case: O ([  + [Equation]  [Equation]
Revised: • in strongly monotone case: O ([  + q q  M +  · Lµ ] log 1 " );

Original: [Equation] ] log [Equation] • inthe monotone case: O ( {+q q  M +  · L k z 
+}
Revised: • in monotone case: O ( {+q q  M +  · L k z 
+}

Original: [Equation]  [Equation]
Revised:  z

Original: [Equation] • inthe non-monotone case: O ([ q  M +  ] L 2 k z 

Revised: ⇤ k 2 " ); • in non-monotone case: O ([ q  M +  ] L 2 k z

Original: 
Revised:

Original: z
Revised: z

Original: ⇤ k 2 ") .
Revised: ⇤ k 2 ") .

Original: We can see that MASHA1 can outperform the uncompressed Extra Gradient method.
Revised: We can see that MASHA1 can outperform the uncompressed Extra Gradient method.

Original: Let us comparethem in the strongly monotone case.
Revised: Let us comparethem in the strongly monotone case.

Original: The communication complexity of the Extra Gradient method is ˜ O
Revised: The communication complexity of the Extra Gradient method is˜ O

Original: ( L / µ ) .
Revised: ( L / µ ) .

Original: MASHA1 has communication complexity ˜
Revised: MASHA1 has communication complexity ˜

Original: O
Revised: O

Original: ( p q /  M + 1 /  · L / µ ) .
Revised: ( p q /  M + 1 /  · L / µ ) .

Original: For practical compressors[10],   q .
Revised: For practical compressors[10],   q .

Original: Then, one can note that the communication complexity of MASHA1 differs fromthe complexity of the uncompressed method by an additional factor ( p 1 / M + 1 /  ) .
Revised: Then, one can note that the communication complexity of MASHA1 differs fromthe complexity of the uncompressed method by an additional factor ( p 1 / M + 1 /  ) .

Original: It is easy tosee that even for a small number of devices M and expected density  , this factor is less than 1 ;hence, MASHA1 outperforms the uncompressed method.
Revised: It is easy tosee that even for a small number of devices M and expected density  , this factor is less than 1 ,hence MASHA1 outperforms the uncompressed method.

Original: We believe that this factor ( p 1 / M + 1 /  )is theoretically unimprovable and optimal – see Section C for details.
Revised: We think that this factor ( p 1 / M + 1 /  ) istheoretically unimprovable and optimal – see Section C for details.

Original: One can also consider the case of bidirectional compression ( q serv 6 = 1 ).
Revised: One can also consider the case of bidirectional compression ( q serv 6 = 1 ).

Original: Table 1 (line 3) shows theresult for q serv =
Revised: Table 1 (line 3) shows theresult for q serv =

Original: q dev m = q ,  serv =
Revised: q dev m = q ,  serv =

Original:  dev m =
Revised:  dev m =

Original:  and 1  ⌧ = 1 /  .
Revised:  and 1  ⌧ = 1 /  .

Original: 4.2 MASHA2 : Handling Contractive Compressors
Revised: 4.2 MASHA2 : Handling Contractive Compressors

Original: The use of contractive compressions is a more complex issue.
Revised: The use of contractive compressions is a more complex issue.

Original: In particular, it is known that if onesimply puts a contractive compressor instead of an unbiased one, the method may diverge even forquadratic problems [10].
Revised: In particular, it is known that if onesimply put a contractive cospressor instead of an unbiased one, the method may diverge even forquadratic problems [10].

Original: To ﬁx this, an error compensation technique [78, 44, 79] is used.
Revised: To ﬁx this, an error compensation technique [78, 44, 79] is used.

Original: Thepoint of this approach is to keep untransmitted information and add it to a new package at thenext iteration.
Revised: Thepoint of this approach is to keep untransmitted information and add it to a new package at thenext iteration.

Original: This is the main difference between MASHA2 and MASHA1 .
Revised: This is the main difference between MASHA2 and MASHA1 .

Original: MASHA2 introducesadditional sequences e k , e km for the server’s and devices’ errors.
Revised: MASHA2 introducesadditional sequences e k , e k m for the server’s and devices’ error.

Original: To deﬁne contractive operators ondevices and on the server, we introduce the following notation: C dev m ,  dev ,  dev and C serv m ,  serv ,  serv .
Revised: To deﬁne contractive operators ondevices and on the server, we introduce the following notation: C dev m ,  dev ,  dev and C serv m ,  serv ,  serv .

Original: Algorithm

Original: Parameters: Stepsize  > 0 , parameter ⌧ , number of iterations K .
Revised: Parameters: Stepsize  > 0 , parameter ⌧ , number of iterations K .

Original: Initialization: Choose z
Revised: Initialization: Choose z

Original: =
Revised: =

Original: w 0 2 Z , e 0 m = 0 , e 0 = 0 .
Revised: w 0 2 Z , e 0 m = 0 , e 0 = 0 .

Original: Devices send F m ( w 0 ) to server and get F ( w 0 ) for k = 0 , 1 , 2 , . .
Revised: Devices send F m ( w 0 ) to server and get F ( w 0 ) for k = 0 , 1 , 2 , . .

Original: , K  1 do for each device m in parallel do z k +1 / 2 =
Revised: , K  1 do for each device m in parallel do z k +1 / 2 =

Original: ⌧ z
Revised: ⌧ z

Original: [Equation]
Revised: [Equation]

Original: + (1  ⌧ ) w k   F ( w k )
Revised: + (1  ⌧ ) w k   F ( w k )

Original: Send g km =
Revised: Sends g km =

Original: C dev m (  F m ( z k +1 / 2 )   F m ( w k ) + e km ) to server e k +1 m =
Revised: C dev m (  F m ( z k +1 / 2 )   F m ( w k ) + e km ) to server e k +1 m =

Original: e km +  F m ( z k +1 / 2 )   F m ( w k )  g km end for for server do
Revised: e km +  F m ( z k +1 / 2 )   F m ( w k )  g km end for for server do

Original: Send to devices g k = C serv h 1 M
Revised: Sends to devices g k = C serv h 1 M

Original: P
Revised: P

Original: Mm = [Equation] g km + e k i e k
Revised: Mm =1 g km + e k i e k

Original: +

Revised: +

Original: =
Revised: =

Original: e k + 1 M
Revised: e k + 1 M

Original: P
Revised: P

Original: Mm =1 g km  g k Send to devices one bit b k : 1 with probability ⌧ , 0 with  probability ⌧ end for for each device m in parallel do z
Revised: Mm =1 g km  g k Sends to devices one bit b k : 1 with probability ⌧ , 0 with with probability ⌧ end for for each device m in parallel do z

Original: [Equation]
Revised: k +1 =

Original:  g k
Revised: z k +1 / 2   g k

Original: If b k = 1 then w k +1 =
Revised: If b k = 1 then w k +1 =

Original: z k , send F m ( w k +1 ) to server and get F ( w k +1 ) else w k +1{++
+} =
Revised: z k , sends F m ( w k +1 ) to server and gets F ( w k +1 ) else w k {++
+} =

Original: w k end for end for
Revised: w k end for end for

Original: In the case of MASHA1 , the key theoretical issue was the choice of a basic method (we discussedthis at the beginning of Section 4.1).
Revised: In the case of MASHA1 , the key theoretical issue was the choice of a basic method (we discussedthis at the beginning of Section 4.1).

Original: MASHA2 raises another problem for theoretical analysis,namely how to combine MASHA1 and the error feedback technique.
Revised: MASHA2 raises another problem for theoretical analysis, howto combine MASHA1 and the error feedback technique.

Original: The analysis of methods witherror compensation for the minimization problem min x f ( x ) is entirely tied to the existence of thefunction f
Revised: The analysis of methods with error compensation for the minimization problem min x f ( x ) is entirely tied to the existence of the functionf [79, 71, 72].

Original: In particular, the differences ( f ( · )  f ( x ⇤ )) appear in the entire analysis andare of key importance in the technical lemmas.
Revised: In particular, the differences ( f ( · )  f ( x ⇤ )) appear in the whole analysis and iskey in the technical lemmas.

Original: As a result, ( f ( · )  f ( x ⇤ )) is used as a convergencecriterion even in the strongly convex case.
Revised: As a result ( f ( · )  f ( x ⇤ )) is used as a convergence criterion even inthe strongly convex case.

Original: However, for VIs, there is no function f , only the operator F (the existence of g ( x, y ) in SPP setup does not save the situation).
Revised: But for VIs there is no function f , only the operator F (the existence ofg ( x, y ) in SPP setup does not save the situation).

Original: This problem is solved in theproof of Theorem 4.3 by using an additional sequence k z k +1 / 2  w k k .
Revised: This problem is solved in the proof of Theorem4.3 by using an additional sequence k z k +1 / 2  w k k .

Original: Theorem 4.3 Let Assumption 3.1 and one case of Assumption 3.2 be satisﬁed.
Revised: Theorem 4.3 Let Assumption 3.1 and one case of Assumption 3.2 are satisﬁed.

Original: Then for some step the following estimates on MASHA2 number of iterations to achieve " -solution hold:
Revised: Then for some step the following estimates on MASHA2 number of iterations to achieve " -solution holds

Original: • inthe strongly monotone case (in terms of E [ k ˆ z K  z ⇤ k 2 ] ⇠ " ): O ([ [Equation] ] log [Equation]
Revised: • in strongly monotone case (in terms of E [ k ˆ z K  z ⇤ k 2 ] ⇠ " ): O ([ 11  ⌧ +  dev  serv ˜ L µ p 1  ⌧ ] log 1 " );

Original: [Equation] • inthe monotone case ( E max z 2 C [ h F ( u ) , ( 1 K {+P K  1 k =
+}
Revised: • in monotone case ( E max z 2 C [ h F ( u ) , ( 1 K {+P K  1 k =
+}

Original: [Equation]{+z k +1 / 2 )  u i ] ⇠ " ): O (  dev  serv ˜ L k z 
  z ⇤ k 2 " p ⌧ );+}
Revised: {+z k +1 / 2 )  u i ] ⇠ " ): O (  dev  serv ˜ L k z 
  z ⇤ k 2 " p ⌧ );+}

Original: • inthe non-monotone case (in terms of E [ 1 K
Revised: • in non-monotone case (in terms of E [ 1 K

Original: [Equation]
Revised: [Equation]

Original: The full listing of the algorithm, as well as the full statement of the theorem withits proof, can befound in Appendix F.
Revised: A full listing of the algorithm, as well as a full statement of the theorem with proof, can be found in Appendix F.

Original: Similarly to Section 4.1, we can consider only devices’ or bidirectional compression.
Revised: The same way as in Section 4.1 we can consider only devices’ or bidirectional compression.

Original: In particular,we put the  results for  serv = 1 ,  dev =
Revised: Inparticular, in the line 2 of Table 1 we put results for  serv = 1 ,  dev =

Original:  , L m =
Revised:  , L m =

Original: ˜ L = L and 1  ⌧ =  in the line 2 of Table 1.
Revised: ˜ L = L and1  ⌧ =  .

Original: Inthe line 4 of Table 1, there are results for  serv =
Revised: In the line 4 of Table 1 there are results for  serv =

Original:  dev =
Revised:  dev =

Original:  , L m =
Revised:  , L m =

Original: ˜ L = L and 1  ⌧ =  .
Revised: ˜ L = L and1  ⌧ =  .

Original:  Experiments
Revised:  Experiments5.1 Bilinear Saddle Point Problem

Original: 5.1 Bilinear Saddle Point Problem

Original: We start our experiments with a distributed bilinear problem, i.e. the problem (5) with
Revised: We start our experiments with a distributed bilinear problem, i.e. the problem (5) with

Original: [Equation]
Revised: [Equation]

Original: where A m 2 R d ⇥ d , a m , b m 2 R d .
Revised: where A m 2 R d ⇥ d , a m , b m 2 R d .

Original: This problem is  -strongly convex–strongly-concave and,moreover, all functions g m are k
Revised: This problem is  -strongly convex–strongly-concave and,moreover, all functions g m are k

Original: A m k 2 -smooth.
Revised: A m k 2 -smooth.

Original: Therefore, such a distributed problem is well suitedfor the primary comparison of our methods.
Revised: Therefore, such a distributed problem is well suitedfor the primary comparison of our methods.

Original: We take d = 100 and generate positive deﬁnite matrices
Revised: We take d = 100 and generate positive deﬁnite matrices

Original: A m and vectors a m , b m randomly,  is chosen as max m k
Revised: A m and vectors a m , b m randomly,  is chosen as max m k

Original: A m k 2 / 10 5 .
Revised: A m k 2 / 10 5 .

Original: The purpose of the experiment is to understand whether the MASHA1 and MASHA2 methods aresuperior to those in the literature.
Revised: The purpose of the experiment is to understand whether the MASHA1 and MASHA2 methods aresuperior to those in the literature.

Original: As a comparison, we take QGD [2] with Random 30%, classical
Revised: As a comparison, we take QGD [2] with Random 30%, classical

Original: Error Feedback [78] with Top 30% compression, as well as CEG (Compressed ExtraGradient,
Revised: Error Feedback [78] with Top 30% compression, as well as CEG (Section I) – Compressed Extra

Original: Section I), in each step of which we use Random 30%.
Revised: Gradient, each step of which we use Random 30%.

Original: For MASHA1 (Algorithm 1), we also used Random 30%, for MASHA2 (Algorithm 2) –
Revised: In MASHA1 (Algorithm 1) we also used Random30%, in MASHA2 (Algorithm 2) –

Original: Top 30%.
Revised: Top 30%.

Original: See Figure 1 for the results of this experiment.
Revised: See Figure 1.

Original: The stepsizes of all methods are chosen for best convergence.
Revised: The stepsizes of all methods are chosenfor best convergence.

Original: ) in iterations and in Mbytes for (7).

Original: We see that methods basedon the gradient descent(QSGD and EF) convergeslowly.

Original: This conﬁrms thatone needs to use methodsspeciﬁcally designed forsaddle point problems (forexample, the extragradientmethod), and not classicaloptimization methods.

Original: Themuch slower convergenceof CEG demonstrates theefﬁciency of our approach, in which we compress the differences F m ( z k +1 / 2 )  F ( w k ) .
Revised: The much slower convergence of CEG shows the efﬁciency of our approach in which we compressthe differences F m ( z k +1 / 2 )  F ( w k ) .

Original: Also,it can be seen that MASHA2 outperforms MASHA1 , which shows that in practice a contractivecompressor can perform better than an unbiased one with the same parameters.
Revised: MASHA2 wins MASHA1 . This shows that in practice acontractive compressor can perform better than an unbiased one with the same parameters.

Original: 5.2 Adversarial Training of Transformers
Revised: 5.2 Adversarial Training of Transformers

Original: We now evaluate how compression performs for variational inequalities (and for saddle point problems, as a special case) in a more practically motivated scenario.
Revised: We now evaluate how compression performs for variational inequalities (and for saddle point problems, as a special case) in a more practically motivated scenario.

Original: Indeed, saddle point problems (aspecial case of VIs) have numerous applications in machine learning, including adversarial training .
Revised: Indeed, saddle point problems(special case of variational inequalities) have sample applications in machine learning, includingadversarial training .

Original: We consider one of these tasks and train a transformer-based masked language model [82, 18, 56]using a ﬂeet of 16 low-cost preemptible workers with T4 GPU and low-bandwidth interconnect.
Revised: We consider one of these tasks and train a transformer-based masked language model [82, 18, 56] using a ﬂeet of 16 low-cost preemptible workers with T4 GPU and lowbandwidth interconnect.

Original: For this task, we use the compute-efﬁcient adversarial training task proposed for transformersby [91, 54].
Revised: For this task, we use the compute-efﬁcient adversarial training regimenproposed for transformers by [91, 54].

Original: Formally, the adversarial formulation of the problem is the min-max problem
Revised: Formally, the adversarial formulation of the problem is themin-max problem

Original: [Equation]
Revised: [Equation]

Original: where w are the weights of the model, { ( x n , y n ) }
Revised: where w are the weights of the model, { ( x n , y n ) }

Original: Nn =1 are pairs of  training data, ⇢ is the so-calledadversarial noise which introduces a perturbation in the data, and  is the regularization hyperparameter.
Revised: Nn =1 are pairs of the training data, ⇢ is the socalled adversarial noise which introduces a perturbation in the data, and  are the regularizationparameters.

Original: To make our setup more realistic, we train ALBERT-large with layer sharing [47], whichwas recently shown to be much more communication-efﬁcient during training [74, 20].
Revised: To make our setup more realistic, we train ALBERT-large with layer sharing [47],which was recently shown to be much more communication-efﬁcient during training [74, 20].

Original: We train ourmodel on a combination of BookCorpus and Wikipedia datasets with the same optimizer (LAMB) and parameters as in the original paper [47], use the adversarial training conﬁguration of [91], andfollow system design considerations for preemptible instances [74].
Revised: Wetrain our model on a combination of Bookcorpus and Wikipedia datasets with the same optimizer(LAMB) and parameters as in the original paper [47], use the adversarial training conﬁgurationof [91], and follow system design considerations for preemptible instances [74].

Original: In terms of communication, we consider 4 different setups for gradient compression: the “baseline”strategy with uncompressed gradients, full 8-bit quantization [17, 50], mixed 8-bit quantization, and
Revised: In terms of communication, we consider 4 different setups for gradient compression: the “baseline”strategy with uncompressed gradients, full 8-bit quantization [17, 50], mixed 8-bit quantization, and

Original: PowerSGD compression [84] with rank r =8 .
Revised: Power compression [84] with rank r =8 .

Original: For mixed 8-bit quantization and PowerSGD, weonly apply compression to gradient tensors with more than 2 16 elements, leaving the smaller onesuncompressed.
Revised: For mixed 8-bit quantization and Power we only applycompression to gradient tensors with more than 2 16 elements, sending smaller ones uncompressed.

Original: These small tensors represent layer biases and LayerNorm scales [6] that collectivelyamount to  1% of the total gradient, but can be more difﬁcult to compress than regular weighttensors.
Revised: These small tensors represent layer biases and LayerNorm scales [6] that collectively amount to 1% of the total gradient, but can be more difﬁcult to compress than regular weight tensors.

Original: Finally, since PowerSGD is a biased compression algorithm, we use error feedback [44, 72]with a modiﬁed formulation proposed by [84].
Revised: Finally,since Power is a biased compression algorithm, we use error feedback [44, 72] with a modiﬁedformulation proposed by [84].

Original: For all experimental setups, we report learningcurves in terms of the model training objective, similarly to [24, 74].
Revised: For all experimental setups, we report learning curves in terms of themodel training objective, similarly to [24, 74].

Original: To quantify the differencesin training quality better, we also evaluate the downstream performance for each model on severalpopular tasks from [85] after each model was trained on approximately 80 billion tokens.
Revised: To quantify the differences in training loss better, wealso evaluate the downstream performance for each model on several popular tasks from [85] aftereach model was trained on approximately 80 billion tokens.

Original: Finally,we measure the communication efﬁciency of each proposed strategy by measuring the average walltime per communication round when all 16 workers are active.[Figure]
Revised: Finally, we measure the communicationefﬁciency of each proposed strategy by measuring the average wall time per communication roundwhen all 16 workers are active.

Original: [Table]
Revised: [Table]

Original: The learning curves in
Revised: The learning curves in

Original: Figure 2 (upper) followa predictable pattern,with more extremecompression techniquesdemonstrating slowerper-iteration convergence.
Revised: Figure 2 (upper) followa predictable pattern,with more extremecompression techniquesdemonstrating slowerper-iteration convergence.

Original: One curiousexception to that is full8-bit quantization, whichwas unable to achievea competitive trainingloss.
Revised: One curiousexception to that is full8-bit quantization, whichwas unable to achievecompetitive trainingloss.

Original: The remaining threesetups converge to similarloss values below 2.
Revised: The remainingthree setups converge tosimilar loss values below2.

Original: Boththe baseline and mixed8-bit compression showsimilar results in termsof downstream performance, with PowerSGDshowing a mild degradation.
Revised: Both the baseline andmixed 8-bit compressionshow similar values interms of downstreamperformance, with Powercompression showing mild degradation.

Original: However, in terms of information transfer time, methods usingcompression (especially PowerSGD) are signiﬁcantly superior to the method without compression.
Revised: But in terms of information transfer time, methods usingcompression (especially Power) are signiﬁcantly superior to the method without compression.

Original: This makes it possible to use such techniques to reduce the training time without sacriﬁcing quality.
Revised: Thismakes it possible to use such techniques to increase the training time without sacriﬁcing quality.

Original:  Conclusion
Revised:  Conclusion

Original: In this paper, we present algorithms with unbiased and contractive compression for solving distributed VIs and SPPs.
Revised: In this paper we present algorithms with unbiased and contractive compressions for solving distributed VIs and SPPs.

Original: Our algorithms are presented in deterministic, stochastic and federated versions.
Revised: Our algorithms are presented in deterministic, stochastic and federated versions.

Original: All basic algorithms and their modiﬁcations support bidirectional compression.
Revised: All basic algorithms and their modiﬁcations support bidirectional compression.

Original: Experimentsconﬁrm the efﬁciency of both our algorithms and the use of compression for solving large-scale VIsin general.
Revised: Experimentsconﬁrm the efﬁciency of both our algorithms and the use of compression for solving large-scale VIsin general.

Original: In future work, we believe that it is important to address the necessity of sending uncompressed information on some iterations.
Revised: In future works it is important to address the issue of the necessity to forward uncompressed information in some iterations.

Original: Although full packages are rarely transmitted, this is a slight limitationof our approach.
Revised: Although full packages are rarely transmitted, this is a slight limitationof our approach.

Original: Lower bounds for compression methods are also an interesting area of research.
Revised: Lower bounds for compression methods are also an interesting area of research.

Original: Atthe moment, there are no such resultsneither for VIs and SPPs nor for minimization.
Revised: At the moment there are neither such results for VIs and SPPs, nor for minimizations.

Original: In Appendix
Revised: In Appendix

Original: C, we only hypothesize the optimality of our methods and back it up with analogies; provable lowerestimates could complete the story with compressed methods.
Revised: C we only hypothesize the optimality of our methods and back it up with analogies, provable lowerestimates could complete the story with compressed methods.