Gradient-Disentangled Embedding Sharing for NLU Pretraining

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

The process of training AI constructs for natural language understanding (NLU) is computationally expensive due to the need to train on vast corpora of language input, such as entire encyclopedias.

Innovation Solution

A language-processing service configured for NLU, utilizing a language model with upstream and downstream sequences of transformer blocks, an input module, and an output module. The pretraining logic adjusts the upstream and downstream data embeddings by computing a gradient of the upstream data embedding disentangled from the gradient of the downstream data embedding.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Reliability

If language models are trained on vast corpora of language input to achieve strong NLU performance, then model performance is improved, but computational resource consumption increases

Engineering Contradiction:
Improvemodel performanceVSAvoidcomputational resource consumption
Core Design Contradiction:
ReliabilityVSUse of energy by moving object

Solution Approach 1:

The patent segments the training process into two distinct phases: pretraining on vast corpora to learn general language patterns, and fine-tuning on task-specific datasets. This segmentation allows the model to achieve strong NLU performance through pretraining while reducing computational resource consumption during the specialized fine-tuning phase by leveraging the pre-learned representations.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent applies preliminary action by performing pretraining on vast language corpora before the actual task-specific training. This preliminary pretraining phase equips the model with general language understanding capabilities, so that subsequent fine-tuning requires fewer computational resources to achieve the same level of performance on specific NLU tasks.

Inventive Principle:
Principle #10Preliminary action

2Adaptability or versatility

If upstream and downstream data embeddings are trained independently, then training flexibility is improved, but training efficiency deteriorates

Engineering Contradiction:
Improvetraining flexibilityVSAvoidtraining efficiency
Core Design Contradiction:
Adaptability or versatilityVSProductivity

Solution Approach 1:

The patent merges the training of upstream and downstream data embeddings by implementing gradient sharing between them. During fine-tuning, gradients from the downstream task flow back through the shared embedding layer, allowing both upstream and downstream embeddings to be updated simultaneously. This combining approach improves training efficiency while preserving the flexibility to handle different tasks through the modular encoder-decoder structure.

Inventive Principle:
Principle #5Merging (Combining)

Solution Approach 2:

The patent creates universal data embeddings that serve multiple functions: they act as input representations for the decoder during generation and as updated parameters during fine-tuning. This multi-functionality allows the same embedding layer to adapt to different tasks while improving training efficiency through shared gradient updates, resolving the contradiction between flexibility and efficiency.

Inventive Principle:
Principle #6Universality (Multi-functionality)

Data Source

PatentUS12223269B2Language-model pretraining with gradient-disentangled embedding sharing
Publication Date: 2025.02.11 MICROSOFT TECHNOLOGY LICENSING LLC
  • US12223269B2 patent drawing
  • US12223269B2 patent drawing
  • US12223269B2 patent drawing

AI summary

A method for training a language model comprises (a) receiving vectorized training data as input to a multitask pretraining problem; (b) generating modified vectorized training data based on the vectorized training data, according to an upstream data embedding; (c) emitting pretraining output based on the modified vectorized training data, according to a downstream data embedding equivalent to the upstream data embedding; and (d) adjusting the upstream data embedding and the downstream data embedding by computing, based on the pretraining output, a gradient of the upstream data embedding disentangled from a gradient of the downstream data embedding, thereby advancing the multitask pretraining problem toward a pretrained state.