BogdanTurbal/synthetic_mri_generation_sd_xl

BRAIN with Alzheimer's MRI generation (LoRA text2image fine-tuning)

These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.

With this model, you can generate an MRI of the brain of 4 classes:

1. NonDemented
2. VeryMildDemented
3. MildDemented
4. ModerateDemeneted

For generation use the following corresponding prompts (or any other modification)

1. MRI of NonDemented brain
2. MRI of VeryMildDemented brain
3. MRI of MildDemented brain
4. MRI of ModerateDemeneted brain

LoRA for the text encoder was enabled: False.

Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.

Intended uses & limitations

How to use

# TODO: add an example code snippet for running this diffusion pipeline

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]