usama10 commited on
Commit
a48c7ac
·
verified ·
1 Parent(s): 1e27037

Upload folder using huggingface_hub

Browse files
.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
+ ![Qualitative comparison](fig_qualitative_comparison.png)
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
+ ![Training dynamics](fig_training_dynamics.png)
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
+ ![Guidance scale](fig_guidance_scale.png)
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

  • SHA256: 6164cc5abead5aedf5e78f70b4210ea73f639361f5c424593edb34495edbd079
  • Pointer size: 131 Bytes
  • Size of remote file: 189 kB
fig_qualitative_comparison.png ADDED

Git LFS Details

  • SHA256: fef6fe834aed4da48d9bf74ee6dc91d65087c2a2f4be894907e7855a00992589
  • Pointer size: 132 Bytes
  • Size of remote file: 5.26 MB
fig_training_dynamics.png ADDED

Git LFS Details

  • SHA256: 4cd68bcf58eaa0adfdadc375f0366b81dd6b3c7975864084251b84a7010583c3
  • Pointer size: 132 Bytes
  • Size of remote file: 1.1 MB