SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient
Open accessAbstract
Many deep learning applications benefit from using large models with billions of parameters. Training these models is notoriously expensive due to the need for specialized HPC clusters. In this work, we consider alternative setups for training large models: using cheap "preemptible" instances or pooling existing resources from multiple regions. We analyze the performance of existing model-parallel algorithms in these conditions and find configurations where training larger models becomes less communication-intensive. Based on these findings, we propose SWARM parallelism, a model-parallel training algorithm designed for poorly connected, heterogeneous and unreliable devices. SWARM creates temporary randomized pipelines between nodes that are rebalanced in case of failure. We empirically validate our findings and compare SWARM parallelism with existing large-scale training approaches. Finally, we combine our insights with compression strategies to train a large Transformer language model with 1B shared parameters (approximately 13B before sharing) on preemptible T4 GPUs with less than 200Mb/s network.
AI snapshot
Written from the paper’s full text.
TL;DR
The paper analyzes how communication and computation scale with model size in pipeline parallelism, deriving a "Square-Cube Law" showing that larger models can be trained with less network overhead per unit of compute. It then proposes SWARM parallelism, a decentralized pipeline-parallel algorithm with stochastic wiring and adaptive rebalancing, and uses it to train a 1B-parameter (≈13B-equivalent) Transformer on preemptible T4 GPUs with under 200 Mb/s network.
Key findings
- Larger models achieve better GPU utilization under the same network conditions: at 500 Mb/s, relative GPU utilization was 18.0% ("base"), 32.1% ("xxlarge"), 82.1% ("GPT-3"), and 89.5% ("Ours" with 8-bit compression); at 100 ms latency these were 2.78%, 14.9%, 60.2%, and 71.5% respectively (Table 1).
- SWARM is competitive with HPC baselines even in idealized homogeneous conditions: for "GPT-3" (4 layers) throughput was 168.3 min/batch (no latency) vs 164.5 for GPipe and 272.7 for ZeRO-Offload; for the full "Ours" model, 432.2 vs 420.0 and 372.0 (Table 2).
- Adaptive rebalancing substantially improves throughput: with T = 300 s, SWARM reached 95.8% of optimal throughput overall (88.9% in the last hour) vs 82.7% (45.4% in the last hour) with no rebalancing; T = 60 s reached 97.6% (Table 5).
- With layer sharing and 8-bit compression, T4 GPUs reached 17.6 samples/s (best-case 19.2) with optimal bandwidth of 317.8 Mb/s upload / 397.9 Mb/s download; A100s reached 16.9 (best-case 25.5) (Table 3). Training convergence on the Pile over ~4 weeks with 400 preemptible T4s matched a DDP baseline on 128 A100s (Figure 4).
Methods
- Design: SWARM parallelism partitions the model into evenly sized pipeline stages ("swarms"); peers within a stage serve the same layers, and stochastic wiring routes each microbatch to peers in proportion to measured throughput using Interleaved Weighted Round-Robin scheduling, with peers discovered via a Distributed Hash Table. Adaptive rebalancing moves peers between stages every T seconds based on queue-size load measurements, and failed peers are banned and rerouted.
- Data/experiments: controlled V100 experiments measuring GPU utilization and network usage across four Transformer configurations (base, xxlarge, GPT-3-scale, and a modified "Ours" with 8-bit quantized activations); a comparison against GPipe and ZeRO-Offload over 16 workers with 4 Transformer layers; large-scale training of a 1.01B-parameter (≈13B-equivalent via layer sharing) Transformer on the Pile with 400 preemptible T4 instances and 7×8 A100 instances; and a simulation of rebalancing using 32 hours of real T4 node-count statistics.
- Analysis: throughput (time to process 6250 sequences of 512 tokens) and All-Reduce time; relative throughput vs a theoretically optimal rebalancing strategy; scaling of throughput with node count (8–128 nodes) and with number of stages (4–32).
Limitations
- Stated: SWARM is by default worse than traditional parallelism in supercomputers due to extra complexity, and is only useful there for heterogeneous devices; for models under 1B parameters, sharded data-parallel approaches can be more optimal; SWARM suffers from the pipeline "bubble" problem with long pipelines; asynchronous updates were not investigated due to potential convergence issues; rebalancing moves only one peer at a time; and throughput drops as stage count increases because variance in preemption is too large relative to peer counts.
- Evident from the text: the rebalancing evaluation (Section 4.4) is a simulation using recorded node-count statistics rather than a live deployment; the large-scale convergence comparison is run for approximately 4 weeks and compared against a single DDP baseline; and the paper notes that its GPU-utilization measurements are not directly comparable to peak-FLOP/s utilization reported by other systems (Appendix A).
Model-written summary. Check it against the paper before relying on it.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
0 of 12 cited sources checked · 0 supported · 0 flagged
- Findingown data
In the earlier Figure 1 experiment, the same model spent most of the time waiting for communication between pipeline stages.
“where the same model spent most of the time waiting for the communication between pipeline stages.”
Quote in the paperNot a substantial claim - Findingown data§ I. Compression-Aware Architectures
Blockwise (de)quantization takes less than 1% of total computation time.
“However, this is not required, as blockwise (de)quantization takes less than 1% of total computation time: see Appendix J for details.”
Quote in the paperNot a substantial claim - Findingown data§ C. Stochastic Wiring Details
The same bandwidth allows for >80% GPU utilization when dealing with GPT-3-sized layers.
“However, that same bandwidth allows for > 80% GPU utilization when dealing with GPT-3-sized layers.”
Quote in the paperSubstantial claim - Findingown data§ C. Stochastic Wiring Details
If 3 GPT-3 layers are colocated per pipeline stage, GPU utilization can be further improved to >90%.
“If we colocate 3”
Quote in the paperNot a substantial claim - Findingown data§ 0 10B 20B 30B 40B 50B
The training dynamics of SWARM parallelism and regular data-parallel training with offloading are similar over approximately 4 weeks of training.
“it can be seen that the training dynamics of two approaches are indeed similar”
Quote in the paperSubstantial claim
- Findingown data§ C. Stochastic Wiring Details
Due to modest communication bandwidth, smaller models spend most of the time waiting for the network.
“Due to a modest communication bandwidth, smaller models spend most of the time waiting for the network.”
Quote in the paperSubstantial claim - Findingreasoning§ 1. Introduction
It is possible to train a billion-scale Transformer language model on preemptible servers with low-power GPUs and network bandwidth of less than 200Mb/s while achieving high training throughput.
“we show that it is possible to train a billion-scale Transformer language model on preemptible servers with low-power GPUs and the network bandwidth of less than 200Mb/s while achieving high training throughput.”
Quote in the paperSubstantial claim - Findingown data§ 1 layer
Large models maintain most of their training efficiency at the 100ms latency.
“In addition, large models maintain most of their training efficiency at the 100ms latency (Table 1), which is roughly equivalent to training on different continents (Verizon, 2021).”
Quote in the paperSubstantial claim - Findingcitation§ I. Compression-Aware Architectures
It is critical to use layer normalization to ensure training without divergence in bottleneck layers.
“We find it critical to use layer normalization (Ba et al., 2016) to ensure training without divergence.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites Ba et al., 2016
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- Findingown data§ 4 stages
The effect of decreasing T from 300 to 60 seconds is not as drastic compared to the increase in DHT data transfer volume.
“However, the effect is not as drastic compared to the increase in DHT data transfer volume.”
Quote in the paperNot a substantial claim - Findingown data
The throughput of SWARM is not very sensitive to the choice of the rebalancing period T and the moving-average parameter alpha.
“However, as we show in Section 4.4, the throughput of SWARM is not very sensitive to the choice of these hyperparameters.”
Quote in the paperNot a substantial claim - Findingown data§ 4 stages
Even with the rebalancing period T = 300, SWARM's approach significantly improves the overall throughput of the pipeline.
“even with the rebalancing period T = 300, our approach significantly improves the overall throughput of the pipeline.”
Quote in the paperSubstantial claim - Findingown data§ Supplementary Material
By default, SWARM is worse than traditional parallelism due to its extra complexity.
“By default, SWARM is worse than traditional parallelism due to its extra complexity (see experiments in Section 4.2).”
Quote in the paperNot a substantial claim - Findingown data§ 1 layer
Some of the larger models trained with pipeline parallelism can already train at peak efficiency with only hundreds of Mb/s bandwidth.
“Section 4.1 demonstrates that some of the larger models trained with pipeline parallelism can already train at peak efficiency with only hundreds of Mb/s bandwidth.”
Quote in the paperSubstantial claim - Interpretationreasoning§ C. Stochastic Wiring Details
The approximately linear scaling can be explained by the high efficiency of stochastic wiring and auxiliary training components such as the DHT and the All-Reduce protocol used for gradient averaging.
“which can be explained by the high efficiency of both the stochastic wiring strategy and the auxiliary training components such as the DHT and the All-Reduce protocol used for gradient averaging.”
Quote in the paperSubstantial claim - Interpretationnone given
Pure pipeline parallelism requires many stages to effectively hide the latency.
“However, pure pipeline parallelism requires many stages to effectively hide the latency.”
Quote in the paperSubstantial claim - Interpretationreasoning§ I. Compression-Aware Architectures
8-bit quantization appears to be a reliable default choice to reduce communication overhead for pipeline parallelism.
“As such, the 8-bit quantization appears to be a reliable default choice to reduce the communication overhead for pipeline parallelism.”
Quote in the paperSubstantial claim - Premisenone given
The communication phase requires at most O(n^2) time to transfer a batch of n×n activations or gradients.
“In turn, the communication phase requires at most O(n2) time to transfer a batch of n×n activations or gradients.”
Quote in the paperSubstantial claim - Interpretationreasoning§ Supplementary Material
Peer failures do not affect forward and backward passes as long as there is at least one peer at the required stage.
“As we mention in Section 3.2, peer failures do not affect forward and backward passes as long as there is at least one peer at the required stage: because of rewiring, it is possible to resend activations or gradients to another worker that has identical model weights by construction.”
Quote in the paperSubstantial claim - Interpretationreasoning§ Supplementary Material
Because of rewiring, it is possible to resend activations or gradients to another worker that has identical model weights by construction.
“because of rewiring, it is possible to resend activations or gradients to another worker that has identical model weights by construction.”
Quote in the paperNot a substantial claim - Interpretationreasoning§ Supplementary Material
Sharded data parallelism can be particularly inefficient for setups where peers have different network bandwidths.
“Thus, sharded data parallelism can be particularly inefficient for setups where peers have different network bandwidths.”
Quote in the paperSubstantial claim - Premisecitation§ 1 layer
100ms latency is roughly equivalent to training on different continents.
“which is roughly equivalent to training on different continents (Verizon, 2021).”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Verizon, 2021
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- Interpretationreasoning§ Supplementary Material
If a peer leaves from the pipeline stage, other workers can compute gradients and replace those accumulated by the disconnected peer, so that the number of examples for an SGD step stays the same.
“Hence, if the peer leaves from the pipeline stage, other workers can compute gradients and replace those accumulated by the disconnected peer, so that the number of examples for an SGD step stays the same.”
Quote in the paperSubstantial claim- The authors state this as an assumption: The i.i.d. assumption in the large-scale training setup holds, so the distribution of gradients is expected to be the same across examples within a batch.
- Interpretationreasoning§ Supplementary Material
When a peer joins the network, it can download the checkpoint corresponding to the current training state from other peers.
“Thus, when a peer joins the network, it can download the checkpoint corresponding to the current training state from other peers.”
Quote in the paperNot a substantial claim - Interpretationreasoning§ C. Stochastic Wiring Details
In case of a latency spike, SWARM workers keep processing previously queued microbatches, maintaining high device utilization.
“In case of a latency spike, workers keep processing previously queued microbatches, maintaining high device utilization.”
Quote in the paperSubstantial claim - Interpretationreasoning
If one pipeline stage requires more compute than others, SWARM will allocate more peers to that stage.
“As a side effect, if one pipeline stage requires more compute than others, SWARM will allocate more peers to that stage.”
Quote in the paperNot a substantial claim - Interpretationcitation§ C. Stochastic Wiring Details
SWARM can still train with competitive training throughput due to large models being trained with increased batch sizes.
“However, as we show in Section 4.3, SWARM can still train with competitive training throughput due to the fact that large models are trained with increased batch sizes (Brown et al., 2020).”
Quote in the paperSubstantial claim0 of 1 sources checked- cites Brown et al., 2020
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- Interpretationreasoning§ Supplementary Material
With slow interconnects, data-parallel synchronizations are not practical.
“Thus, with slow interconnects, such synchronizations are not practical.”
Quote in the paperSubstantial claim - Interpretationreasoning§ C. Stochastic Wiring Details
For practical numbers of peers (<10,000) and stages (<100), Algorithm 2 incurs negligible overhead on performance.
“For practical numbers of both peers (e.g., < 10,000) and stages (fewer than 100), this incurs a negligible overhead on performance, as all communication and computation is done in parallel with the actual forward and backward passes.”
Quote in the paperSubstantial claim - Interpretationreasoning§ I. Compression-Aware Architectures
The bottleneck layer requires m/c times less communication compared to the original model.
“The parameter matrix wc resides in one stage and its outputs are transferred to the next stage that holds the parameters wd, which requires m/c times less communication compared to the original model.”
Quote in the paperSubstantial claim - Interpretationreasoning
Because of layer sharing, the 1.01B-parameter model is equivalent to a 13B model from Brown et al. (2020) in terms of compute costs.
“Because of layer sharing, this model is equivalent to a 13B model from Brown et al. (2020) in terms of compute costs.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites Brown et al., 2020
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- Premisereasoning§ Supplementary Material
For 1B parameters in 16-bit precision, a total of 2 GB need to be synchronized for both the forward and backward pass.
“Thus, for 1B parameters in 16-bit precision, a total of 2 GB need to be synchronized for both the forward and backward pass.”
Quote in the paperSubstantial claim - Interpretationreasoning§ 1. Introduction
Increasing model size leads computation costs to grow faster than the network footprint, making household-grade connection speeds more practical than one might think.
“increasing the size leads computation costs to grow faster than the network footprint, thus making household-grade connection speeds more practical than one might think.”
Quote in the paperSubstantial claim - Interpretationown data§ 5. Conclusion
Training large models with SWARM parallelism and compression-aware architectures enables high utilization of cheap preemptible instances with slow interconnect.
“We also show that training large models with SWARM parallelism and compression-aware architectures enables high utilization of cheap preemptible instances with slow interconnect.”
Quote in the paperSubstantial claim - Interpretationown data
SWARM is competitive with HPC baselines even in an idealized homogeneous environment.
“Overall, SWARM is competitive to HPC baselines even in an idealized homogeneous environment.”
Quote in the paperSubstantial claim - Interpretationreasoning§ Supplementary Material
20% GPU idle time for SWARM means that it can reach approximately 0.8x the training throughput compared to training with an infinitely fast network.
“Since any realistic deep learning workload cannot achieve 100% peak FLOP/s, 20% GPU idle time for SWARM means that it can reach ≈ 0.8x the training throughput compared to training with an infinitely fast network.”
Quote in the paperNot a substantial claim- Rests on an assumption: Any realistic deep learning workload cannot achieve 100% peak FLOP/s.
- Interpretationown data§ 5. Conclusion
High-throughput training of billion-scale neural networks on unreliable peers with low network bandwidth can be possible with very large models and pipeline parallelism.
“We find that training in this setup can be possible with very large models and pipeline parallelism.”
Quote in the paperSubstantial claim - Interpretationreasoning§ C. Stochastic Wiring Details
Stochastic wiring allows SWARM to mitigate network latency.
“An important observation is that stochastic wiring allows SWARM to mitigate network latency.”
Quote in the paperSubstantial claim - Interpretationreasoning§ C. Stochastic Wiring Details
The overall network load of the adaptive rebalancing migration procedure is relatively small.
“Also, notice that only one migrating peer needs to stop processing requests and download the weights and optimizer statistics of the pipeline stage it starts serving: this means that the overall network load of this procedure is relatively small, as all DHT requests handle scalar data and do not exceed the number of active peers for each worker.”
Quote in the paperSubstantial claim - Interpretationown data§ Supplementary Material
By using no quantization (regular 16-bit activations), the network overhead increases approximately by a factor of two.
“By using no quantization, which means using regular 16-bit activations, the network overhead increases approximately by a factor of two.”
Quote in the paperNot a substantial claim
- May combine two claims. 19
- The quote only partly states this claim. 3
- Classified as probably not a substantial claim. 27
May combine two claims.
Findingown data§ I. Compression-Aware Architectures8-bit compression does not degrade the time to 22 perplexity and maintains close to the final perplexity of the baseline.
“We can see that 8-bit compression does not degrade the time to 22 perplexity and maintains close to the final perplexity of the baseline.”
Quote in the paperSubstantial claimMay combine two claims.
Findingown data§ I. Compression-Aware ArchitecturesAdding a bottleneck only adds two linear layers for the forward pass and decreases the size of MLP activations; its computational overhead is negligible (less than 1% for typical sizes).
“Note that adding a bottleneck only adds two linear layers for the forward pass and decreases the size of MLP activations; thus, its computational overhead is negligible (less than 1% for typical sizes, see Appendix J).”
Quote in the paperSubstantial claimThe quote only partly states this claim.
Findingreasoning§ C. Stochastic Wiring DetailsTraining performance exhibits an approximately linear scaling pattern with the number of nodes from 8 to 128.
“It can be seen that the training performance exhibits an approximately linear scaling pattern”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Findingnone given§ Supplementary MaterialSome configurations in Section 4.1 measure less than 20% GPU idle time.
“Some configurations in Section 4.1 measure less than 20% GPU idle time”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Findingown data§ 4 stagesFor approximately 4.5% of the experiment, the performance of the unbalanced pipeline exceeded the throughput of the balanced one due to random choice of disconnecting peers.
“However, this held true only for ≈ 4.5% of the experiment and was quickly mitigated by adaptive rebalancing.”
Quote in the paperNot a substantial claimMay combine two claims.
Findingown data§ 4. ExperimentsLarger models achieve better GPU utilization rate in the same network conditions, since their communication load grows slower than computation.
“As depicted in Figure 1 (right) and Figure 3, larger models achieve better GPU utilization rate in the same network conditions, since their communication load grows slower than computation.”
Quote in the paperSubstantial claimMay combine two claims.
Findingown data§ 4 stagesRebalancing partially mitigates the performance drop with more stages: the rebalanced pipeline regains its performance over time and achieves a higher overall throughput than the baseline.
“Still, rebalancing allows to partially mitigate the issue: while we observe a more consistent downward trend for the baseline strategy, the rebalanced pipeline regains its performance over time and achieves a higher overall throughput.”
Quote in the paperSubstantial claimMay combine two claims.
Findingown data§ I. Compression-Aware ArchitecturesWhile the bottleneck architecture is competitive with maxout for a compression factor of 2x with two stages, maxout has better perplexities if more stages or a higher compression ratio is used.
“We can see that while the bottleneck architecture is competitive with maxout for a compression factor of 2x with two stages, maxout has better perplexities if more stages or a higher compression ratio is used.”
Quote in the paperSubstantial claimMay combine two claims. The quote only partly states this claim.
Findingown data§ 4 stagesBoth the rebalancing and the no-rebalancing strategies drop in performance as the number of pipeline stages increases from 4 to 32.
“both strategies drop in performance with the increase in the stage count”
Quote in the paperNot a substantial claimMay combine two claims.
Findingown data§ I. Compression-Aware ArchitecturesThe compression-aware bottleneck and maxout architectures perform equal to each other, but degrade final perplexity slightly and increase time to a perplexity of 22 by 26-28%.
“The compression-aware bottleneck and maxout architectures perform equal to each other, but degrade final perplexity slightly and increase time to a perplexity of 22 by 26–28%.”
Quote in the paperSubstantial claimMay combine two claims.
Findingown data§ 4 stagesDecreasing T from 300 to 60 seconds improves both the overall throughput and the speed of convergence to optimal pipeline performance.
“As expected, decreasing T from 300 to 60 seconds improves both the overall throughput and the speed of convergence to optimal pipeline performance.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S4 (based).
Interpretationnone givenIf any single pipeline stage loses too many peers, the remaining ones will face an increased processing load, which will inevitably form a bottleneck.
“If any single pipeline stage loses too many peers, the remaining ones will face an increased processing load, which will inevitably form a bottleneck.”
Quote in the paperNot a substantial claim- Rests on an assumption: Workers can join and leave training at any time.
Probably not a substantial claim: may not pass S4 (based).
Premisenone given§ Supplementary MaterialAll-to-all communication latency is determined by the slowest peer.
“This is exacerbated further, because all-to-all communication latency is determined by the slowest peer.”
Quote in the paperNot a substantial claimClassified as probably not a substantial claim.
Interpretationnone given§ AbstractSWARM parallelism is a model-parallel training algorithm designed for poorly connected, heterogeneous and unreliable devices.
“we propose SWARM parallelism1, a model-parallel training algorithm designed for poorly connected, heterogeneous and unreliable devices.”
Quote in the paperSubstantial claim- Rests on an assumption: The training devices are poorly connected, heterogeneous, and unreliable.
Probably not a substantial claim: may not pass S3 (load-bearing).
Interpretationreasoning§ Supplementary MaterialUsing long pipelines can result in a reduced training throughput.
“While its design allows for any number of stages, using long pipelines can result in a reduced training throughput.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Study design and conductown dataThe adaptive rebalancing evaluation ran 10 simulations with different random seeds and averaged the resulting trajectories.
“We run 10 simulations with different random seeds and average the resulting trajectories.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S4 (based).
Interpretationnone given§ Supplementary MaterialThe data loader state can be recomputed from the last known SGD step.
“Similarly, the data loader state can be recomputed from the last known SGD step.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Study design and conductnone given§ C. Stochastic Wiring DetailsAll results in Section 4.1 are averaged over 1000 consecutive batches, and the standard deviations are below 0.1%.
“All results are averaged over 1000 consecutive batches; the standard deviations are below 0.1%.”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationown data§ Supplementary MaterialTraining with SWARM can accelerate training while also allowing to train larger models.
“Training with SWARM can accelerate training while also allowing to train larger models; see Appendix E for a detailed comparison.”
Quote in the paperSubstantial claimMay combine two claims.
InterpretationreasoningZeRO-Offload takes longer to aggregate gradients because each peer must aggregate the entire model, whereas in SWARM and GPipe peers aggregate a single pipeline stage.
“likely because each peer must aggregate the entire model, whereas in SWARM and GPipe, peers aggregate a single pipeline stage.”
Quote in the paperSubstantial claimClassified as probably not a substantial claim.
Interpretationreasoning§ C. Stochastic Wiring DetailsAll surviving peers use non-stale aggregated gradients over the specified batch size when performing the optimizer step.
“This procedure ensures that all surviving peers use non-stale aggregated gradients over the specified batch size when performing the optimizer step.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
PremisecitationTraining large models over heterogeneous devices can be optimized with global scheduling.
“Moreover, training large models over heterogeneous devices can be optimized with global scheduling (Yuan et al., 2022).”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Yuan et al., 2022
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing).
Interpretationreasoning§ C. Stochastic Wiring DetailsThe total complexity of Algorithm 2 is O(MS), where M is the highest number of peers over all stages and S is the number of stages.
“These operations are nested in the loop over all stages, which means that the total complexity of the algorithm is O(MS).”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing), S4 (based).
Premisenone given§ Supplementary MaterialBottleneck and maxout compression are general compression techniques that can be applied to any layer in any architecture.
“Bottleneck and maxout compression are general compression techniques that can be applied to any layer in any architecture.”
Quote in the paperNot a substantial claimClassified as probably not a substantial claim.
Interpretationnone givenIf any node fails or leaves training prematurely, it will stall the entire training procedure.
“On top of that, if any node fails or leaves training prematurely, it will stall the entire training procedure.”
Quote in the paperSubstantial claim- Rests on an assumption: Workers can join and leave training at any time.
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
InterpretationreasoningTraining with small batch sizes is more sensitive to the All-Reduce time, since the algorithm needs to run All-Reduce more frequently, and vice versa.
“Intuitively, training with small batch sizes is more sensitive to the All-Reduce time (since the algorithm needs to run All-Reduce more frequently) and vice versa.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
PremisecitationLayer sharing reduces the number of aggregated gradients by a factor of how many times each layer is reused.
“Besides gradient compression, another effective technique is to use layer sharing (Lan et al., 2020), which reduces the number of aggregated gradients by a factor of how many times each layer is reused.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Lan et al., 2020
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Classified as probably not a substantial claim.
Premisenone givenThe compute time for naïve matrix multiplication scales as O(n^3).
“The compute time for naïve matrix multiplication scales as O(n3).”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing), S4 (based). The quote only partly states this claim.
Interpretationnone given§ Supplementary MaterialFor smaller models, a sharded data-parallel approach can be more optimal than SWARM.
“For smaller models, a sharded data-parallel approach can be more optimal.”
Quote in the paperNot a substantial claimClassified as probably not a substantial claim.
Premisecitation§ Supplementary MaterialNarayanan et al. (2021) measures GPU utilization as a fraction of theoretical peak FLOP/s of their GPUs.
“Narayanan et al. (2021) measures GPU utilization as a fraction of theoretical peak FLOP/s of their GPUs.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites Narayanan et al. (2021)
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
May combine two claims.
Interpretationreasoning§ Supplementary MaterialAs a rule of thumb, SWARM will run at a 20% slower speed than systems described by Narayanan et al. (2021) using infrastructure that is several times cheaper.
“As a rule of thumb, one can say that SWARM will run at a 20% slower speed than systems described by Narayanan et al. (2021) using the infrastructure that is several times cheaper.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Narayanan et al. (2021)
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
May combine two claims.
Interpretationreasoning§ Supplementary MaterialAs long as there is at least one operational peer at every pipeline stage and at least one trainer, SWARM can work without any issues.
“As long as there is at least one operational peer at every pipeline stage and at least one trainer, SWARM can work without any issues.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing), S4 (based).
Interpretationnone given§ Supplementary MaterialElastic sharded data parallelism is better than SWARM if the number of expected failures is relatively low.
“Elastic sharded data parallelism is better than SWARM if the number of expected failures is relatively low.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
PremisecitationDettmers (2016) uses 8-bit quantization to compress gradients before communication.
“Alternatively, Dettmers (2016) uses 8-bit quantization to compress gradients before communication.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Dettmers, 2016
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
May combine two claims.
Interpretationreasoning§ I. Compression-Aware ArchitecturesIf network bandwidth is unlimited, maxout or bottleneck compression would result in slower training.
“However, the same two techniques would result in slower training in a setup where network bandwidth is unlimited.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisenone given§ Supplementary MaterialSWARM's design allows for any number of pipeline stages.
“While its design allows for any number of stages, using long pipelines can result in a reduced training throughput.”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationreasoning§ Supplementary MaterialFor low-bandwidth devices with 100 Mb/s speed, sharded data parallelism would entail an overhead of 5.5 minutes per forward/backward pass, which is difficult to overlap with computation.
“For low-bandwidth devices with 100 Mb/s speed, this would entail an overhead of 5.5 minutes per forward/backward pass, which is difficult to overlap with computation.”
Quote in the paperSubstantial claimMay combine two claims.
Premisecitation§ I. Compression-Aware ArchitecturesDynamic 8-bit quantization with blockwise scaling reduces communication by approximately 2x and 4x for half and full precision, respectively.
“Instead, we use dynamic 8-bit quantization with blockwise scaling from (Dettmers et al., 2022). This technique reduces communication by ≈2x and ≈4x for half and full precision, respectively.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites Dettmers et al., 2022
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
PremisecitationDeep Gradient Compression sparsifies gradients and corrects momentum after synchronization.
“Deep Gradient Compression (Lin et al., 2018) sparsifies the gradients and corrects the momentum after synchronization”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Lin et al., 2018
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing), S4 (based).
Interpretationnone given§ Supplementary MaterialThe effectiveness of bottleneck and maxout compression may vary depending on where in the model they are applied and what kind of model they are applied to.
“However, their effectiveness may vary depending on where in the model they are applied and what kind of model these are applied to (for example, CNNs vs. RNNs vs. Transformers).”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Interpretationreasoning§ Supplementary MaterialSWARM can be useful in case of supercomputers that have heterogeneous devices.
“However, SWARM can be useful in case of supercomputers that have heterogeneous devices.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing), S4 (based).
Interpretationnone given§ Supplementary MaterialWithout layer sharing, the overhead within each pipeline stage to synchronize the gradients is increased by the number of layers not being shared.
“Without layer sharing, the overhead within each pipeline stage to synchronize the gradients is increased by the number of layers not being shared.”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationreasoning§ C. Stochastic Wiring DetailsSWARM produces exactly the same stepwise updates as conventional distributed training algorithms and will therefore achieve a solution in the same number of steps.
“That is, SWARM produces exactly the same stepwise updates as conventional distributed training algorithms and will therefore achieve a solution in the same number of steps.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ Supplementary MaterialMany HPC systems only achieve approximately 80% GPU utilization.
“while many HPC systems only achieve ≈ 80% GPU utilization.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Narayanan et al. (2021)
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
The paper relies on this without support.
Workers can join and leave training at any time.
“As we described in Section 2.2, our workers can join and leave training at any time.”
The paper relies on this without support.
The training devices are poorly connected, heterogeneous, and unreliable.
“we propose SWARM parallelism1, a model-parallel training algorithm designed for poorly connected, heterogeneous and unreliable devices.”
The authors state this as an assumption.
In the rebalancing simulation, when a worker is removed from the pipeline, the stage it was removed from is chosen uniformly at random over the four pipeline stages.
“When a worker is removed from the pipeline, we randomly choose the stage it was removed from: that is, removing N peers corresponds to N samples from the uniform distribution over four pipeline stages.”
The paper relies on this without support.
Executing a single pipeline stage 3 times locally on a single server and multiplying the single-node estimates by the number of nodes yields a valid estimate of the ideal best-case pipeline performance with infinite bandwidth and zero latency.
“The ideal case is emulated by executing a single pipeline stage 3 times locally on a single server and multiplying the single-node estimates by the number of nodes.”
The authors state this as an assumption.
The i.i.d. assumption in the large-scale training setup holds, so the distribution of gradients is expected to be the same across examples within a batch.
“because of the i.i.d. assumption in the large-scale training setup, the distribution of gradients is expected to be the same.”
The paper relies on this without support.
Any realistic deep learning workload cannot achieve 100% peak FLOP/s.
“Since any realistic deep learning workload cannot achieve 100% peak FLOP/s”
The paper relies on this without support.
The square-cube law for distributed training holds.
“When considered together with the square-cube law for distributed training and SWARM parallelism”
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Dynamic blockwise quantization halves the bandwidth requirements without any noticeable effect on generation quality.
“Dynamic blockwise quantization halves the bandwidth requirements without any noticeable effect on generation quality.”
From Petals: Collaborative Inference and Fine-tuning of Large Models (2022) · cites this paper as Ryabinin et al., 2023
In Appendix I (p. 23), Table 7 reports that dynamic blockwise 8-bit quantization halved data transfer relative to the 16-bit baseline and produced similar final perplexity (21.13 versus 21.02). The paper evaluates perplexity, not generated-output quality, and the halving applies to activation data transfer rather than necessarily to total bandwidth requirements. (LLM, conf=0.85)
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
Each contradiction between two passages listed here was confirmed by an independent check.
Figures label the "Ours" pipeline stage as 12 layers, but the setup says 3
MinorFigures 1 and 3 label the "Ours" configuration as having 12 layers at a stage, while the experiment description says each of its four servers hosts 3 layers. The discrepancy affects the setup behind the reported GPU-utilization result.
“ours 4096 units 12 layers”
“in the last one, there are 4 servers, each hosting 3 layers.”
Appendix describes four GPUs for configurations using twelve
MinorAppendix F says all four GPUs were on different servers, but its setup specifies 12 single-GPU servers for each of the first three configurations. Four GPUs describes only the "Ours" configuration, not all four benchmarked configurations.
“All four GPUs are in the same data center but on different servers.”
“For simplicity, we set up each experiment with 12 Trans- former layers using 12 servers (4 for "Ours") with a sin- gle V100-PCIE GPU each.”
Cites
The publisher hasn’t deposited this paper’s reference list.
Part of this paper’s reference list could not be split into single entries.
Research works printed but not linked: 1
- [148]148. IEEE, 2020. Zheng, L., Li, Z., Zhang, H., Zhuang, Y., Chen, Z., Huang, Y., Wang, Y., Xu, Y., Zhuo, D., Xing, E. P., Gonzalez, J. E., and Stoica, I. Alpa: Automating inter- and intra- operator parallelism for distributed deep learning, 2022. URL https://arxiv.org/abs/2201.12023. 17 https://openreview.net/forum?id=UHoGOaGjEq https://openreview.net/forum?id=UHoGOaGjEq https://arxiv.org/abs/2106.04560 https://arxiv.org/abs/2106.04560 https://doi.org/10.1007/978-3-319-20119-1_2 https://doi.org/10.1007/978-3-319-20119-1_2 https://arxiv.org/abs/2201.12023 SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient · DOI
Cited by
Sorted by citations
Friha, O., Ferrag, M. A., Kantarcı, B., Çakmak, B., Ozgun, A., & Ghoualmi‐Zine, N. (2024) · IEEE Open Journal of the Communications Society · doi:10.1109/ojcoms.2024.3456549 · Cited by 209 · see citing works
A Survey on Scheduling Techniques in Computing and Network Convergence
Tang, S., Yu, Y., Wang, H., Wang, G., Chen, W., Xu, Z., Guo, S., & Gao, W. (2023) · IEEE Communications Surveys & Tutorials · doi:10.1109/comst.2023.3329027 · Cited by 68 · see citing works
Petals: Collaborative Inference and Fine-tuning of Large Models
Borzunov, A., Baranchuk, D., Dettmers, T., Riabinin, M., Belkada, Y., Chumachenko, A., Samygin, P., & Raffel, C. (2023) · Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 3: System Demonstrations) · doi:10.18653/v1/2023.acl-demo.54 · Cited by 52 · see citing works
Self-citationHelix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flow
Mei, Y., Zhuang, Y., Miao, X., Yang, J., Jia, Z., & Vinayak, R. (2025) · Proceedings of the 30th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 1 · doi:10.1145/3669940.3707215 · Cited by 35 · see citing works
Achieving Peak Performance for Large Language Models: A Systematic Review
Rostam, Z. R. K., Szénási, S., & Kertész, G. (2024) · IEEE Access · doi:10.1109/access.2024.3424945 · Cited by 39 · see citing works
Sustainable Carbon-Aware and Water-Efficient LLM Scheduling in Geo-Distributed Cloud Datacenters
Moore, H., Qi, S., Hogade, N., Milojicic, D. S., Bash, C. E., & Pasricha, S. (2025) · ACM Great Lakes Symposium on VLSI (GLSVLSI) · doi:10.1145/3716368.3735301 · Cited by 11 · see citing works
Petals: Collaborative Inference and Fine-tuning of Large Models
Borzunov, A., Baranchuk, D., Dettmers, T., Ryabinin, M., Belkada, Y., Chumachenko, A., Samygin, P., & Raffel, C. (2022) · arXiv · doi:10.48550/arxiv.2209.01188 · Cited by 6 · see citing works
Self-citationPolygon training architecture for foundation models with network- and device-level heterogeneity
Li, C., Liu, F., Wu, X.-M., Huo, J., Wang, C., Liang, A., Zhao, Z., & Gao, L. (2025) · Information Fusion · doi:10.1016/j.inffus.2025.103264 · Cited by 2 · see citing works
Kulp, G., Gonzales, D., Smith, E., Heim, L., Puri, P., Vermeer, M. J. D., & Winkelman, Z. (2024) · RAND Corporation eBooks · doi:10.7249/wra3056-1 · Cited by 2 · see citing works
Distributed LLM Serving on Consumer-Grade GPUs by Reconciling Computation and Communication
Jin, L., Zhang, K., Chen, Y., Zhuoyifan, Li, R., Gao, Y., Yang, B., Cai, Z., & Dong, W. (2025) · Findings of the Association for Computational Linguistics: EMNLP 2025 · doi:10.18653/v1/2025.findings-emnlp.957
- Authors
- Ryabinin, M., Dettmers, T., Diskin, M., & Borzunov, A.
- Affiliations
- National Research University Higher School of Economics (RU)
- University of Washington (US)
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2023
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 2301.11913
- OpenAlex
- W4318621310
- Cited by
- 8
- References
- 0
- Citations by year
- 2025: 3 · 2024: 3 · 2023: 2
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Topics
- Physical Sciences › Computer Science › Hardware and Architecture › Parallel Computing and Optimization Techniques
- Physical Sciences › Computer Science › Artificial Intelligence › Stochastic Gradient Optimization Techniques
- Physical Sciences › Computer Science › Information Systems › Cloud Computing and Resource Management
- Keywords
- communication efficiency, heterogeneous devices, pipeline parallelism, parameter sharing, transformer language models, network communication, fault tolerance
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-31.
Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-31.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review