Skip to content
ModelListedExternal
Download

Imagenet Sae Top K 64 Patches Only Layer 10 Hook Resid Post 64 80

This model is a sparse autoencoder trained on CLIP's internal representations.

KindOtherVersionRevision 0e6eedeCgrade

Model source

Kind
Other
File size
288 MB
Version
vRevision 0e6eede
Updated
Oct 6, 2026
Downloads
10
Source
Hugging Face

Source description

CLIP Sparse Autoencoder Checkpoint

This model is a sparse autoencoder trained on CLIP's internal representations.

Model Details

Architecture

  • Layer: 10
  • Layer Type: hook_resid_post
  • Model: open-clip:laion/CLIP-ViT-B-32-DataComp.XL-s13B-b90K
  • Dictionary Size: 49152
  • Input Dimension: 768
  • Expansion Factor: 64
  • CLS Token Only: False

Training

  • Training Images: 1299936
  • Learning Rate: 0.0004
  • L1 Coefficient: 0.0002
  • Batch Size: 4096
  • Context Size: 49

Performance Metrics

Sparsity

  • L0 (Active Features): 64.0000
  • Dead Features: 0
  • Mean Log10 Feature Sparsity: -3.1933
  • Features Below 1e-5: 28
  • Features Below 1e-6: 2
  • Mean Passes Since Fired: 0.3781

Reconstruction

  • Explained Variance: 0.8094
  • Explained Variance Std: 0.0674
  • MSE Loss: 0.0047
  • L1 Loss: 0
  • Overall Loss: 0.0047

Training Details

  • Training Duration: 3956 seconds
  • Final Learning Rate: 0.0000
  • Warm Up Steps: 500
  • Gradient Clipping: 1

Additional Information

Sources

1 source

Model artifacts

1 artifact

weights.pt

pt · 288 MB · SHA-256 bc8320f1e6d3…82b2 · Hugging Face

Download

Source excerpts

3 excerpts

# CLIP Sparse Autoencoder Checkpoint This model is a sparse autoencoder trained on CLIP's internal representations. ## Model Details ### Architecture - **Layer**: 10 - **Layer Type**: hook_resid_post - **Model**: open-clip:laion/CLIP-ViT-B-32-DataComp.XL-s13B-b90K - **Dictionary Size**: 49152 - **Input Dimension**: 768 - **Expansion Factor**: 64 - **CLS Token Only**: False ### Training - **Training Images**: 1299936 - **Learning Rate**: 0.0004 - **L1 Coefficient**: 0.0002 - **Batch Size**: 4096 - **Context Size**: 49 ## Performance Metrics ### Sparsity - **L0 (Active Features)**: 64.0000 - **Dead Features**: 0 - **Mean Log10 Feature Sparsity**: -3.1933 - **Features Below 1e-5**: 28 - **Features Below 1e-6**: 2 - **Mean Passes Since Fired**: 0.3781 ### Reconstruction - **Explained Variance**: 0.8094 - **Explained Variance Std**: 0.0674 - **MSE Loss**: 0.0047 - **L1 Loss**: 0 - **Overall Loss**: 0.0047 ## Training Details - **Training Duration**: 3956 seconds - **Final Learning Rate**: 0.0000 - **Warm Up Steps**: 500 - **Gradient Clipping**: 1 ## Additional Information - **Original Checkpoint Path**: /network/scratch/p/praneet.suresh/celeba_checkpoints/d6ee4316-tinyclip_sae_16_hyperparam_sweep_lr/n_images_1300020.pt - **Wandb Run**: https://wandb.ai/perceptual-alignment/imagenet-sweep-topk-patches_all_layers/runs/fci8e1q5 - **Random Seed**: 42

Source context: 10 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_10-hook_resid_post-64-80

Source context: 4 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_10-hook_resid_post-64-80