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

Claim 10:
11. A computer program product for augmenting a dataset, used by a data augmentation application, to train a neural network, the computer program product comprising:
a computer readable storage medium having program instructions embodied therewith, the program instructions executable by a processing unit, as part of a data augmentation application, to cause the processing unit to:
obtain 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;
cluster the input data to a plurality of clusters;
obtain, for each of the plurality of clusters, a second feature vector based on the first model;
estimate, for each cluster of the plurality of clusters, a similarity between the first feature vector and the second feature vector;
determine at least one cluster of the plurality of clusters for which the similarity is lower than a similarity threshold; and
perform data augmentation to the at least one cluster to train a neural network, wherein the program instructions executable by a processing unit to further cause the processing unit to:
for each of the at least one cluster,
perform a plurality of augmentation methods on the cluster to obtain a plurality of augmented clusters;
for each cluster of the plurality of augmented clusters,
obtain a third feature vector for the augmented cluster based on the first model;
calculate a loss based on the first feature vector and the third feature vector;
select one of the plurality of augmented clusters for which the loss is the lowest, as an optimal augmented cluster; and
combine the input data with the respective optimal augmented clusters to train a second model.