Lightweight Prompt Encoder for Frozen Embedding Models

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Existing embedding models require significant learning costs due to their large scale, even when fine-tuning, which hinders efficient data embedding and performance improvement.

Innovation Solution

A method involving a lightweight prompt encoder that generates prompts for a pretrained embedding model, allowing the model to be updated while freezing the embedding model, thereby reducing learning costs and enhancing embedding performance.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Measurement precision

If a large-scale embedding model is used to improve embedding performance, then embedding performance is improved, but learning cost increases significantly

Engineering Contradiction:
Improveembedding performanceVSAvoidlearning cost
Core Design Contradiction:
Measurement precisionVSUse of energy by moving object

Solution Approach 1:

The patent segments the embedding system into two distinct components: a frozen pretrained embedding model (e.g., BERT, RoBERTa) that provides the core embedding capabilities, and a lightweight prompt encoder that generates prompts for specific tasks. This segmentation allows the heavy computational burden to be divided, where the pretrained model handles general embedding tasks with its fixed parameters, while only the lightweight prompt encoder requires updating during fine-tuning, significantly reducing the learning cost while maintaining high embedding performance.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent uses the concept of copying by creating a prompt encoder that generates prompts as representations of input data. These prompts are then input to the frozen embedding model to generate embedding representations. This copying approach allows the system to leverage the pretrained model's knowledge without requiring full fine-tuning, as the prompts serve as compressed representations that guide the embedding generation process.

Inventive Principle:
Principle #26Copying

2Adaptability or versatility

If a pretrained embedding model is fine-tuned to adapt to specific tasks, then task-specific performance is improved, but computational resources required increase

Engineering Contradiction:
Improvetask-specific performanceVSAvoidcomputational resources
Core Design Contradiction:
Adaptability or versatilityVSPower

Solution Approach 1:

The patent extracts the task-specific adaptation component from the full embedding model by separating it into a dedicated prompt encoder. The pretrained embedding model remains frozen, extracting only the necessary adaptability functionality into the prompt encoder. This allows task-specific performance to be improved through prompt generation without requiring computational resources to be expended on retraining the entire large-scale embedding model.

Inventive Principle:
Principle #2Taking out (Extraction)

Solution Approach 2:

Instead of performing full fine-tuning of the embedding model (excessive action), the patent applies partial action by only updating the prompt encoder parameters. This partial adaptation approach is sufficient for achieving task-specific performance improvements while consuming significantly fewer computational resources than would be required for complete model fine-tuning.

Inventive Principle:
Principle #16Partial or excessive action

Data Source

PatentUS20240020578A1Method for embedding data and system thereof
Publication Date: 2024.01.18 SAMSUNG SDS CO LTD
  • US20240020578A1 patent drawing
  • US20240020578A1 patent drawing
  • US20240020578A1 patent drawing

AI summary

Methods and apparatuses for embedding data. The method for embedding data includes: acquiring a pretrained embedding model; generating a prompt associated with a data sample through a prompt encoder, the prompt encoder being lighter than the embedding model; generating an embedding representation of the data sample by inputting the prompt and the data sample to the embedding model; calculating a task loss by performing a predefined task by using the embedding representation; and updating the prompt encoder based on the task loss.