Figure 4: Type #1 of data augmentation consists of dataset generation/dataset expansion. This is a less common form of data augmentation. The first type of data augmentation is what I call dataset generation or dataset expansion.. As you know machine learning models, and especially neural networks, can require quite a bit of training data — but what if you don't have very much training.