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

# CLIP Sparse Autoencoder Checkpoint This model is a sparse autoencoder trained on CLIP's internal representations. ## Model Details ### Architecture - **Layer**: 4 - **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**: -3.2837 - **Features Below 1e-5**: 9 - **Features Below 1e-6**: 0 - **Mean Passes Since Fired**: 1.2363 ### Reconstruction - **Explained Variance**: 0.8367 - **Explained Variance Std**: 0.0284 - **MSE Loss**: 0.0013 - **L1 Loss**: 0 - **Overall Loss**: 0.0013 ## Training Details - **Training Duration**: 4337 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/d5ec2e8b-tinyclip_sae_16_hyperparam_sweep_lr/n_images_1300020.pt - **Wandb Run**: https://wandb.ai/perceptual-alignment/imagenet-sweep-topk-patches_all_layers/runs/yipxajew - **Random Seed**: 42

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

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