Data augmentation based on attention

An electronic device, comprising a computer processor coupled to a computer-readable memory unit, the memory unit comprising instructions that when executed by the computer processor implements a method for data augmentation, comprising:

• generating mixed data from first data and second data, the mixed data comprising a first portion from the first data and a second portion from the second data;
• dividing the mixed data into a plurality of data blocks based on a mask describing a boundary between the first and second portions in the mixed data, wherein dividing the mixed data further comprises dividing the mixed data based on any of:
• a size of any of the first and second data, and the mixed data;
• a size of any of the first and second portions;
• a predetermined size;
• and
• a predetermined number;
• obtaining an attention map for the plurality of data blocks in the mixed data based on distributions of the first and second portions in the mixed data, the attention map describing contributions of the first and second data to the mixed data;
• and
• determining a label for the mixed data based on a first label for the first data, a second label for the second data, and the attention map.

A computer program product, the computer program product comprising a non-transitory computer readable storage medium having program instructions embodied therewith, the program instructions executable by an electronic device to cause the electronic device to perform a method for data augmentation, the method comprising:

• generating mixed data from first data and second data, the mixed data comprising a first portion from the first data and a second portion from the second data;
• dividing the mixed data into a plurality of data blocks based on a mask describing a boundary between the first and second portions in the mixed data, wherein dividing the mixed data further comprises dividing the mixed data based on any of:
• a size of any of the first and second data, and the mixed data;
• a size of any of the first and second portions;
• a predetermined size;
• and
• a predetermined number;
• obtaining an attention map for the plurality of data blocks in the mixed data based on distributions of the first and second portions in the mixed data, the attention map describing contributions of the first and second data to the mixed data;
• and
• determining a label for the mixed data based on a first label for the first data, a second label for the second data, and the attention map.