Federated Training Checkpoint Migration Across Edge Servers

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Existing federated learning systems face challenges in managing device mobility during training, leading to increased training times and reduced accuracy due to the need for restarting computations when devices move between edge servers.

Innovation Solution

A mechanism for migrating partially trained models from an origin edge server to a destination edge server, allowing computational devices to resume training without restarting, thereby reducing training time and maintaining model accuracy.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Adaptability or versatility

If devices move between edge servers during federated learning training, then device mobility is supported, but training time increases and accuracy decreases due to computation restart

Engineering Contradiction:
Improvedevice mobility supportVSAvoidtraining time
Core Design Contradiction:
Adaptability or versatilityVSLoss of time

Solution Approach 1:

The system performs preliminary actions by creating data checkpoints during the training process at the origin edge server. These checkpoints capture the state of model training at regular intervals, allowing the device to resume from the latest checkpoint rather than restarting from scratch when moving to a new edge server. This preliminary preparation of checkpoint data eliminates the need for full training restart.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The invention uses copying by transferring data checkpoints from the origin edge server to the destination edge server. Instead of transferring the entire training process or raw data, the system creates and copies checkpoint snapshots that represent the training state, enabling efficient resumption of training at the new server without repeating previous training work.

Inventive Principle:
Principle #26Copying

2Adaptability or versatility

If devices move between edge servers during federated learning training, then device mobility is supported, but model accuracy decreases due to computation restart

Engineering Contradiction:
Improvedevice mobility supportVSAvoidmodel accuracy
Core Design Contradiction:
Adaptability or versatilityVSManufacturing precision

Solution Approach 1:

The system performs preliminary actions by creating data checkpoints during the training process at the origin edge server. These checkpoints capture the state of model training at regular intervals, allowing the device to resume from the latest checkpoint rather than restarting from scratch when moving to a new edge server. This preliminary preparation of checkpoint data eliminates the need for full training restart.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The invention uses copying by transferring data checkpoints from the origin edge server to the destination edge server. Instead of transferring the entire training process or raw data, the system creates and copies checkpoint snapshots that represent the training state, enabling efficient resumption of training at the new server without repeating previous training work.

Inventive Principle:
Principle #26Copying

3Reliability

If full training restart is performed when devices move between edge servers, then training consistency is maintained, but training time increases significantly

Engineering Contradiction:
Improvetraining consistencyVSAvoidtraining efficiency
Core Design Contradiction:
ReliabilityVSProductivity

Solution Approach 1:

The system performs preliminary actions by creating data checkpoints during the training process at the origin edge server. These checkpoints capture the state of model training at regular intervals, allowing the device to resume from the latest checkpoint rather than restarting from scratch when moving to a new edge server. This preliminary preparation of checkpoint data eliminates the need for full training restart.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The invention uses copying by transferring data checkpoints from the origin edge server to the destination edge server. Instead of transferring the entire training process or raw data, the system creates and copies checkpoint snapshots that represent the training state, enabling efficient resumption of training at the new server without repeating previous training work.

Inventive Principle:
Principle #26Copying

4Loss of time

If data checkpoints are transferred between edge servers, then training resumption is enabled, but network overhead increases

Engineering Contradiction:
Improvetraining timeVSAvoidnetwork overhead
Core Design Contradiction:
Loss of timeVSLoss of energy

Solution Approach 1:

The system extracts only the essential checkpoint data needed for training resumption rather than transferring complete training datasets or full model states. By taking out only the critical checkpoint information (model parameters, optimization state, and metadata), the system minimizes network transfer overhead while enabling effective training continuation.

Inventive Principle:
Principle #2Taking out (Extraction)

Solution Approach 2:

The invention uses copying by transferring data checkpoints from the origin edge server to the destination edge server. Instead of transferring the entire training process or raw data, the system creates and copies checkpoint snapshots that represent the training state, enabling efficient resumption of training at the new server without repeating previous training work.

Inventive Principle:
Principle #26Copying

Data Source

PatentUS12493789B2Cooperative training migration
Publication Date: 2025.12.09 RAKUTEN MOBILE INC
  • US12493789B2 patent drawing
  • US12493789B2 patent drawing
  • US12493789B2 patent drawing

AI summary

Cooperative training migration is performed by training, cooperatively with a computational device through a network, the neural network model, creating, during the iterations of training, a data checkpoint, the data checkpoint including the gradient values and the weight values of the server partition, the loss value, and an optimizer state, receiving, during the iterations of training, a migration notice, the migration notice including an identifier of a second edge server, and transferring, during the iterations of training, the data checkpoint to the second edge server.