Skip to content

Search algorithms

AutoAlbument uses the following algorithms to search for augmentation policies.

Faster AutoAugment

"Faster AutoAugment: Learning Augmentation Strategies using Backpropagation" by Ryuichiro Hataya, Jan Zdenek, Kazuki Yoshizoe, and Hideki Nakayama. Paper | Original implementation