Need more varied training data without collecting more images? Data augmentation creates modified examples, such as noisy or rotated images, to improve robustness and reduce overfitting. section 9.1
A practical image-classification augmentation setup can:
- Add random noise to existing training images. section 9.1
- Apply random rotations to increase dataset diversity. section 9.1
- Use JAX's functional random-number keys rather than hidden mutable generator state. section 9.0
Noisy and rotated images
Imagine training an image classifier. Randomly adding noise and rotating each image produces varied training examples, helping the network become more robust while acting as implicit regularization. section 9.1
Figure 9.1 Scheme of the data augmentation process
Use the process diagram to visualize how original images become varied training examples. section 9.1








