Neural Memory Training Loss for Escaping Local Minima

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Machine learning systems, such as Neural Turing Machines (NTMs), often converge to local minima during training, leading to poor generalization capabilities and limited memory utilization, especially when dealing with data sequences longer than the training data.

Innovation Solution

Implement a modified loss function that includes a penalty term associated with the difference and magnitude of read and write gradient differences, and restart training if convergence to local minima is detected, using gradient descent methods to reinitialize the system.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Productivity

If standard gradient descent training is used for machine learning systems, then training speed is improved, but the system converges to local minima resulting in poor generalization

Engineering Contradiction:
Improvetraining speedVSAvoidgeneralization capability
Core Design Contradiction:
ProductivityVSReliability

Solution Approach 1:

The patent implements a feedback mechanism by monitoring gradient statistics (mean and variance) during training and using this information to dynamically adjust the learning rate. This feedback loop prevents convergence to local minima by adapting the training process based on observed gradient patterns, thereby maintaining both training efficiency and generalization capability.

Inventive Principle:
Principle #23Feedback

Solution Approach 2:

The patent dynamically changes the learning rate parameter based on gradient statistics during training. By adjusting this critical parameter in response to observed gradient patterns, the system can escape local minima while maintaining efficient training progression, resolving the contradiction between training speed and generalization capability.

Inventive Principle:
Principle #35Parameter changes

2Duration of action of stationary object

If training continues without modification, then training completeness is improved, but the system utilizes only a fraction of available memory locations

Engineering Contradiction:
Improvetraining completenessVSAvoidmemory utilization
Core Design Contradiction:
Duration of action of stationary objectVSAdaptability or versatility

Solution Approach 1:

The patent introduces dynamic adjustments to the training process by modifying the learning rate based on gradient statistics. This dynamic approach prevents the system from settling into static patterns that would limit memory utilization, allowing the system to continue exploring different memory locations throughout training and achieving both completeness and adaptability.

Inventive Principle:
Principle #15Dynamics

3Stability of the object's composition

If the learning rate is kept constant for stable training, then training stability is improved, but the system cannot escape local minima

Engineering Contradiction:
Improvetraining stabilityVSAvoidconvergence to global minimum
Core Design Contradiction:
Stability of the object's compositionVSReliability

Solution Approach 1:

The patent transforms the static learning rate into a dynamic parameter that adapts during training based on gradient statistics. This allows the system to maintain stability when gradients are consistent while automatically increasing exploration when gradient variance indicates potential local minima, thus resolving the contradiction between stability and convergence to global minimum.

Inventive Principle:
Principle #15Dynamics

Solution Approach 2:

The patent changes the learning rate parameter dynamically during training based on observed gradient patterns. This parameter adaptation allows the system to maintain stability during stable training phases while escaping local minima when gradient statistics indicate convergence issues, achieving both stability and global optimization.

Inventive Principle:
Principle #35Parameter changes

Data Source

PatentUS12518153B2Training machine learning systems
Publication Date: 2026.01.06 INTERNATIONAL BUSINESS MACHINE CORPORATION
  • US12518153B2 patent drawing
  • US12518153B2 patent drawing
  • US12518153B2 patent drawing

AI summary

Training machine learning systems using a training data set, gradient descent, and a loss function. The machine learning system includes memory and reads and writes to memory according to read and write profiles. The loss function is associated with machine learning system memory read and write profile gradients. The loss function includes a loss function penalty term, the loss function penalty term being associated with the read and write profile gradient differences. Trained machine learning systems are then provided.