Adversarial momentum-contrastive pre-training

Cong Xu,Dan Li,Min Yang
DOI: https://doi.org/10.1016/j.patrec.2022.07.005
IF: 4.757
2022-08-01
Pattern Recognition Letters
Abstract:Recently proposed adversarial self-supervised learning methods usually require big batches and long training epochs to extract robust features, which will bring heavy computational overhead on platforms with limited resources. In order to help the network learn more powerful feature representations in smaller batches and fewer epochs, this paper proposes a novel adversarial momentum contrastive learning method, which introduces two memory banks corresponding to clean samples and adversarial samples, respectively. These memory banks can be dynamically incorporated into the training process to track invariant features among historical mini-batches. Compared with the previous adversarial pre-training model, our method achieves superior performance with smaller batch size and less training epochs. In addition, the model outperforms some state-of-the-art supervised defensive methods on multiple benchmark datasets after being fine-tuned on downstream classification tasks.
computer science, artificial intelligence
What problem does this paper attempt to address?