Hardware-Aware Neural Network Sparsity for Target Platform Execution

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Existing machine learning models, particularly neural networks, are not optimized for specific hardware platforms, leading to suboptimal performance and efficiency when executed on accelerators other than those used for training, and current sparsity techniques fail to consider hardware architecture, resulting in models with high sparsity but poor execution on target hardware.

Innovation Solution

A hardware-aware model training controller applies reinforcement learning to identify sparsity ratios for neural network layers based on the architecture of the target hardware platform, optimizing sparsity and performance for CPUs, GPUs, and VPUs.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Productivity

If existing machine learning models are trained without hardware awareness, then model training is simpler and faster, but execution performance and efficiency on target hardware platforms deteriorate

Engineering Contradiction:
Improveexecution performanceVSAvoidtraining complexity
Core Design Contradiction:
ProductivityVSDevice complexity

Solution Approach 1:

The patent applies preliminary action by incorporating hardware architecture information into the model training process beforehand. The training controller receives hardware configuration data and uses it to guide the reinforcement learning agent in optimizing sparsity ratios during training, rather than attempting to optimize after training is complete. This preliminary integration of hardware awareness enables the model to be pre-optimized for target hardware, resolving the contradiction between training simplicity and execution performance.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The patent employs parameter changes by dynamically adjusting sparsity ratios based on hardware architecture characteristics. The reinforcement learning agent learns to modify the sparsity parameters of different network layers according to the specific hardware platform's features (such as GPU vs. CPU architectures). This parameter optimization during training improves execution performance on target hardware without requiring complete retraining, thus managing the trade-off between training complexity and performance.

Inventive Principle:
Principle #35Parameter changes

2Quantity of substance

If sparsity techniques are applied without considering hardware architecture, then model size is reduced, but execution efficiency on target hardware deteriorates

Engineering Contradiction:
Improvemodel sizeVSAvoidexecution efficiency
Core Design Contradiction:
Quantity of substanceVSProductivity

Solution Approach 1:

The patent applies local quality by implementing different sparsity ratios for different layers of the neural network based on their specific characteristics and the target hardware architecture. Rather than applying uniform sparsity across all layers, the reinforcement learning agent learns to assign appropriate sparsity levels to specific layers (e.g., higher sparsity for convolutional layers, lower for fully connected layers) considering the hardware's computational capabilities. This localized optimization maintains model size reduction while preserving execution efficiency on the target hardware platform.

Inventive Principle:
Principle #3Local quality

Solution Approach 2:

The patent employs dynamics by making sparsity ratios adaptive rather than static. The reinforcement learning framework enables the system to dynamically determine optimal sparsity ratios during training based on hardware configuration and performance feedback. This dynamic approach allows the model to achieve efficient execution on specific hardware platforms while maintaining reasonable model size, resolving the contradiction between compression and efficiency.

Inventive Principle:
Principle #15Dynamics

3Productivity

If reinforcement learning is used to optimize sparsity ratios based on hardware architecture, then execution performance on target hardware is improved, but training time and computational resources increase

Engineering Contradiction:
Improveexecution performanceVSAvoidtraining time
Core Design Contradiction:
ProductivityVSLoss of time

Solution Approach 1:

The patent applies feedback by implementing a reinforcement learning framework where the training controller receives feedback about hardware execution performance and uses it to adjust sparsity ratios iteratively. The system monitors the performance impact of different sparsity configurations on the target hardware and feeds this information back to the agent, which then refines its sparsity decisions. This feedback mechanism enables the system to achieve optimal execution performance while managing training time by learning from performance outcomes rather than exhaustive searching.

Inventive Principle:
Principle #23Feedback

Solution Approach 2:

The patent employs partial action by applying reinforcement learning optimization selectively to critical layers of the neural network rather than uniformly to all layers. The system identifies which layers benefit most from hardware-aware sparsity optimization and focuses computational resources there, while using simpler optimization methods for less critical layers. This partial application of the complex reinforcement learning process reduces overall training time while still achieving significant execution performance improvements on target hardware.

Inventive Principle:
Principle #16Partial or excessive action

Data Source

PatentUS20250371349A1Methods and apparatus for hardware-aware machine learning model training
Publication Date: 2025.12.04 INTEL CORP
  • US20250371349A1 patent drawing
  • US20250371349A1 patent drawing
  • US20250371349A1 patent drawing

AI summary

Methods, apparatus, systems, and articles of manufacture are disclosed for hardware-aware machine learning model training. An example apparatus includes a configuration determiner to determine a hardware configuration of a target hardware platform on which the machine learning model is to be executed, a layer generator to assign sparsity configurations to layers of the machine learning model based on the hardware configuration, and a deployment controller to deploy the machine learning model to the target hardware platform in response to outputs of the machine learning model satisfying respective thresholds, the outputs including a quantity of clock cycles to execute the machine learning model with the layers having the assigned sparsity configurations.