ASYNCHRONOUS EDGE-CLOUD MACHINE LEARNING MODEL MANAGEMENT WITH UNSUPERVISED DRIFT DETECTION
Techniques described herein relate to a method for updating ML models based on drift detection. The method may include training a ML model; storing the trained ML model associated with a confidence threshold and a fresh indication; receiving a drift signal from an edge node; making a determination, that drift is detected for the ML model; updating the trained ML model in the shared communication layer to be associated with a drifted indication; receiving batch data from edge nodes in response to the updating; generating an updated data set comprising previous data and the batch data; training the ML model using the updated data set; updating the trained ML model in the shared communication layer to be associated with an outdated indication; and storing, by the model coordinator, the updated trained ML model in the shared communication layer associated with a confidence threshold and a fresh indication.
1 . A method for updating machine learning (ML) models based on drift detection, the method comprising:
training, by a model coordinator, a ML model using a historical data set to obtain a trained ML model;
storing, by the model coordinator, the trained ML model in a shared communication layer associated with a first confidence threshold and a first fresh indication;
receiving, by the model coordinator, a drift signal from an edge node of a plurality of edge nodes executing the trained ML model;
making a determination, by the model coordinator and based on receiving the drift signal, that drift is detected for the trained ML model;
updating, by the model coordinator, the trained ML model in the shared communication layer to be associated with a drifted indication;
receiving, by the model coordinator, batch data from the plurality of edge nodes in response to the updating;
generating, by the model coordinator, an updated historical data set comprising at least a portion of the historical data set and the batch data;
training the ML model using the updated historical data set to obtain an updated trained ML model;
updating, by the model coordinator, the trained ML model in the shared communication layer to be associated with an outdated indication; and
storing, by the model coordinator, the updated trained ML model in the shared communication layer associated with a second confidence threshold and a second fresh indication.
2 . The method of claim 1 , wherein the edge node sends the drift signal based on the determination that an edge node confidence value associated with execution of the trained ML model is lower than the first confidence threshold.
3 . The method of claim 1 , wherein the batch data is provided from the plurality of edge nodes to the model coordinator using the shared communication layer.
4 . The method of claim 1 , wherein, based on the outdated indication being associated with the trained ML model, the plurality of edge nodes obtain and execute the updated trained ML model.
5 . The method of claim 1 , wherein the determination based on receiving the drift signal is made after receiving a plurality of other drift signals, and wherein the drift signal and the plurality of other drift signals are a quantity equal to a minimum threshold of drift signals required for drift detection.
6 . The method of claim 1 , further comprising, before generating the updated historical data set, making a second determination, by the model coordinator, that a required amount of batch data has been received from the plurality of edge nodes.
7 . The method of claim 1 , wherein:
the first confidence threshold is a percentage of a confidence value associated with the trained ML model, and
the confidence value is obtained by using an arbitrary statistic of values of a softmax layer associated with the trained ML model.
8 . A non-transitory computer readable medium comprising computer readable program code, which when executed by a computer processor enables the computer processor to perform a method for updating machine learning (ML) models based on drift detection, the method comprising:
training, by a model coordinator, a ML model using a historical data set to obtain a trained ML model;
storing, by the model coordinator, the trained ML model in a shared communication layer associated with a first confidence threshold and a first fresh indication;
receiving, by the model coordinator, a drift signal from an edge node of a plurality of edge nodes executing the trained ML model;
making a determination, by the model coordinator and based on receiving the drift signal, that drift is detected for the trained ML model;
updating, by the model coordinator, the trained ML model in the shared communication layer to be associated with a drifted indication;
receiving, by the model coordinator, batch data from the plurality of edge nodes in response to the updating;
generating, by the model coordinator, an updated historical data set comprising at least a portion of the historical data set and the batch data;
training the ML model using the updated historical data set to obtain an updated trained ML model;
updating, by the model coordinator, the trained ML model in the shared communication layer to be associated with an outdated indication; and
storing, by the model coordinator, the updated trained ML model in the shared communication layer associated with a second confidence threshold and a second fresh indication.
9 . The non-transitory computer readable medium of claim 8 , wherein the edge node sends the drift signal based on the determination that an edge node confidence value associated with execution of the trained ML model is lower than the first confidence threshold.
10 . The non-transitory computer readable medium of claim 8 , wherein the batch data is provided from the plurality of edge nodes to the model coordinator using the shared communication layer.
11 . The non-transitory computer readable medium of claim 8 , wherein, based on the outdated indication being associated with the trained ML model, the plurality of edge nodes obtain and execute the updated trained ML model.
12 . The non-transitory computer readable medium of claim 8 , wherein the determination based on receiving the drift signal is made after receiving a plurality of other drift signals, and wherein the drift signal and the plurality of other drift signals are a quantity equal to a minimum threshold of drift signals required for drift detection.
13 . The non-transitory computer readable medium of claim 8 , wherein the method performed by executing the computer readable program code further comprises, before generating the updated historical data set, making a second determination, by the model coordinator, that a required amount of batch data has been received from the plurality of edge nodes.
14 . The non-transitory computer readable medium of claim 8 , wherein the first confidence threshold is a fraction of a confidence value associated with the trained ML model.
15 . A system for updating machine learning (ML) models based on drift detection, the system comprising:
a model coordinator, executing on a processor comprising circuitry, operatively connected to a shared communication layer and a plurality of edge nodes, and configured to:
train a ML model using a historical data set to obtain a trained ML model;
store the trained ML model in the shared communication layer associated with a first confidence threshold and a first fresh indication;
receive a drift signal from an edge node of the plurality of edge nodes executing the trained ML model;
make a determination, based on receiving the drift signal, that drift is detected for the trained ML model;
update the trained ML model in the shared communication layer to be associated with a drifted indication;
receive batch data from the plurality of edge nodes in response to the updating;
generate an updated historical data set comprising at least a portion of the historical data set and the batch data;
train the ML model using the updated historical data set to obtain an updated trained ML model;
update the trained ML model in the shared communication layer to be associated with an outdated indication; and
store the updated trained ML model in the shared communication layer associated with a second confidence threshold and a second fresh indication.
16 . The system of claim 15 , wherein the edge node sends the drift signal based on the determination that an edge node confidence value associated with execution of the trained ML model is lower than the first confidence threshold.
17 . The system of claim 15 , wherein the batch data is provided from the plurality of edge nodes to the model coordinator using the shared communication layer.
18 . The system of claim 15 , wherein, based on the outdated indication being associated with the trained ML model, the plurality of edge nodes obtain and execute the updated trained ML model.
19 . The system of claim 15 , wherein the determination based on receiving the drift signal is made after receiving a plurality of other drift signals, and wherein the drift signal and the plurality of other drift signals are a quantity equal to a minimum threshold of drift signals required for drift detection.
20 . The system of claim 15 , wherein, before generating the updated historical data set, the model coordinator is further configured to make a second determination that a required amount of batch data has been received from the plurality of edge nodes.