BUILDING GENERALIZED MACHINE LEARNING MODELS FROM MACHINE LEARNING MODEL EXPLANATIONS
A system includes a memory and a processing device, operatively coupled to the memory, to receive, from a client device via a user interface, input data comprising an initial version of a machine learning model, initialize an operating mode of the user interface for machine learning model building, and generate an enhanced version of the machine learning model in accordance with the operating mode.
1 . A system comprising:
a memory; and
a processing device, operatively coupled to the memory, to:
receive, from a client device via a user interface, input data comprising an initial version of a machine learning model;
initialize an operating mode of the user interface for machine learning model building; and
generate an enhanced version of the machine learning model in accordance with the operating mode.
2 . The system of claim 1 , wherein the enhanced version of the machine learning model is generated based on an explanation indicative of feature importance.
3 . The system of claim 2 , wherein the explanation is a local interpretable model-agnostic explanation (LIME)-based explanation.
4 . The system of claim 2 , wherein, to generate the enhanced version of the machine learning model, the processing device is further to:
send the explanation to the client device;
receive, from the client device, user input relating to the explanation, wherein the user input comprises a set of ground truth features; and
generate the enhanced version of the machine learning model based on the user input relating to the explanation.
5 . The system of claim 1 , wherein, to generate the enhanced version of the machine learning model, the processing device is further to implement incremental learning.
6 . The system of claim 5 , wherein the incremental learning is regularization-based elastic weight consolidation (EWC) incremental learning.
7 . The system of claim 1 , wherein the processing device is further to:
generate a first evaluation of the initial version of the machine learning model; and
evaluate the enhanced version of the machine learning model by comparing the first evaluation to a second evaluation of the enhanced version of the machine learning model.
8 . A method comprising:
receiving, by at least one processing device from a client device via a user interface, input data comprising an initial version of a machine learning model;
initializing, by the at least one processing, an operating mode of the user interface for machine learning model building; and
generating, by the at least one processing device based on the input data, an enhanced version of the machine learning model in accordance with the operating mode.
9 . The method of claim 8 , wherein the enhanced version of the machine learning model is generated based on an explanation indicative of feature importance.
10 . The method of claim 9 , wherein the explanation is a local interpretable model-agnostic explanation (LIME)-based explanation.
11 . The method of claim 9 , wherein generating the enhanced version of the machine learning model further comprises:
sending the explanation to the client device;
receiving, from the client device, user input relating to the explanation, wherein the user input comprises a set of ground truth features; and
generate the enhanced version of the machine learning model based on the user input relating to the explanation.
12 . The method of claim 8 , wherein generating the enhanced version of the machine learning model comprises implementing incremental learning.
13 . The method of claim 12 , wherein the incremental learning is regularization-based elastic weight consolidation (EWC) incremental learning.
14 . The method of claim 8 , further comprising:
generating, by the at least one processing device, a first evaluation of the initial version of the machine learning model; and
evaluating, by the at least one processing device, the enhanced version of the machine learning model by comparing the first evaluation to a second evaluation of the enhanced version of the machine learning model.
15 . A non-transitory computer-readable storage medium comprising instructions that, when executed by a processing device, cause the processing device to:
receive, from a client device via a user interface, input data comprising an initial version of a machine learning model;
initialize an operating mode of the user interface for machine learning model building; and
generate an enhanced version of the machine learning model in accordance with the operating mode.
16 . The non-transitory computer-readable storage medium of claim 15 , wherein the enhanced version of the machine learning model is generated based on an explanation indicative of feature importance.
17 . The non-transitory computer-readable storage medium of claim 16 , wherein, to generate the enhanced version of the machine learning model, the processing device is further to:
send the explanation to the client device;
receive, from the client device, user input relating to the explanation, wherein the user input comprises a set of ground truth features; and
generate the enhanced version of the machine learning model based on the user input relating to the explanation.
18 . The non-transitory computer-readable storage medium of claim 15 , wherein, to generate the enhanced version of the machine learning model, the processing device is to implement incremental learning.
19 . The non-transitory computer-readable storage medium of claim 18 , wherein the incremental learning is regularization-based elastic weight consolidation (EWC) incremental learning.
20 . The non-transitory computer-readable storage medium of claim 15 , further comprising instructions that, when executed by the processing device, cause the processing device to:
generate a first evaluation of the initial version of the machine learning model; and
evaluate the enhanced version of the machine learning model by comparing the first evaluation to a second evaluation of the enhanced version of the machine learning model.