Dialogue System Memory Reduction via Model Segmentation

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Dialogue systems face challenges due to large memory requirements and long training times, especially when adapting to new tasks with scarce annotated data, as existing models are inefficient and require significant resources.

Innovation Solution

A memory-efficient value extractor model is developed, utilizing a transformer-based neural architecture with a Conditional Random Fields (CRF) approach, which segments user inputs into subword units and employs attention layers for slot value extraction, allowing for efficient training and adaptation to new domains with limited data through pre-training and fine-tuning.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Reliability

If large models with many trainable parameters are used for slot labelling, then the model performance is improved, but the memory requirements increase and training time increases

Engineering Contradiction:
Improvemodel performanceVSAvoidmemory requirements
Core Design Contradiction:
ReliabilityVSWeight of stationary object

Solution Approach 1:

The model is divided into two separate components: a language model that processes user inputs and a slot labelling model that extracts slot values. This segmentation allows each component to be optimized independently, reducing the overall memory footprint while maintaining performance. The language model can be smaller since it only processes text, while the slot labelling model focuses specifically on extraction tasks.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The slot labelling functionality is extracted as a separate model component rather than being integrated into a single large model. This extraction allows the system to use a smaller language model combined with a specialized slot labelling model, reducing total memory requirements while maintaining extraction performance through dedicated optimization.

Inventive Principle:
Principle #2Taking out (Extraction)

2Reliability

If large models with many trainable parameters are used for slot labelling, then the model performance is improved, but the training time increases

Engineering Contradiction:
Improvemodel performanceVSAvoidtraining time
Core Design Contradiction:
ReliabilityVSLoss of time

Solution Approach 1:

By segmenting the model into separate language processing and slot labelling components, each model can be trained independently and more efficiently. The slot labelling model can be trained separately on annotated data without requiring the entire large model to be retrained, significantly reducing training time while maintaining performance.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The language model is pre-trained on general language data before being combined with the slot labelling model. This preliminary training allows the system to leverage pre-computed language understanding, avoiding the need to retrain the entire model from scratch when adding slot labelling capabilities, thus reducing training time.

Inventive Principle:
Principle #10Preliminary action

3Adaptability or versatility

If existing models are used for new tasks, then task coverage is expanded, but the training becomes challenging due to scarcity of annotated data

Engineering Contradiction:
Improvetask coverageVSAvoidtraining difficulty
Core Design Contradiction:
Adaptability or versatilityVSEase of manufacture

Solution Approach 1:

The slot labelling model is extracted as a separate trainable component that can be fine-tuned on domain-specific annotated data. This allows the system to adapt to new tasks by training only the slot labelling model on scarce annotated data for the new domain, while reusing the pre-trained language model, making training easier and more feasible for new tasks.

Inventive Principle:
Principle #2Taking out (Extraction)

Solution Approach 2:

The system allows for parameter changes in the slot labelling model by training it on domain-specific annotated data. This enables adaptation to new tasks and domains by adjusting the slot labelling model's parameters to match the specific task requirements, while maintaining the benefit of the pre-trained language model's general knowledge.

Inventive Principle:
Principle #35Parameter changes

Data Source

PatentUS11132988B1Dialogue system, a dialogue method, and a method of training
Publication Date: 2021.09.28 POLYAI LTD
  • US11132988B1 patent drawing
  • US11132988B1 patent drawing
  • US11132988B1 patent drawing

AI summary

A computer implemented method comprising: receiving input data relating to a speech or text signal originating from a user; representing the input data as a first sequence of first representations, each representing a unit of the input data; representing the input data as a second sequence of second representations, each representing one of the units of the input data; using a model to determine a tag sequence from the first sequence of first representations, wherein the model comprises an attention layer using the second sequence of second representations, wherein the tag sequence comprises one or more tags from a set of tags comprising a first tag; if one or more units of the input data correspond to the first tag, determining a system dialogue act based on the part of the input data corresponding to the first tag; and outputting speech or text information specified by the determined dialogue act.