Visa Inc.
Method, System, and Computer Program Product for Multi-Task Learning in Deep Neural Networks

Abstract:

Provided are methods for multi-task learning (MTL) in deep neural networks. An exemplary method may include receiving an MTL model; receiving a testing data set comprising testing data items for the MTL model, each testing data item comprising a plurality of elements, each element associated with a respective feature; grouping the features into a plurality of groups based on an impact of each feature on the tasks of the MTL model, determining an overall accuracy score and task-specific accuracy scores based on inputting the testing data to the MTL model; applying feature reduction evaluation (FRE) to provide a feature score for each feature; and adjusting the feature scores based on a respective grouping associated with the respective feature and at least one of the overall accuracy score, the task-specific accuracy scores, or any combination thereof to provide an adjusted feature score. Systems and computer program products are also disclosed.