Skip to content
ModelListedExternal
Download

Imagenet Sae Top K 64 Patches Only Layer 11 Hook Resid Post 128 79

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

KindOtherVersionWeightsCgrade

Model source

Kind
Other
File size
288 MB
Version
vWeights
Updated
Oct 6, 2026
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: 11
  • 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.0001
  • L1 Coefficient: 0.0002
  • Batch Size: 4096
  • Context Size: 49

Performance Metrics

Sparsity

  • L0 (Active Features): 128.0000
  • Dead Features: 0
  • Mean Log10 Feature Sparsity: -2.8426
  • Features Below 1e-5: 2
  • Features Below 1e-6: 0
  • Mean Passes Since Fired: 0.2142

Reconstruction

  • Explained Variance: 0.7933
  • Explained Variance Std: 0.0639
  • MSE Loss: 0.0053
  • L1 Loss: 0
  • Overall Loss: 0.0053

Training Details

  • Training Duration: 4260 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 510618ff2af7…ccb9 · Hugging Face

Download

Source excerpts

2 excerpts

# CLIP Sparse Autoencoder Checkpoint This model is a sparse autoencoder trained on CLIP's internal representations. ## Model Details ### Architecture - **Layer**: 11 - **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.0001 - **L1 Coefficient**: 0.0002 - **Batch Size**: 4096 - **Context Size**: 49 ## Performance Metrics ### Sparsity - **L0 (Active Features)**: 128.0000 - **Dead Features**: 0 - **Mean Log10 Feature Sparsity**: -2.8426 - **Features Below 1e-5**: 2 - **Features Below 1e-6**: 0 - **Mean Passes Since Fired**: 0.2142 ### Reconstruction - **Explained Variance**: 0.7933 - **Explained Variance Std**: 0.0639 - **MSE Loss**: 0.0053 - **L1 Loss**: 0 - **Overall Loss**: 0.0053 ## Training Details - **Training Duration**: 4260 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/e238a05a-tinyclip_sae_16_hyperparam_sweep_lr/n_images_1300020.pt - **Wandb Run**: https://wandb.ai/perceptual-alignment/imagenet-sweep-topk-patches_all_layers/runs/5llrenyy - **Random Seed**: 42

Source context: 8 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_11-hook_resid_post-128-79