Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification
Paper • 2607.12987 • Published
How to use hcarrion/graft-vs-host_disease with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", dtype=torch.bfloat16, device_map="cuda")
pipe.load_textual_inversion("hcarrion/graft-vs-host_disease")This repository contains the weights for graft-vs-host_disease trained on top of stabilityai/stable-diffusion-2-1-base.
This model is part of the cgDDI (Controllable Generation of Diverse Dermatological Imagery) framework, introduced in the paper: Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification (MICCAI 2026).
If you find this work helpful, please consider citing:
@inproceedings{carrion2026cgddi,
title = {Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification},
author = {Carri{\\'o}n, H{\\'e}ctor and Norouzi, Narges},
booktitle = {Medical Image Computing and Computer-Assisted Intervention (MICCAI)},
year = {2026},
publisher = {Springer},
series = {Lecture Notes in Computer Science}
}
Base model
stabilityai/stable-diffusion-2-1-base