Skip to content
ModelListedExternal
Download

Imagenet Sae Top K 64 Patches Only Layer 9 Hook Resid Post 64 82

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

KindOtherVersionRevision be9a43aCgrade

Model source

Kind
Other
File size
288 MB
Version
vRevision be9a43a
Updated
Oct 7, 2026
Downloads
11
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: 9
  • 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.0015
  • 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.6212
  • Features Below 1e-5: 407
  • Features Below 1e-6: 8
  • Mean Passes Since Fired: 2.6682

Reconstruction

  • Explained Variance: 0.8289
  • Explained Variance Std: 0.0560
  • MSE Loss: 0.0038
  • L1 Loss: 0
  • Overall Loss: 0.0038

Training Details

  • Training Duration: 4155 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 d7f9c4140b39…c7af · 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**: 9 - **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.0015 - **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.6212 - **Features Below 1e-5**: 407 - **Features Below 1e-6**: 8 - **Mean Passes Since Fired**: 2.6682 ### Reconstruction - **Explained Variance**: 0.8289 - **Explained Variance Std**: 0.0560 - **MSE Loss**: 0.0038 - **L1 Loss**: 0 - **Overall Loss**: 0.0038 ## Training Details - **Training Duration**: 4155 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/6fa7c4bd-tinyclip_sae_16_hyperparam_sweep_lr/n_images_1300020.pt - **Wandb Run**: https://wandb.ai/perceptual-alignment/imagenet-sweep-topk-patches_all_layers/runs/hl7n2nav - **Random Seed**: 42

Source context: 11 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_9-hook_resid_post-64-82

Source context: 5 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_9-hook_resid_post-64-82