Skill Discovery for Imitation Learning with Noisy Data

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Imitation learning faces challenges in training models with a small number of high-quality demonstrations and a larger number of noisy demonstrations, where noisy demonstrations may not follow the best strategy, leading to inaccurately trained models.

Innovation Solution

A method involving skill discovery using a combination of known-good and noisy demonstrations to generate skills, where a unidirectional skill embedding model is trained while keeping skill matching model and low-level policy parameters constant, and then trained together in an end-to-end fashion to extract useful information from noisy demonstrations.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Measurement precision

If a model is trained using a small number of high-quality demonstrations, then the model training accuracy is improved, but the quantity of training data is insufficient

Engineering Contradiction:
Improvemodel training accuracyVSAvoidquantity of training demonstrations
Core Design Contradiction:
Measurement precisionVSQuantity of substance

Solution Approach 1:

The patent segments the training demonstrations into two distinct categories: high-quality demonstrations and noisy demonstrations. This segmentation allows the system to process and learn from different types of data separately, extracting useful information from noisy demonstrations while maintaining the quality standards set by high-quality demonstrations. The segmentation resolves the contradiction by enabling the model to leverage both small amounts of high-quality data and large amounts of noisy data without compromising training accuracy.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent introduces skill discovery as an intermediary process that bridges high-quality and noisy demonstrations. By first discovering skills from high-quality demonstrations and then using these skills to guide learning from noisy demonstrations, the system acts as a mediator that filters and processes information. This intermediary approach allows the model to benefit from the quantity of noisy demonstrations while maintaining the quality standards established by the smaller set of high-quality demonstrations.

Inventive Principle:
Principle #24Intermediary (Mediator)

2Quantity of substance

If noisy demonstrations are used for training, then the quantity of training data is increased, but the model training accuracy deteriorates

Engineering Contradiction:
Improvequantity of training demonstrationsVSAvoidmodel training accuracy
Core Design Contradiction:
Quantity of substanceVSMeasurement precision

Solution Approach 1:

The patent converts the harmful effect of noisy demonstrations into a benefit by using skill discovery to extract useful patterns and skills from the noise. Instead of discarding noisy demonstrations due to their low quality, the system processes them through a skill-based framework that identifies and learns from valid action sequences embedded within the noise. This transforms the previously harmful noisy data into a valuable training resource, increasing data quantity without sacrificing accuracy.

Inventive Principle:
Principle #22Blessing in disguise (Convert harm into benefit)

Solution Approach 2:

The skill matching model serves as an intermediary that filters noisy demonstrations before they reach the main training process. By matching demonstrations against discovered skills, the system identifies and utilizes only the useful portions of noisy demonstrations while filtering out the noise. This intermediary filtering mechanism enables the model to leverage large quantities of noisy data while maintaining training accuracy through selective learning.

Inventive Principle:
Principle #24Intermediary (Mediator)

3Loss of information

If skill discovery is performed using both high-quality and noisy demonstrations, then the interpretability of the model is improved, but the complexity of the training process increases

Engineering Contradiction:
Improvemodel interpretabilityVSAvoidtraining process complexity
Core Design Contradiction:
Loss of informationVSDevice complexity

Solution Approach 1:

The patent segments the training process into distinct phases: skill discovery from high-quality demonstrations, skill matching with noisy demonstrations, and end-to-end training. This segmentation provides clear interpretability for each stage while managing complexity through structured progression. Each segment has a specific purpose and can be analyzed independently, maintaining interpretability even as the overall process complexity increases.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent maintains continuity of useful action by ensuring that skills discovered in the first phase continue to be utilized and refined in subsequent phases. The skill matching model continuously refines the discovered skills using noisy demonstrations, and the end-to-end training further optimizes them. This continuous refinement process maintains interpretability by building upon previously established skills rather than restarting the discovery process, thereby managing complexity through iterative improvement.

Inventive Principle:
Principle #20Continuity of useful action

Data Source

PatentUS20240062070A1Skill discovery for imitation learning
Publication Date: 2024.02.22 NEC LABORATORIES AMERICA INC
  • US20240062070A1 patent drawing
  • US20240062070A1 patent drawing
  • US20240062070A1 patent drawing

AI summary

Methods and systems for training a model include performing skill discovery, using a set of demonstrations that includes known-good demonstrations and noisy demonstrations, to generate a set of skills. A unidirectional skill embedding model is trained in a first training while parameters of a skill matching model and low-level policies that relate skills to actions are held constant. The unidirectional skill embedding model, the skill matching model, and the low-level policies are trained together in an end-to-end fashion in a second training.