Doktorandské kolokvium KAI - Štefan Pócoš (4.12.2023)
v pondelok 4.12.2023 o 13:10 hod. v miestnosti I/9
Od: Damas Gruska
Prednášajúci: Štefan Pócoš
Názov: RecViT: Enhancing Vision Transformer with Top-Down Information Flow
Termín: 4.12.2023, 13:10 hod., I/9
We propose and analyse a novel neural network architecture — recurrent vision transformer (RecViT). Building upon the popular vision transformer (ViT), we add a biologically inspired top-down connection, letting the network ‘reconsider’ its initial prediction. Moreover, using a recurrent connection creates space for feeding multiple similar, yet slightly modified or augmented inputs into the network, in a single forward pass. As it has been shown that a top-down connection can increase accuracy in case of convolutional networks, we analyse our architecture, combined with multiple training strategies, in the adversarial examples (AEs) scenario. Our results show that some versions of RecViT may indeed be more robust than the baseline ViT. We also leverage the fact that transformer networks have certain level of inherent explainability. By visualising attention maps of various input images, we gain some insight into the inner workings of our network.