Skip to content
ModelListedExternal
Download

Imagenet Sae Top K 64 Patches Only Layer 8 Hook Resid Post 128 83

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

KindOtherVersionRevision 4642b4bCgrade

Model source

Kind
Other
File size
288 MB
Version
vRevision 4642b4b
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: 8
  • 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.0002
  • 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.8240
  • Features Below 1e-5: 0
  • Features Below 1e-6: 0
  • Mean Passes Since Fired: 0.0485

Reconstruction

  • Explained Variance: 0.8354
  • Explained Variance Std: 0.0434
  • MSE Loss: 0.0027
  • L1 Loss: 0
  • Overall Loss: 0.0027

Training Details

  • Training Duration: 4213 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 b9041fae061e…50eb · 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**: 8 - **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.0002 - **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.8240 - **Features Below 1e-5**: 0 - **Features Below 1e-6**: 0 - **Mean Passes Since Fired**: 0.0485 ### Reconstruction - **Explained Variance**: 0.8354 - **Explained Variance Std**: 0.0434 - **MSE Loss**: 0.0027 - **L1 Loss**: 0 - **Overall Loss**: 0.0027 ## Training Details - **Training Duration**: 4213 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/4977d2e9-tinyclip_sae_16_hyperparam_sweep_lr/n_images_1300020.pt - **Wandb Run**: https://wandb.ai/perceptual-alignment/imagenet-sweep-topk-patches_all_layers/runs/ur12csb4 - **Random Seed**: 42

Source context: 10 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_8-hook_resid_post-128-83

Source context: 6 downloads · 0 likes · Repo Prisma-Multimodal/imagenet-sae-top_k-64-patches_only-layer_8-hook_resid_post-128-83