Diffusion Models | PyTorch Implementation

Поділитися
Вставка
  • Опубліковано 8 лют 2025

КОМЕНТАРІ • 185

  • @outliier
    @outliier  2 роки тому +21

    Link to the code: github.com/dome272/Diffusion-Models-pytorch

    • @bao-dai
      @bao-dai 2 роки тому +2

      21:56 The way you starred your own repo makes my day bro 🤣🤣 really appreciate your work, just keep going!!

    • @outliier
      @outliier  2 роки тому

      @@bao-dai xd

    • @leif1075
      @leif1075 Рік тому

      @@outliier Thanks for sharing but how do you not get bored or tired of doing the same thing for so long and deal with all the math?

    • @outliier
      @outliier  Рік тому

      @@leif1075 I love to do it. I don’t get bored

    • @ananpinya835
      @ananpinya835 Рік тому

      After I saw your next video "Cross Attention | method and math explained", I would like to see ControlNet's openpose in PyTorch Implementation which control posing on image of a dogs. Or if it is too complicate, you may simplify it to control 2 - 3 branches shape of a tree.

  • @javiersolisgarcia
    @javiersolisgarcia Рік тому +9

    This videos is crazy! I don't get tired of recommend it to anyone interesting in diffusion models. I have recently started to research with these type of models and I think your video as huge source of information and guidance in this topic. I find myself recurrently re-watching your video to revise some information. Incredible work, we need more people like you!

    • @outliier
      @outliier  Рік тому +3

      Thank you so much for the kind words!

  • @aladinwunderlampe7478
    @aladinwunderlampe7478 2 роки тому +58

    Hello, this has become a great video once again. We didn't understand much, but it's still nice to watch. Greetings from home say Mam & Dad. ;-))))

  • @AICoffeeBreak
    @AICoffeeBreak 2 роки тому +35

    Great, this video is finally out! Awesome coding explanation! 👏

  • @astrophage381
    @astrophage381 7 місяців тому +2

    These implementation videos are marvelous. You really should do more of them. Big fan of your channel!

  • @terencelee6492
    @terencelee6492 Рік тому +7

    We chose Diffusion Model as part of our course project, and your videos do save much of my time to understand the concepts and have more focus on implementing the main part. I am really grateful for your contribution.

  • @MrScorpianwarrior
    @MrScorpianwarrior 7 місяців тому +1

    Hey! I am start my CompSci Masters program in the Fall, and just wanted to say that I love this video.
    I've never really had time to sit down and learn PyTorch, so the brevity of this video is greatly appreciated! It gives me a fantastic starting point that I can tinker around with, and I have an idea on how I can apply this in a non-conventional way that I haven't seen much research on...
    Thanks again!

    • @outliier
      @outliier  7 місяців тому +1

      Love to hear that
      Good luck on your journey!

  • @Mandollr
    @Mandollr Рік тому +1

    After my midterm week i wanna study diffusion models with your videos im so exited .thanks a lot for good explanation

  • @ethansmith7608
    @ethansmith7608 Рік тому

    this is the most underrated channel i've ever seen, amazing explanation !

  • @yingwei3436
    @yingwei3436 Рік тому +1

    thank you so much for your detailed explaination of the code. It helped me a lot on my way of learning diffusion model. Wish there are more youtubers like you!

  • @mmouz2
    @mmouz2 Рік тому

    Sincere gratitude for this tutorial, this has really helped me with my project. Please continue with such videos.

  • @stevemurch3245
    @stevemurch3245 2 роки тому +2

    Incredible. Very thorough and clear. Very, very well done.

  • @potisseslikitap7605
    @potisseslikitap7605 2 роки тому +3

    This channel seems to be growing very fast. Thanks for this amazing tutorial.🤩

  • @Miurazzo
    @Miurazzo 2 роки тому +11

    Hi, @Outlier , thank you for the awesome explanation !
    Just one observation, I believe in line 50 of your code (at 19:10) it should be:
    uncond_predicted_noise = model(x,t,None)
    😁

    • @outliier
      @outliier  2 роки тому +7

      good catch thank you. (It's correct in the github code tho :))

    • @saltukkezer5100
      @saltukkezer5100 25 днів тому

      Haha, thanks. I also saw the same error and wanted to point it out!

  • @rewixx69420
    @rewixx69420 2 роки тому +1

    I was wating for so long i learnd about condicional difusion models

  • @subtainmalik5182
    @subtainmalik5182 Рік тому

    most informative and easy to understand video on diffusion models on youtube, Thanks Man

  • @FLLCI
    @FLLCI 2 роки тому +1

    This video is really timely and needed. Thanks for the implementation and keep up the good work!

  • @gaggablagblag9997
    @gaggablagblag9997 Рік тому

    Dude, you're amazing! Thanks for uploading this!

  • @haoxu3204
    @haoxu3204 Рік тому

    The best video for diffusion! Very Clear

  • @TheAero
    @TheAero Рік тому

    This is my first few days of trying to understand diffusion models. Coding was kinda fun on this one. I will take a break for 1-2 months and study something related like GANs or VAE, or even energy-based models. Then comeback with more general understanding :) Thanks !

    • @zenchiassassin283
      @zenchiassassin283 Рік тому

      And transformers for the attention mechanisms + positional encoding

    • @TheAero
      @TheAero Рік тому

      I got that snatched in the past 2 months. Gotta learn the math, what is actually a distribution etc.@@zenchiassassin283

  • @prabhavkaula9697
    @prabhavkaula9697 2 роки тому

    Thank you for sharing the implementation since authentic resources are rare

  • @manuelsebastianriosbeltran972
    @manuelsebastianriosbeltran972 2 роки тому +1

    Congrats, This is a great channel!! hope to see more of these videos in the future.

  • @pratyanshvaibhav
    @pratyanshvaibhav 7 місяців тому

    The Under rated OG channel

  • @NickSergievskiy
    @NickSergievskiy 2 роки тому

    Thank you. Best explanation with good DNN models

  • @vinc6966
    @vinc6966 Рік тому

    Amazing tutorial, very informative and clear, nice work!

  • @ZhangzhiPeng-x8r
    @ZhangzhiPeng-x8r 2 роки тому +2

    great tutorial! looking to seeing more of this! keep it up!

  • @smnvrs
    @smnvrs Рік тому

    Thanks, this implementation really helped clear things up.

  • @947973
    @947973 Рік тому

    Very helpful walk-through. Thank you!

  • @DiogoSanti
    @DiogoSanti Рік тому

    Very well done! Keep the great content!!

  • @yuhaowang9846
    @yuhaowang9846 Рік тому

    Thank you so much for this sharing, that was perfect!

  • @qq-mf9pw
    @qq-mf9pw Рік тому

    Incredible explanation, thanks a lot!

  • @LMonty-do9ud
    @LMonty-do9ud Рік тому

    Thank you very much, it has solved my urgent need

  • @dylanwattles7303
    @dylanwattles7303 Рік тому

    nice demonstration, thanks for sharing

  • @Kooshiar
    @Kooshiar Рік тому

    best diffusion youtube

  • @talktovipin1
    @talktovipin1 2 роки тому

    Looking forward for some video on Classifier Guidance as well. Thanks.

  • @yazou3896
    @yazou3896 Рік тому

    It's definitely cool and helpful! Thanks!!!

  • @WendaoZhao
    @WendaoZhao 7 місяців тому +1

    one CRAZY thing to take from this code (and video)
    GREEK LETTERS ARE CAN BE USED AS VARIABLE NAME IN PYTHON

  • @henrywong741
    @henrywong741 Рік тому +2

    Could you please explain the paper "High Resolution Image Synthesis With Latent Diffusion Models" and its implementations? Your explanations are exceptionally crystal.

  • @rachelgardner1799
    @rachelgardner1799 Рік тому

    Fantastic video!

  • @junghunkim8467
    @junghunkim8467 Рік тому

    it is very helpful!! You are a genius.. :) thank you!!

  • @xuefengdu6926
    @xuefengdu6926 2 роки тому

    thanks for your amazing efforts!

  • @talktovipin1
    @talktovipin1 2 роки тому

    Very nicely explained. Thanks.

  • @houbenbub
    @houbenbub 2 роки тому

    This is GOLD

  • @orestispapanikolaou9798
    @orestispapanikolaou9798 2 роки тому

    Great video!! You make coding seem like playing super mario 😂😂

  • @spartancoder
    @spartancoder 2 роки тому

    This video is priceless.

  • @ParhamEftekhar
    @ParhamEftekhar 8 місяців тому

    Awesome video.

  • @nez2884
    @nez2884 2 роки тому

    awesome implementation!

  • @pedrambazrafshan9598
    @pedrambazrafshan9598 3 місяці тому

    Could you also make a video on how to implement DDIM? Or make a GitHub repository about it?

  • @kerenye955
    @kerenye955 Рік тому

    Great video!

  • @spyrosmarkesinis443
    @spyrosmarkesinis443 2 роки тому

    Amazing stuff!

  • @Neptutron
    @Neptutron 2 роки тому +1

    Thank you!!

  • @andrewluo6088
    @andrewluo6088 2 роки тому +1

    The best

  • @mcpow6614
    @mcpow6614 Рік тому +2

    Can you do one for tensorflow too btw very good explaination

  • @decode168
    @decode168 2 місяці тому

    Very great topic. Could you please make another video of generating text to text images? Example: cat with “hello world “, so the model could generate the picture. Thanks ❤

    • @outliier
      @outliier  2 місяці тому

      @@decode168 usually this is learnt when you train a big model on a lot of data automatically. So there is no specific technique for this

  • @sandravu1541
    @sandravu1541 2 роки тому

    great video, you got one new subscriber

  • @김남형산업공학과한양
    @김남형산업공학과한양 2 роки тому

    Thank you for sharing!

  • @salehgholamzadeh3368
    @salehgholamzadeh3368 4 місяці тому

    Great video
    I faced a question at 19:10 line 50 of the code. why do we call
    ```model(x,label,None)```
    what happened to t? shouldn't we instead call it like ```model(x,t,None)``` ??
    also line 17 in ema (20:31) ```retrun old * self.beta +(1+self.beta) * new``` why 1+self.beta? shouldnt it be 1-self.beta?

  • @chickenp7038
    @chickenp7038 2 роки тому +2

    great walkthrough. but where would i implement dynamic or static thresholding as described in the imagen paper? the static thresholding clips all values larger then 1 but my model regularly outputs numbers as high as 5. but it creates images and loss decreases to 0.016 with SmoothL1Loss.

  • @versusFliQq
    @versusFliQq Рік тому

    Really nice video! I also enjoyed your explanation video - great work in general :)
    However, I noticed at around 5:38, you are defining sample_timesteps with low=1. I am pretty sure that this is wrong, as Python indexes at 0 meaning you skip the first noising step every time you access alpha, alpha_cumprod etc. Correct me if I am wrong but all the other implementations also utilise zero-indexing.

    • @arpanpoudel
      @arpanpoudel Рік тому

      this function sample the timesteps of the denoising step. selecting time=0 is the original image itself. there is no point in taking 0 timestep.

  • @SkyHighBeyondReach
    @SkyHighBeyondReach 6 місяців тому

    Thanks alot :)

  • @satpalsinghrathore2665
    @satpalsinghrathore2665 2 роки тому

    Super cool

  • @jinhengfeng6440
    @jinhengfeng6440 2 роки тому

    terrific!

  • @scotth.hawley1560
    @scotth.hawley1560 Рік тому

    Wonderful video! I notice that at 18:50, the equation for the new noise seems to differ from Eq. 6 in the CFG paper, as if the unconditioned and conditioned epsilons are reversed. Can you comment on that?

  • @SeonhoonKim
    @SeonhoonKim Рік тому +1

    Hello, thanks for your a lot contribution ! But a bit confused, At 06:04, just sampling from N(0, 1) totally randomly would not have any "trace" of an image. How come the model infer the image from the totally random noise ?

    • @outliier
      @outliier  Рік тому +1

      Hey there, that is sort of the "magic" of diffusion models which is hard to grasp your mind around. But since the model is trained to always see noise between 0% and 100% it sees full noise during training for which it is then trained to denoise it. And usually when you provide conditioning to the model such as class labels or text information, the model has more information than just random noise. But still, unconditional training still works.

  • @Gruell
    @Gruell 9 місяців тому

    Sorry if I am misunderstanding, but at 19:10, shouldn't the code be:
    "uncond_predicted_noise = model(x, t, None)" instead of "uncond_predicted_noise = model(x, labels, None)"
    Also, according to the CFG paper's formula, shouldn't the next line be: "predicted_noise = torch.lerp(predicted_noise, uncond_predicted_noise, -cfg_scale)" under the definition of lerp?
    One last question: have you tried using L1Loss instead of MSELoss? On my implementation, L1 Loss performs much better (although my implementation is different than yours). I know the ELBO term expands to essentially an MSE term wrt predicted noise, so I am confused as to why L1 Loss performs better for my model.
    Thank you for your time.

    • @Gruell
      @Gruell 9 місяців тому

      Great videos by the way

    • @Gruell
      @Gruell 9 місяців тому

      Ah, I see you already fixed the first question in the codebase

  • @Soso65929
    @Soso65929 11 місяців тому

    So the process of adding noise and removing it happens in a loop

  • @doctorshadow2482
    @doctorshadow2482 Рік тому

    Thank you for the review. So, what is the key to make a step from text description to image? Can you please pinpoint where it is explained?

  • @jamesfogwill1455
    @jamesfogwill1455 6 місяців тому +1

    Roughly how long does an Epoch take for you? I am using rtx3060 mobile and achieving an epoch every 24 minutes. Also i cannot work with a batch size greater than 8 and a img size greater than 64 because it overfills my GPUs 6gb memory. I thought this was excessive for such small batch and img size?

  • @maybritt-sch
    @maybritt-sch 2 роки тому +1

    Great videos on diffusion models, very understandable explanations! For how many hours did you train it? I tried adjusting your conditional model and train with a different dataset, but it seems to take forever :D

    • @outliier
      @outliier  2 роки тому +1

      Yea it took quite long. On the 3090 it trained a couple days (2-4 days I believe)

    • @maybritt-sch
      @maybritt-sch 2 роки тому

      @@outliier Thanks for the feedback. Ok seems like I didn't do a mistake, but only need more patience!

    • @outliier
      @outliier  2 роки тому

      @@maybritt-sch Yea. Let me know how it goes or if you need help

  • @LonLat1842
    @LonLat1842 2 роки тому

    Nice tutorial

  • @LeeYuanZ
    @LeeYuanZ Рік тому

    Thank you so much for this amazing video! In mention that the first DDPM paper show no necessary of lower bound formulation, could you tell me the specific place in the paper? thanks!

  • @gordondou2286
    @gordondou2286 Рік тому

    Can you please explain how to use Woodfisher technique to approximate second-order gradients? Thanks

  • @ankanderia4999
    @ankanderia4999 10 місяців тому

    `
    x = torch.randn((n, 3, self.img_size, self.img_size)).to(self.device)
    predicted_noise = model(x, t)
    `
    in the deffusion class why you create an noise and pass that noise into the model to predict noise ... please explain

  • @Sherlock14-d6x
    @Sherlock14-d6x 7 місяців тому

    Why is the bias off in the initial convolutional block?

  • @khangvutien2538
    @khangvutien2538 Рік тому

    People in Earth Observation know that images from Synthetic Aperture Radar have random speckles. People have tried removing the speckles using wavelets. I wonder how Denoising Diffusion would fare. The difficulty that I see is the need for x0 the un-noised image.
    What do you think?

  • @muhammadawais2173
    @muhammadawais2173 Рік тому

    thanks for the easiest implementation. could you plz tell us how to find FID and IS score for these images?

    • @outliier
      @outliier  Рік тому +1

      I think you would just sample 10-50k images from the trained model and then take 10-50k images from the original dataset and then calculate the FID and IS

    • @muhammadawais2173
      @muhammadawais2173 Рік тому

      @@outliier thanks

  • @anonymousperson9757
    @anonymousperson9757 2 роки тому +1

    Thank you so much for this amazing video! You mention that changing the original DDPM to a conditional model should be as simple as adding in the condition at some point during training. I was just wondering if you had any experience with using DDPM to denoise images? I was planning on conditioning the model on the input noisy data by concatenating it to yt during training. I am going to try and play around with your github code and see if I can get something to work with denoising. Wish me luck!

  • @kashishmathukiya8091
    @kashishmathukiya8091 Рік тому

    8:38 in the UNet section, how do you decide on the number of channels to set in both input and output to the Down and Up classes. Why just 64,128, etc. ?

    • @outliier
      @outliier  Рік тому +1

      People just go with powers of 2 usually. And usually you go to more channels in the deeper layers of the network.

    • @kashishmathukiya8091
      @kashishmathukiya8091 Рік тому

      @@outliier oh okay got it. Thank you so much for clearing that and for the video! I had seen so many videos / read articles for diffusion but yours were the best and explained every thing which others considered prerequisites!! Separating the paper explanation and implementation was really helpful.

  • @sweetautumnfox
    @sweetautumnfox 11 місяців тому

    With this training method, wouldn't there be a possibility of some timesteps not being trained in an epoch? wouldn't it be better to shuffle the whole list of timesteps and then sample sequentially with every batch?

  • @khyatinkadam8032
    @khyatinkadam8032 7 місяців тому

    hey can we use an image as a condition

  • @ovrava
    @ovrava 2 роки тому

    Great Video,
    On what Data did you train your model again?

  • @andonso
    @andonso Рік тому

    How can i increase the img size to 128 pixels square?

  • @signitureDGK
    @signitureDGK Рік тому

    Very cool. How would DDIM models be different? Do they use a deterministic denoising sampler?

  • @susdoge3767
    @susdoge3767 11 місяців тому

    having hard time to understand the mathematical and code aspect of diffusion model although i have a good high level understanding...any good resource i can go through? id appreciate it

  • @orangethemeow
    @orangethemeow 2 роки тому +1

    Like your channel, please make more videos

  • @nomaannafi7561
    @nomaannafi7561 Рік тому

    How can i increase the size of the generated image here?

  • @Laszer271
    @Laszer271 Рік тому

    There is a slight bug at 19:11
    it should be
    uncond_predicted_noise = model(x, t, None)
    and not
    uncond_predicted_noise = model(x, labels, None)

    • @outliier
      @outliier  Рік тому

      Yes correct. Good catch

  • @wizzy1996pl
    @wizzy1996pl Рік тому

    last self attention layer (64, 64) changes my training type from 5 minutes to hours per epoch, do you know why?
    training on a single rtx 3060 TI gpu

  • @gabrielchan3255
    @gabrielchan3255 2 роки тому

    Thx Mr gigachad

  • @zedtarwu3074
    @zedtarwu3074 2 роки тому

    Great video! How long did it take to train the models?

    • @outliier
      @outliier  2 роки тому

      About 3-4 days on an rtx 3090.

  • @janevirahman9904
    @janevirahman9904 11 місяців тому

    Hi , I want to use a single underwater image dataset what changes do i have to implement on the code?

  • @luchaoqi
    @luchaoqi 2 роки тому +1

    Awesome! How did you type Ɛ in code?

  • @UnbelievableRam
    @UnbelievableRam 9 місяців тому

    Hi! Can you please explain why the output is getting two stitched images?

    • @outliier
      @outliier  9 місяців тому

      What do you mean with two stitched images?

  • @NoahElRhandour
    @NoahElRhandour 2 роки тому +1

    pog!!!

  • @egoistChelly
    @egoistChelly Рік тому

    I think your code bugs when adjust image_size?

  • @remmaria
    @remmaria 2 роки тому

    Your videos are a blessing. Thank you very much!!! Have you tried using DDIM to accelerate predictions? Or any other idea to decrease the number of steps needed?

    • @outliier
      @outliier  2 роки тому +1

      I have not tried any speedups in any way. But feel free to try it out and tell me / us what works best. In the repo I do linked a fork which implements a couple additions which make the training etc. faster. You can check that out too here: github.com/tcapelle/Diffusion-Models-pytorch

    • @remmaria
      @remmaria 2 роки тому

      @@outliier Thank you! I will try it for sure.

  • @homataha5626
    @homataha5626 2 роки тому

    Thank you for the video.
    How can we use diffusion model for inpainting?

  • @Naira-ny9zc
    @Naira-ny9zc 2 роки тому

    Thank you...U just made diffusion so easy to understand... I would like to ask ; What changes do I need to make in order to give an image as condition rather than a label as condition. I mean how to load ground Truth from GT repository as label (y).

    • @outliier
      @outliier  2 роки тому +1

      Depends on your task. Could you specify what you want to achieve? Super resolution? Img2Img?

    • @Naira-ny9zc
      @Naira-ny9zc 2 роки тому

      @@outliier I want to generate thermal IR images conditioned on their respective RGB images . I know that in order to achieve this task i.e ; Image (RGB) to Image (Thermal IR) translation, I have to concat the input to U-net (which of course is thermal noise image ) with its corresponding RGB (condition image) and give this concatenated output as final input to the unet ; but problem is that I am not able to put this all together in the code (especially concatenating each RGB image (condition) from RGB image folder with its corresponding Thermal noise images so that I can pass the concatenated resultant image as final input to Unet as my aim is to generate RGB conditioned Thermal image using Diffusion.

  • @mic9657
    @mic9657 Рік тому

    great video. please can you list the creators of the other helpful videos at 00:52? thanks

    • @outliier
      @outliier  Рік тому

      There are from Yannick Kilcher (on the right side), the one in the lower left is from AICoffeeBreak, the one in the top right corner is the first video that comes when you google „diffusion models explained“ and I forgot the middle one sorry. But shouldnt be hard to find

  • @colintsang-ww6mz
    @colintsang-ww6mz Рік тому

    Thank you very much for this very easy-to-understand implementation. I have one question: I don't understand the function def noise_images.
    Assume that we have img_{0}, img_{1}, ..., img_{T}, which are obtained from adding the noise iteratively. I understand that img{t} is given by the formula "sqrt_alpha_hat * img_{0} + sqrt_one_minus_alpha_hat * Ɛ".
    However, I don't understand the function "def noise_images(self, x, t)" in [ddpm.py].
    It return Ɛ, where Ɛ = torch.randn_like(x). So, this is just a noise signal draw directly from the normal distribution. I suppose this random noise is not related to the input image? It is becasue randn_like() returns a tensor with the same size as input x that is filled with random numbers from a normal distribution with mean 0 and variance 1
    In training, the predicted noise is compared to this Ɛ (line 80 in [ddpm.py]).
    Why we are predicting this random noise? Shouldn't we predict the noise added at time t, i.e. "img_{t} - img_{t-1}"?

    • @Laszer271
      @Laszer271 Рік тому +1

      I had the same misconception before. It was actually explained by "AI Coffee Break with Letitia" channel in a video titled "How does Stable Diffusion work? - Latent Diffusion Models EXPLAINED".
      Basically, the model tries to predict the WHOLE noise added to the image to go from noised image to a fully denoised image in ONE STEP. Because it's a hard task to do, the model does not excel at that so at inference we denoise it iteratively, each time subtracting only a small fraction of the noise predicted by the model. In this way, the model produces much better quality samples. At least that's how I understood it :P

    • @rikki146
      @rikki146 Рік тому

      @@Laszer271 While I understand it predicts the "whole noise", this "whole noise" is newly generated and I suppose the ground truth is (img_{t} - img_{0)).. still can't wrap my head around it.