Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Potential feature: One-shot generative models based on single training image #337

Open
RichardObi opened this issue Mar 22, 2023 · 2 comments

Comments

@RichardObi
Copy link

For use-cases where there are very few training images (e.g. perhaps rare diseases), it could be nice to augment the data using one-shot generative models. Could this be something of interest to have in MONAI GenerativeModels?
Example models include sinfusion, SinGAN, OneShotGAN, ConSinGAN

@Warvito
Copy link
Collaborator

Warvito commented Mar 23, 2023

That sounds like a good idea, @RichardObi ^^ I need to get more familiar with these methods yet. For our package, which one of them do you think we start with? Which one looks more promising for medical data (2D and 3D)?

@RichardObi
Copy link
Author

RichardObi commented Apr 5, 2023

Cool :) Often these methods learn from patches of the training image or train multiple generators that iteratively scale the training image up with some variation in each step.

  • SinGAN is the most widely used and cited. So far, it has been used to generate 2D medical imaging data (ref1, ref2, ref3, ref4).
  • The other one-shot generative models such as OneShotGAN and ConSinGAN extend upon SinGAN. I could not find an application of them to medical images yet.
  • sinfusion is based on a diffusion model rather than a GAN. It is still quite recent (Nov 2022) and (therefore) has not yet been tested widely with medical images. In the paper, it is also applied to video data and perhaps the video methodology could guide its application to 3D medical data. Therefore, sinfusion could be a great addition to GenerativeModels and perhaps some of the diffusion model training and inference code that is already in GenerativeModels could be reused to integrate sinfusion.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants