nielsr HF Staff commited on
Commit
d952f2b
·
verified ·
1 Parent(s): 9f39163

Improve model card: add paper, code, dataset links and metadata

Browse files

Hi!

This PR improves the model card for `hcarrion/keloid` by:
- Adding `library_name: diffusers` and `pipeline_tag: text-to-image` to the YAML metadata.
- Linking the model card to the corresponding paper ([Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification](https://huggingface.co/papers/2607.12987)).
- Linking to the official GitHub repository and Hugging Face dataset.
- Specifying the trigger token (`<keloid-class>`) to make it clear how to use the model.
- Providing the BibTeX citation.

Files changed (1) hide show
  1. README.md +44 -7
README.md CHANGED
@@ -1,17 +1,54 @@
1
-
2
  ---
3
- license: creativeml-openrail-m
4
  base_model: stabilityai/stable-diffusion-2-1-base
 
 
 
5
  tags:
6
  - stable-diffusion
7
  - stable-diffusion-diffusers
8
- - text-to-image
9
- - diffusers
10
  - textual_inversion
11
  inference: true
12
  ---
13
-
14
- # Textual inversion text2image fine-tuning - hcarrion/keloid
15
- These are textual inversion adaption weights for stabilityai/stable-diffusion-2-1-base. You can find some example images in the following.
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
 
2
  base_model: stabilityai/stable-diffusion-2-1-base
3
+ license: creativeml-openrail-m
4
+ library_name: diffusers
5
+ pipeline_tag: text-to-image
6
  tags:
7
  - stable-diffusion
8
  - stable-diffusion-diffusers
 
 
9
  - textual_inversion
10
  inference: true
11
  ---
 
 
 
12
 
13
+ # cgDDI: Keloid Textual Inversion Model
14
+
15
+ This repository contains the textual inversion adaptation weights for `stabilityai/stable-diffusion-2-1-base` representing the **keloid** disease concept.
16
+
17
+ It 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](https://huggingface.co/papers/2607.12987) (MICCAI 2026).
18
+
19
+ ## Model Description
20
+
21
+ cgDDI is a hybrid framework designed to address the systematic lack of expertly annotated dermatological images across diverse skin tones and rare diseases. This model learns the specific concept of **keloid** through textual inversion and fine-tuning, allowing the generation of diverse, realistic, and skin-tone-controlled synthetic samples.
22
+
23
+ - **Disease Category:** Keloid
24
+ - **Trigger Token/Concept:** `<keloid-class>`
25
+ - **Base Model:** `stabilityai/stable-diffusion-2-1-base`
26
+
27
+ ## cgDDI Framework Overview
28
+
29
+ <p align="center">
30
+ <img src="https://raw.githubusercontent.com/hectorcarrion/ControllableGenDDI/main/assets/cgddi_framework_colors_2.png" width="900" alt="cgDDI framework: healthy synthetics, lesion-mapped synthetics, semantic synthetics, and fair classification">
31
+ </p>
32
+
33
+ 1. **Healthy Synthetics:** Synthesizes realistic healthy skin samples without disturbing other input properties.
34
+ 2. **Lesion Mapping:** Maps single-sample rare lesions onto novel skin-tones and locations non-parametrically.
35
+ 3. **Semantic Synthetics:** Allows for efficient parametric generation of targeted dermatological conditions using textual inversion and LoRA.
36
+
37
+ ## Resources
38
+
39
+ - **Paper:** [Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification](https://huggingface.co/papers/2607.12987)
40
+ - **Code:** [GitHub Repository](https://github.com/hectorcarrion/ControllableGenDDI)
41
+ - **Dataset:** [Hugging Face Dataset](https://huggingface.co/datasets/hcarrion/ControllableGenDDI)
42
+
43
+ ## Citation
44
 
45
+ ```bibtex
46
+ @inproceedings{carrion2026cgddi,
47
+ title = {Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification},
48
+ author = {Carri{\'o}n, H{\'e}ctor and Norouzi, Narges},
49
+ booktitle = {Medical Image Computing and Computer-Assisted Intervention (MICCAI)},
50
+ year = {2026},
51
+ publisher = {Springer},
52
+ series = {Lecture Notes in Computer Science}
53
+ }
54
+ ```