Hybrid Scheduling for Deep Learning Accelerators

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

There is a need for an optimal scheduling method to enhance resource efficiency in deep learning accelerators (DLAs) for AI workloads, as existing methods fail to effectively manage resource allocation and energy consumption.

Innovation Solution

A hybrid scheduling method that generates an input vector including resource status and metadata, uses an actor network to determine actions, performs first and second resource scheduling based on priority and heuristic algorithms, and evaluates performance using a critic network to select the best resource allocation for each task.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Productivity

If a single scheduling method is used for resource allocation, then the scheduling process is simple, but resource efficiency and energy consumption optimization are insufficient

Engineering Contradiction:
Improveresource efficiencyVSAvoidscheduling method complexity
Core Design Contradiction:
ProductivityVSDevice complexity

Solution Approach 1:

The patent combines multiple scheduling methods (reinforcement learning-based scheduling and heuristic-based scheduling) into a hybrid scheduling system. The reinforcement learning component handles complex optimization for energy consumption and resource efficiency, while the heuristic component provides fast, rule-based scheduling for immediate task allocation. This merging allows the system to achieve high resource efficiency without being overwhelmed by a single complex method.

Inventive Principle:
Principle #5Merging (Combining)

Solution Approach 2:

The scheduling system is segmented into multiple independent scheduling modules, each handling specific aspects of resource allocation. The reinforcement learning module focuses on long-term optimization and energy efficiency, while heuristic modules handle specific scheduling rules and constraints. This segmentation allows the system to manage complexity by dividing the scheduling problem into smaller, manageable parts that can be optimized independently.

Inventive Principle:
Principle #1Segmentation

2Loss of energy

If resource allocation is optimized for energy consumption, then energy efficiency improves, but scheduling time and computational overhead increase

Engineering Contradiction:
Improveenergy consumptionVSAvoidscheduling time
Core Design Contradiction:
Loss of energyVSLoss of time

Solution Approach 1:

The system applies partial optimization by using reinforcement learning for high-level resource allocation decisions that have significant energy impact, while using faster heuristic methods for detailed scheduling adjustments. This partial application of complex optimization only where most needed allows the system to reduce overall energy consumption without incurring the full computational overhead of optimizing every scheduling detail through reinforcement learning.

Inventive Principle:
Principle #16Partial or excessive action

Solution Approach 2:

The scheduling system dynamically switches between different scheduling strategies based on current system conditions. When energy optimization is critical and system load is manageable, the system uses reinforcement learning-based scheduling. When speed is critical or system state changes rapidly, the system switches to faster heuristic-based scheduling. This dynamic adaptation allows the system to optimize energy consumption without being permanently constrained by slow scheduling times.

Inventive Principle:
Principle #15Dynamics

3Manufacturing precision

If detailed resource scheduling is performed for each task, then scheduling granularity and resource allocation precision improve, but computational complexity and processing time increase

Engineering Contradiction:
Improveresource allocation precisionVSAvoidcomputational complexity
Core Design Contradiction:
Manufacturing precisionVSDevice complexity

Solution Approach 1:

The hybrid scheduling system applies different levels of detail and precision to different aspects of resource allocation. The reinforcement learning component provides high-level strategic allocation decisions with optimized precision for energy efficiency, while heuristic components handle detailed tactical scheduling with rule-based precision for immediate execution. This local differentiation of quality allows the system to achieve precise resource allocation without uniformly applying maximum computational complexity throughout the entire scheduling process.

Inventive Principle:
Principle #3Local quality

Data Source

PatentUS12056525B2Hybrid scheduling method for deep learning workloads, and computing apparatus with hybrid scheduling
Publication Date: 2024.08.06 SAMSUNG ELECTRONICS CO LTD
  • US12056525B2 patent drawing
  • US12056525B2 patent drawing
  • US12056525B2 patent drawing

AI summary

A scheduling method performed by a computing apparatus includes: generating an input vector including a resource status and metadata of each of tasks for parallel execution; determining an action for the input vector by executing an actor network based on the input vector; performing first resource scheduling for each of the tasks based on the determined action; performing second resource scheduling for each of the tasks based on the input vector; evaluating performance of first resource scheduling results of the first resource scheduling and second resource scheduling results of the second resource scheduling, for each of the tasks, using a critic network; selecting one of the first and second resource scheduling results for each of the tasks based on a result of the evaluating; and allocating resources to each of the tasks based on a resource scheduling result selected for each of the tasks.