Pipeline Protocol for Distributed ML Worker Nodes

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Deep neural networks (DNNs) training has become increasingly compute and communication intensive, leading to resource inefficiencies and idle time due to the complexity of models and current aggregation protocols that limit training speed and cause resource stalling.

Innovation Solution

Implementing a pipeline-based protocol flow for aggregation trees that allows worker nodes to continue gradient calculations and send data for alternate models while initial gradient data propagates, using middleware protocols to manage communication and resource utilization efficiently.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Productivity

If traditional aggregation protocols are used for DNN training, then model gradient calculations can be performed, but resource stalling and idle time occur due to sequential processing requirements

Engineering Contradiction:
Improvetraining speedVSAvoididle time
Core Design Contradiction:
ProductivityVSLoss of time

Solution Approach 1:

The patent segments the gradient aggregation process into multiple parallel pipelines. Instead of having all worker nodes wait for sequential aggregation, the system divides worker nodes into different pipelines that can independently aggregate and update models simultaneously. This segmentation eliminates the sequential bottleneck and prevents resource stalling, allowing continuous utilization of worker nodes across multiple parallel model training operations.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent implements a pipeline-based protocol that enables continuous useful action by allowing worker nodes to immediately begin gradient calculations for alternate models after sending initial gradient data, without waiting for aggregation completion. The pipeline architecture ensures that while one model's gradients are being aggregated, worker nodes are already processing gradients for other models, eliminating idle time and maintaining continuous productive work.

Inventive Principle:
Principle #20Continuity of useful action

2Productivity

If current aggregation protocols are used, then gradient data can be aggregated from worker nodes, but resource utilization is inefficient due to worker nodes being blocked during data propagation

Engineering Contradiction:
Improveresource utilizationVSAvoidprotocol complexity
Core Design Contradiction:
ProductivityVSDevice complexity

Solution Approach 1:

The patent introduces dynamic pipeline management where worker nodes can dynamically switch between different model training tasks based on pipeline status. The system dynamically assigns worker nodes to different pipelines and models, allowing flexible resource allocation that maximizes utilization. This dynamic approach replaces the static, blocked workflow of traditional protocols with an adaptive system that continuously optimizes resource usage.

Inventive Principle:
Principle #15Dynamics

Solution Approach 2:

The patent introduces pipeline managers as intermediary components that coordinate between worker nodes and model aggregation processes. These intermediaries handle the complexity of pipeline coordination, model assignment, and gradient routing, freeing worker nodes from protocol management overhead. The intermediary layer absorbs the protocol complexity while presenting a simplified interface to worker nodes, improving resource utilization without burdening the computing resources with protocol management.

Inventive Principle:
Principle #24Intermediary (Mediator)

3Duration of action of moving object

If worker nodes process single model sequentially, then gradient aggregation is simplified, but training time increases due to inability to process multiple models simultaneously

Engineering Contradiction:
Improvetraining timeVSAvoidprocessing throughput
Core Design Contradiction:
Duration of action of moving objectVSProductivity

Solution Approach 1:

The patent segments the model training workload across multiple parallel pipelines, allowing worker nodes to process gradients for multiple models simultaneously rather than sequentially. Each pipeline independently handles a subset of models, and worker nodes can contribute to multiple pipelines concurrently. This segmentation multiplies the effective processing throughput while maintaining manageable aggregation complexity within each pipeline.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent transitions from single-dimension sequential model processing to multi-dimensional parallel processing by introducing the pipeline dimension. Instead of processing models one after another in a single sequence, the system creates multiple parallel dimensions (pipelines) where model training occurs simultaneously. This dimensional expansion allows worker nodes to operate on multiple models at once, dramatically reducing total training time while maintaining coordination through the pipeline structure.

Inventive Principle:
Principle #17Another dimension (Dimensionality change)

Data Source

PatentUS20230129511A1Methods, systems, articles of manufacture and apparatus to improve distributed machine learning efficiency
Publication Date: 2023.04.27 INTEL CORP
  • US20230129511A1 patent drawing
  • US20230129511A1 patent drawing
  • US20230129511A1 patent drawing

AI summary

Methods, apparatus, systems, and articles of manufacture are disclosed to improve distributed machine learning efficiency. An example apparatus includes train management circuitry to cause a first vector to be sent from a worker node to an in-network-aggregator (INA) after completion of a first processing iteration requested by a parameter server. The example apparatus also includes protocol configuration circuitry to prohibit a second processing iteration when an availability status of the INA is false, and permit the second processing iteration when (a) an acknowledgement (ACK) from the INA corresponding to the first vector is received and (b) the availability status of the INA is true.