TopoOpt: Co-optimizing Network Topology and Parallelization Strategy for Distributed Training Jobs
Not on the map. Why?
Abstract (arXiv)
We propose TopoOpt, a novel direct-connect fabric for deep neural network (DNN) training workloads. TopoOpt co-optimizes the distributed training process across three dimensions: computation, communication, and network topology. We demonstrate the mutability of AllReduce traffic, and leverage this property to construct efficient network topologies for DNN training jobs. TopoOpt then uses an alternating optimization technique and a group theory-inspired algorithm called TotientPerms to find the best network topology and routing plan, together with a parallelization strategy. We build a fully functional 12-node direct-connect prototype with remote direct memory access (RDMA) forwarding at 100 Gbps. Large-scale simulations on real distributed training models show that, compared to similar-cost Fat-Tree interconnects, TopoOpt reduces DNN training time by up to 3.4x.
Key facts. Cited by 30 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Feb 2022.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2022: 1; 2023: 7; 2024: 8; 2025: 10; 2026: 4.
30 papers.
Sort:
By area:
- 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 198
- 2023CASSINI: Network-Aware Job Scheduling in Machine Learning Clusterscited by 26
- 2023THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compressioncited by 15
- 2024Forecasting GPU Performance for Deep Learning Training and Inferencecited by 15
- 2023MAD Max Beyond Single-Node: Enabling Large Machine Learning Model Acceleration on Distributed Systemscited by 14
- 2023Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameterscited by 13
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Surveycited by 9
- 2024Swing: Short-cutting Rings for Higher Bandwidth Allreducecited by 8
- 2025MixNet: A Runtime Reconfigurable Optical-Electrical Fabric for Distributed Mixture-of-Experts Trainingcited by 7
- 2025InfiniteHBD: Building Datacenter-Scale High-Bandwidth Domain for LLM with Optical Circuit Switching Transceiverscited by 7
- 2022Efficient Direct-Connect Topologies for Collective Communicationscited by 5
- 2025Characterizing the Efficiency of Distributed Training: A Power, Performance, and Thermal Perspectivecited by 5
18 more
- 2023TACOS: Topology-Aware Collective Algorithm Synthesizer for Distributed Machine Learningcited by 4
- 2023Holmes: Towards Distributed Training Across Clusters with Heterogeneous NIC Environmentcited by 4
- 2025HierMoE: Accelerating MoE Training with Hierarchical Token Deduplication and Expert Swapcited by 4
- 2024Communication Optimization for Distributed Training: Architecture, Advances, and Opportunitiescited by 3
- 2025Towards Easy and Realistic Network Infrastructure Testing for Large-scale Machine Learningcited by 3
- 2024ForestColl: Throughput-Optimal Collective Communications on Heterogeneous Network Fabricscited by 2
- 2024Credence: Augmenting Datacenter Switch Buffer Sharing with ML Predictionscited by 1
- 2025Efficient AllReduce with Stragglerscited by 1
- 2026NetKV: Network-Aware Decode Instance Selection for Disaggregated LLM Inferencecited by 1
- 2024MLCC: A Congestion Control Technique to Accelerate ML Trainingcited by 0
- 2024VcLLM: Video Codecs are Secretly Tensor Codecscited by 0
- 2025Routing for Large ML Modelscited by 0
- 2025LUMION: Fast Fault Recovery for ML Jobs Using Programmable Optical Fabricscited by 0
- 2025Morphlux: Transforming Torus Fabrics for Efficient Multi-tenant MLcited by 0
- 2025Enabling Reconfiguration-Communication Overlap for Collective Communication in Optical Networkscited by 0
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Servingcited by 0
- 2026Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimizationcited by 0
- 2026AoiZora: Topology-Aware Auto-Parallel Optimization for Inference of Diffusion Transformerscited by 0
- 2026AoiZora: Topology-Aware Auto-Parallel Optimization for Inference of Diffusion Transformers16 Jun 2026
- 2026NetKV: Network-Aware Decode Instance Selection for Disaggregated LLM Inference2 Jun 2026
- 2026Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization25 May 2026
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Serving30 Apr 2026
- 2025Enabling Reconfiguration-Communication Overlap for Collective Communication in Optical Networks22 Oct 2025
- 2025Characterizing the Efficiency of Distributed Training: A Power, Performance, and Thermal Perspective12 Sep 2025
- 2025HierMoE: Accelerating MoE Training with Hierarchical Token Deduplication and Expert Swap13 Aug 2025
- 2025Morphlux: Transforming Torus Fabrics for Efficient Multi-tenant ML20 Jul 2025
- 2025Efficient AllReduce with Stragglers29 May 2025
- 2025LUMION: Fast Fault Recovery for ML Jobs Using Programmable Optical Fabrics29 May 2025
- 2025Towards Easy and Realistic Network Infrastructure Testing for Large-scale Machine Learning29 Apr 2025
- 2025Routing for Large ML Models7 Mar 2025
18 more
- 2025InfiniteHBD: Building Datacenter-Scale High-Bandwidth Domain for LLM with Optical Circuit Switching Transceivers6 Feb 2025
- 2025MixNet: A Runtime Reconfigurable Optical-Electrical Fabric for Distributed Mixture-of-Experts Training7 Jan 2025
- 2024Forecasting GPU Performance for Deep Learning Training and Inference18 Jul 2024
- 2024VcLLM: Video Codecs are Secretly Tensor Codecs29 Jun 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024Communication Optimization for Distributed Training: Architecture, Advances, and Opportunities12 Mar 2024
- 2024MLCC: A Congestion Control Technique to Accelerate ML Training14 Feb 2024
- 2024ForestColl: Throughput-Optimal Collective Communications on Heterogeneous Network Fabrics9 Feb 2024
- 2024Swing: Short-cutting Rings for Higher Bandwidth Allreduce17 Jan 2024
- 2024Credence: Augmenting Datacenter Switch Buffer Sharing with ML Predictions5 Jan 2024
- 2023Holmes: Towards Distributed Training Across Clusters with Heterogeneous NIC Environment6 Dec 2023
- 2023MAD Max Beyond Single-Node: Enabling Large Machine Learning Model Acceleration on Distributed Systems4 Oct 2023
- 2023CASSINI: Network-Aware Job Scheduling in Machine Learning Clusters1 Aug 2023
- 2023Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameters22 Jul 2023
- 2023TACOS: Topology-Aware Collective Algorithm Synthesizer for Distributed Machine Learning11 Apr 2023
- 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddings4 Apr 2023
- 2023THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compression16 Feb 2023
- 2022Efficient Direct-Connect Topologies for Collective Communications7 Feb 2022
- 2023Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameterscited by 13
- 2025MixNet: A Runtime Reconfigurable Optical-Electrical Fabric for Distributed Mixture-of-Experts Trainingcited by 7
- 2025HierMoE: Accelerating MoE Training with Hierarchical Token Deduplication and Expert Swapcited by 4
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Servingcited by 0
- 2026Rethinking Network Topologies for Cost-Effective Mixture-of-Experts LLM Serving30 Apr 2026
- 2025HierMoE: Accelerating MoE Training with Hierarchical Token Deduplication and Expert Swap13 Aug 2025
- 2025MixNet: A Runtime Reconfigurable Optical-Electrical Fabric for Distributed Mixture-of-Experts Training7 Jan 2025
- 2023Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameters22 Jul 2023
- 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 198
- 2023CASSINI: Network-Aware Job Scheduling in Machine Learning Clusterscited by 26
- 2023THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compressioncited by 15
- 2024Forecasting GPU Performance for Deep Learning Training and Inferencecited by 15
- 2023MAD Max Beyond Single-Node: Enabling Large Machine Learning Model Acceleration on Distributed Systemscited by 14
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Surveycited by 9
- 2024Swing: Short-cutting Rings for Higher Bandwidth Allreducecited by 8
- 2025InfiniteHBD: Building Datacenter-Scale High-Bandwidth Domain for LLM with Optical Circuit Switching Transceiverscited by 7
- 2022Efficient Direct-Connect Topologies for Collective Communicationscited by 5
- 2025Characterizing the Efficiency of Distributed Training: A Power, Performance, and Thermal Perspectivecited by 5
- 2023TACOS: Topology-Aware Collective Algorithm Synthesizer for Distributed Machine Learningcited by 4
- 2023Holmes: Towards Distributed Training Across Clusters with Heterogeneous NIC Environmentcited by 4
12 most cited of 23 All 23 (JSON)
- 2025Enabling Reconfiguration-Communication Overlap for Collective Communication in Optical Networks22 Oct 2025
- 2025Characterizing the Efficiency of Distributed Training: A Power, Performance, and Thermal Perspective12 Sep 2025
- 2025Morphlux: Transforming Torus Fabrics for Efficient Multi-tenant ML20 Jul 2025
- 2025Efficient AllReduce with Stragglers29 May 2025
- 2025LUMION: Fast Fault Recovery for ML Jobs Using Programmable Optical Fabrics29 May 2025
- 2025Towards Easy and Realistic Network Infrastructure Testing for Large-scale Machine Learning29 Apr 2025
- 2025Routing for Large ML Models7 Mar 2025
- 2025InfiniteHBD: Building Datacenter-Scale High-Bandwidth Domain for LLM with Optical Circuit Switching Transceivers6 Feb 2025
- 2024Forecasting GPU Performance for Deep Learning Training and Inference18 Jul 2024
- 2024VcLLM: Video Codecs are Secretly Tensor Codecs29 Jun 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024Communication Optimization for Distributed Training: Architecture, Advances, and Opportunities12 Mar 2024
12 newest of 23 All 23 (JSON)