VQ-VAEs: Neural Discrete Representation Learning | Paper + PyTorch Code Explained

Sdílet
Vložit
  • čas přidán 29. 08. 2024

Komentáře • 77

  • @DCentFN
    @DCentFN Před 3 měsíci +1

    Combining the code with the paper explanation helps the understanding immensely. Allows for concept as well as application. Thank you

  • @pawnagon4874
    @pawnagon4874 Před 3 lety +18

    I wish I had one of these videos for every paper I read, awesome work

  • @user-hv2xy2zt1k
    @user-hv2xy2zt1k Před 3 lety +28

    Great explanation! Especially useful explanation of the code! Please keeping doing the code part! You are a life saver!

    • @TheAIEpiphany
      @TheAIEpiphany  Před 3 lety

      Super valuable thanks! I'll consider maybe doing a walk-through of some code feel free to suggest something!

    • @iceinmylean3947
      @iceinmylean3947 Před 2 lety

      @@TheAIEpiphany Id be really interested in anything related to the autoregressive model part also mentioned in this video, maybe something like training a transformer?

  • @user-sz1hf9rv1u
    @user-sz1hf9rv1u Před 2 lety +7

    I truly appreciate your explanations, especially PyTorch implementation part, which reduce the gap between concepts and real world implementations. Finding this channel is like finding treasures to me, I've recommended this channel to all my friends. Look forward to your weekly update, thanks :)

    • @TheAIEpiphany
      @TheAIEpiphany  Před 2 lety +2

      Thanks man! 🙏 Yup I am getting back on track with CZcams I had a weird period over the last month. 😄

  • @mehmetonur7925
    @mehmetonur7925 Před 3 lety +6

    Code part is pretty good.It has made paper more clear.

  • @ramensusho
    @ramensusho Před dnem +1

    This was some nice explanation

  • @artikeshari5441
    @artikeshari5441 Před 2 lety +3

    Thanks for explaining it very clearly. Code explanation makes the concept more robust.

  • @yimingqu2403
    @yimingqu2403 Před 3 lety +7

    Appreciate your work! Both paper and code parts are very helpful.
    Two suggestion to make the code more concise
    - pytorch has built in function to calculate pairwise distance `torch.cdist`.
    - directly using `index_select` to get the quantized matrix may be more convenient.

    • @TheAIEpiphany
      @TheAIEpiphany  Před 3 lety

      Not my implementation - I agree why not reuse the existing library code when possible

    • @kyde8392
      @kyde8392 Před 2 lety

      Your suggestions are really neat 👌

  • @ShravanKumar147
    @ShravanKumar147 Před rokem +2

    Thank you for such a great explanation, adding code into this format is really helpful to digest the concepts more intuitively. Please keep them coming the same way.

  • @AmirHosseinAlamdar
    @AmirHosseinAlamdar Před 16 dny

    Great! Also the code section was a very good idea. Saved a lot of time thanks.

  • @christiannowak7094
    @christiannowak7094 Před 3 lety +1

    Brilliant, never got so close to understand what's going on. Really well done

  • @jasdeepsinghgrover2470
    @jasdeepsinghgrover2470 Před 2 lety +2

    Thanks for the amazing video... You can make them longer and more detailed if needed... Really fun to watch

  • @elbayo421
    @elbayo421 Před 7 měsíci

    Amazing explanation! Thank you very much. I was a bit troubled about understanding how this model can be used to generate new images but after reading around I think I get it now

  • @skymanaditya
    @skymanaditya Před 2 lety +1

    Loved the explanation, especially the part where you covered all the important aspects and showed them in the code. Subscribed and looking forward to more of this content!

  • @hoanglinh96nl
    @hoanglinh96nl Před 8 měsíci

    00:01 VQ-VAE is a crucial model for AI research and used in various novel works.
    02:02 Variational autoencoders use a stochastic bottleneck layer.
    05:58 VQ-VAEs impose structure into the latent space for continuous and meaningful interpolation.
    07:50 Discrete representations are a natural fit for many modalities and enable complex reasoning and predictive learning.
    11:40 Using l2 norm to find closest vector and approximate posterior
    13:33 The likelihood assumption and the loss terms in VQ-VAEs
    17:18 Conversion of bchw tensor to standard representation
    18:59 VQ-VAEs use flat input and an embedding table to find distance to codebook vectors.
    22:27 Explanation of implementing straight through gradient
    24:09 The approximate posterior z given x is a deterministic function.
    27:37 The model is an autoregressive token predictor for generating novel images.
    29:11 VQ-VAEs compress data to a discrete space with code size k=512.
    32:24 VQ-VAE v2 has hierarchical structure for better reconstructions
    34:04 VQ-VAEs capture high-resolution images with some distortion
    Crafted by Merlin AI.

  • @vladimirtchuiev2218
    @vladimirtchuiev2218 Před rokem

    Finally a good explanation on how the autoregressive prior part works :X

  • @stefanmai9879
    @stefanmai9879 Před rokem

    You're a great teacher! Glad you came back to this paper and love the format with the code walkthroughs. Very thorough!

  • @letianwang5141
    @letianwang5141 Před 10 měsíci +1

    best explanation ever, unbiased comment

  • @michelspeiser5789
    @michelspeiser5789 Před 5 měsíci +1

    Instead of argmin on the distance to the closest embedding, couldn't we just use a softmax instead?

  • @gougenot
    @gougenot Před rokem

    Very nice code part. Truly helped me to understand, what is happening

  • @mehdizahedi2810
    @mehdizahedi2810 Před 28 dny

    awesome presentation, thanks.

  • @Prashantserai
    @Prashantserai Před 11 měsíci

    Fantastic in every way, including the code explanation as well!

  • @alexijohansen
    @alexijohansen Před 2 lety

    Thank you. Love the code, love the in depth explanation! Explaining the math is also great for a beginner like me.

  • @sarathmohan3143
    @sarathmohan3143 Před 2 lety

    Thank a lot sir.
    Simple and concise explanation by covering the related basics also.

  • @amonkotaro1723
    @amonkotaro1723 Před rokem +1

    the distance looks like (a - b)^2 19:30

  • @evgenydyshlyuk5604
    @evgenydyshlyuk5604 Před 3 lety +1

    Great choice of the article, thank you, was very interesting!

  • @MrMIB983
    @MrMIB983 Před 3 lety +3

    Great, we also need VQ-GAN, TransGAN and GANsformer

    • @TheAIEpiphany
      @TheAIEpiphany  Před 3 lety +2

      VQ-GAN coming soon as well as DALL-E. I'll add the other 2 to my list. 😂 Thanks!

    • @varunsai9736
      @varunsai9736 Před 3 lety +2

      Can you also do clip+ vqgan

    • @TheAIEpiphany
      @TheAIEpiphany  Před 3 lety

      @@varunsai9736 Sure I'll see whether I can cram it into VQGAN video

  • @user-co6pu8zv3v
    @user-co6pu8zv3v Před 2 lety +1

    Great explanation!!! Thank you!

  • @user-tt7mp4dk9w
    @user-tt7mp4dk9w Před rokem

    One thing that confused me is -> why do they convert BCHW to BHWC and then combine BHW x C => (16K, 64)? Should the quantization be done per image in the batch? It seems the entire batch is merged and quantized instead.

  • @IgorAherne
    @IgorAherne Před 10 měsíci

    @TheAIEpiphany Man, that's such an epic explanation. Thank you so much for your help!
    One thing that I am struggling with, is 28:00 - by tweaking the prior, does that mean that we can trick the model about what "was" in the image? (what is is expected).
    The concept of predicting the next token is easy for me, - but what are we predicting? a next discrete-embedding vector from the table? But these vectors weren't guaranteed to be in any order...
    Or are we predicting the next word? In that case, how do we associate word token to the discrete-embedding vector?
    During teaching this autoregressive model, how do we know which one is the target/correct vector, that we want to be predicted?

    • @IgorAherne
      @IgorAherne Před 10 měsíci

      If anyone else has this question, the autoregressive model is an addition which doesn't "improve" the quality of the VQ-VAE.
      But, we can swap it instead of the encoder+codebook, and use it to produce new images. So basically, "autoregressiveModel+decoder".
      You have to remember that once VQ-VAE is learned, the codebook vectors will be frozen forever. They will not be shuffled etc.
      So, when deployed into production, the Autoregressive model doesn't care what encoder does.
      Instead, the autoregressive model has learnt to look at the few code-book indices (we pick them arbitrarily), and to generate remaining indices of codebook that it thinks will be relevant.
      For example, if we gave it and index describing sky, it might decide that a following index describing a cloud will be more likely, than, say, of a fish.
      Once the autoregressive model produced all the needed indices, we feed the chosen codebook-vectors into the decoder.
      This allows us to generate images.

  • @MuhammadAli-mi5gg
    @MuhammadAli-mi5gg Před 2 lety

    Thanks a lot, it was an awesome explanation.
    And yes the code part is necessary as far as I think, and would highly recommend that.
    Moreover, it would be great if you can also make some content regarding these distributions, because I have tried to understand them, but still, they sound quite fuzzy to me.
    Thanks again!

  • @kirtipandya4618
    @kirtipandya4618 Před 2 lety +1

    Nice video. Please do more videos like this. 👍🏻

  • @TuanNguyen-su5ty
    @TuanNguyen-su5ty Před 7 měsíci

    This video is invaluable. Thank you

  • @sahilgoyal3811
    @sahilgoyal3811 Před 8 měsíci

    much helpful!

  • @yinghaohu8784
    @yinghaohu8784 Před 5 měsíci

    You mentioned posterior and prior, can you provide some reference, why they model it in this way ?

  • @user-my6yf1st8z
    @user-my6yf1st8z Před 3 lety +1

    THANK YOU BROTHER AMAZING

  • @djabort
    @djabort Před 2 lety

    thank you a lot. i like the format with code

  • @bdennyw1
    @bdennyw1 Před 2 lety

    Love the Pytorch code!

  • @hassenzaayra5419
    @hassenzaayra5419 Před rokem

    thank you very much for this explanation.
    I would like to know how the creation of the codebook is going

  • @johnpope1473
    @johnpope1473 Před 3 lety +1

    You’re smashing it. Take some pauses. Pacing conveys a lot / gives space to digest content. Consider you want to cause people to have a light bulb moment. You can’t give people the answer so quickly. I’m looking forward to pytorch stuff. Maybe do some meditation before you record / stillness. Pause.

    • @TheAIEpiphany
      @TheAIEpiphany  Před 3 lety

      Thanks for the feedback! I agree I need to work on me being less hectic haha I guess.

  • @modyngs1256
    @modyngs1256 Před 8 měsíci

    Hi,
    What's the application you are using to write on the PDF? i mean the way you write something in side with the original pdf in the black side of the pdf?

  • @user-gz5ym6lb4l
    @user-gz5ym6lb4l Před rokem

    Thanks for your amazing&simple explanation. It realy helpful.
    In some paper based on VQVAE, they use perplexity for measurement. But i can not understand what perplexity means in VQVAE model. So if you are not busy can i request explain 'what perplexity means in VQVAE?'
    Thanks again for your wonderful explain!

  • @drtristanbehrens
    @drtristanbehrens Před 2 lety

    A great video! Thanks for sharing!

  • @KarimaKadaoui
    @KarimaKadaoui Před 2 lety +1

    Thank you so much for the explanation! I wanted to ask how you get to understand some of the details that are not mentioned in the paper, like how the KL Div ends up being equal to log K?

    • @TheAIEpiphany
      @TheAIEpiphany  Před 2 lety

      🙏 Well, analyzing these I bring in my understanding and background from elsewhere to better understand what is going on in this particular paper.

  • @abdelrahmanwaelhelaly1871

    Thank you

  • @apollozou9809
    @apollozou9809 Před 2 lety

    Overall great explanation. One thing I find confused though. In the paper, loss2 and loss3 are something between Codebook(embeddings vector) and the encoding after CNN. However, in the code, it is something between Quantized encoding after CNN and the encoding after CNN. Can you explain why they are the same thing?

  • @redone9553
    @redone9553 Před 2 lety +1

    Code is nice

  • @kirtipandya4618
    @kirtipandya4618 Před 2 lety +1

    Which software are you using for paper review? One side paper and you can draw and put code next to it.

  • @mathkernel5136
    @mathkernel5136 Před rokem

    How do we generate new images from the VQ-VAE model. Can you do a tutorial on the pix2pix model for generating new image samples? Thanks

  • @srinathtangudu4899
    @srinathtangudu4899 Před rokem

    awesome

  • @zongtaowang7840
    @zongtaowang7840 Před 2 lety

    thank you for your explaining and code. When run the code there is an ERROR: Could not open requirements file: [Errno 2] No such file or directory: 'requirements.txt'. it seems there is no requirement.txt file there

  • @peterkonig9537
    @peterkonig9537 Před 2 lety +1

    cool video

  • @deep.extrospection
    @deep.extrospection Před 2 lety +1

    Very good explanation. And with an implementation to support it.
    Thanks a lot!

  • @sarvagyagupta1744
    @sarvagyagupta1744 Před 3 lety

    This is a good explanation of VQVAE. I do have a question though. OpenAI's Jukebox is based on VQVAE and they pass gradients through the latent space in their loss function. So is there any difference or what do you think is going on?

  • @hernanperez8427
    @hernanperez8427 Před 2 lety

    thanks!! it really likes me, very usefull!

  • @terryr9052
    @terryr9052 Před 2 lety

    I have been thinking about the VQ-VAE for generating music and it seems to me that one large limitation of quantizing your latent vectors is that you lose the ability to see interesting results that lay between clusters of latent vectors. For example, I train my model on both reggae and death metal songs and the resulting latent space shows two clusters. It would be nice to then hear songs that interpolate between the 2 clusters but it seems that the quantizing step will force any new vectors (our desired hybrid) to adopt the established codebook vectors which are only representative of the "pure" songs. Am I correct in this line of thinking? Has anyone seen any more info on this at all?

  • @manuobelleiro7711
    @manuobelleiro7711 Před 2 lety

    Hello, great video! I had a question regarding the token prediction training. Can this be used to generate images from a text description? If so, where in the code is this implemented? I'm having trouble understanding this last part

  • @eranjitkumar11
    @eranjitkumar11 Před 2 lety

    Hi, thank you for your work. Can you explain how they incorporate pixelcnn (or wavenet)?

  • @djaym7
    @djaym7 Před 2 lety

    +1 on code part

  • @razvanrotaru2285
    @razvanrotaru2285 Před 2 lety

    i love you