Gradient-Disentangled Embedding Sharing for NLU Pretraining
Find Innovative SolutionsGenerate 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
Engineering 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
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.
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.
2Adaptability or versatility
If upstream and downstream data embeddings are trained independently, then training flexibility is improved, but training efficiency deteriorates
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.
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.
Data Source
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.


