I spent the weekend reading this super interesting paper "An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale", and made this video to explain it a bit. I hope it can be helpful for those who're also interested in how the transformer architecture (de-facto model in NLP) can be applied in computer vision.
Thanks for this super detailed video! I've seen some of your others in the past and always found them helpful. Yannick's video on this paper is great as well.
It's still an interesting paper, but I was disappointed they were "just" concatenating an imagine generator with a language model. I'm really excited for when someone figures out concurrently trained models, say, alternating between training passes of GPT-3 and iGPT, such that the very same attention layers deal with both language and and visual/spatial conceptualization. I expect common sense reasoning capabilities to take a huge leap, at that point.
11 comments
[ 3.1 ms ] story [ 40.4 ms ] threadIf anyone is looking for a written walkthrough, I did a quick summary/explainer on my personal site: https://whoisnnamdi.com/transformers-image-recognition/
I wish someone would code this model and walk through it along with the training and inference procedures.
1. Transformer 2. How to apply the Transformer to image data
As for the first one, I made a video to go through Transformer architecture, if you're interested https://youtu.be/ELTGIye424E
For the second one, what you need to code except for the Transformer is a pre-processing pipeline to break an image into a sequence of patches.