Upload folder using huggingface_hub
Browse files- .gitattributes +3 -0
- README.md +18 -0
- fig_guidance_scale.png +3 -0
- fig_qualitative_comparison.png +3 -0
- fig_training_dynamics.png +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
fig_guidance_scale.png filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
fig_qualitative_comparison.png filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
fig_training_dynamics.png filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -36,6 +36,24 @@ This model adapts Stable Diffusion 1.5 for longitudinal retinal image prediction
|
|
| 36 |
|
| 37 |
Evaluated on 110 held-out test image pairs.
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
## Usage
|
| 40 |
|
| 41 |
```python
|
|
|
|
| 36 |
|
| 37 |
Evaluated on 110 held-out test image pairs.
|
| 38 |
|
| 39 |
+
## Qualitative Results
|
| 40 |
+
|
| 41 |
+

|
| 42 |
+
|
| 43 |
+
Each row shows a different test patient. Columns: baseline fundus, ground-truth follow-up, our prediction, Regression U-Net, and Pix2Pix. Our diffusion model generates sharper, more realistic retinal textures compared to deterministic baselines.
|
| 44 |
+
|
| 45 |
+
## Training Dynamics
|
| 46 |
+
|
| 47 |
+

|
| 48 |
+
|
| 49 |
+
(a, b) Stage 1: VAE fine-tuning over 50 epochs reaching SSIM 0.954. (c-f) Stage 2: U-Net training over 500 epochs with cosine LR schedule and warmup.
|
| 50 |
+
|
| 51 |
+
## Guidance Scale Sweep
|
| 52 |
+
|
| 53 |
+

|
| 54 |
+
|
| 55 |
+
SSIM peaks at guidance scale 7.5, while FID increases monotonically with stronger guidance, reflecting the fidelity-diversity tradeoff.
|
| 56 |
+
|
| 57 |
## Usage
|
| 58 |
|
| 59 |
```python
|
fig_guidance_scale.png
ADDED
|
Git LFS Details
|
fig_qualitative_comparison.png
ADDED
|
Git LFS Details
|
fig_training_dynamics.png
ADDED
|
Git LFS Details
|