Patent ID: 11947570
Assignee: INTERNATIONAL BUSINESS MACHINES CORPORATION
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 5:
6. A system for augmenting a dataset, used by a data augmentation application, to train a neural network, the system comprising:
one or more processing units;
a memory coupled to at least one of the processing units; and
a set of computer program instructions stored in the memory as part of a data augmentation application and executed by at least one of the processing units in order to perform actions of:
obtaining a first feature vector for input data based on a first model, wherein the first model comprises one or more filters capable of detecting specific objects comprising at least one of eyes, ears, feathers, fur, or hair;
clustering the input data to a plurality of clusters;
obtaining, for each of the plurality of clusters, a second feature vector based on the first model;
estimating, for each cluster of the plurality of clusters, a similarity between the first feature vector and the second feature vector;
determining at least one cluster of the plurality of clusters for which the similarity is lower than a similarity threshold; and
performing data augmentation to the at least one cluster to train a neural network, wherein the actions further comprise:
for each cluster of the at least one cluster,
performing a plurality of augmentation methods on the cluster to obtain a plurality of augmented clusters;
for each cluster of the plurality of augmented clusters,
obtaining a third feature vector for the augmented cluster based on the first model;
calculating a loss based on the first feature vector and the third feature vector;
selecting one of the plurality of augmented clusters for which the loss is the lowest, as an optimal augmented cluster; and
combining the input data with the respective optimal augmented clusters to train a second model.