sanket-mhatre commited on
Commit
81da898
·
verified ·
1 Parent(s): 2fd0fc1

Upload gated-50m-9lang/README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. gated-50m-9lang/README.md +60 -0
gated-50m-9lang/README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sparse-autoencoder
4
+ - multilingual
5
+ - safety
6
+ - interpretability
7
+ language: [arb, ben, deu, eng, fra, hin, jpn, spa, zho]
8
+ ---
9
+
10
+ # Gated SAE — Multilingual Safety Features
11
+
12
+ Sparse autoencoder trained on **CohereLabs/tiny-aya-global** (layer 20) for cross-lingual safety representation analysis.
13
+
14
+ ## Training Config
15
+
16
+ | Parameter | Value |
17
+ |-----------|-------|
18
+ | Architecture | Gated SAE |
19
+ | d_model | 2048 |
20
+ | d_sae | 16384 |
21
+ | Expansion | 8x |
22
+ | L1 coefficient | 0.0005 |
23
+ | Training tokens | 50,000,008 |
24
+ | Languages | 9 (arb, ben, deu, eng, fra, hin, jpn, spa, zho) |
25
+
26
+ ## Metrics
27
+
28
+ | Metric | Value |
29
+ |--------|-------|
30
+ | CE recovered | 0.9989 |
31
+ | Cosine similarity | 0.9508 |
32
+ | Explained variance | 0.9861 |
33
+ | L0 (eval) | 1223.8 |
34
+ | Dead features | 0.0183% |
35
+ | Total steps | 12161 |
36
+
37
+ ## Language Balance
38
+
39
+ | Language | Tokens |
40
+ |----------|--------|
41
+ | arb | 5,555,597 |
42
+ | ben | 5,555,660 |
43
+ | deu | 5,555,565 |
44
+ | eng | 5,555,259 |
45
+ | fra | 5,555,561 |
46
+ | hin | 5,555,584 |
47
+ | jpn | 5,555,595 |
48
+ | spa | 5,555,600 |
49
+ | zho | 5,555,587 |
50
+
51
+ ## Usage
52
+
53
+ ```python
54
+ import torch
55
+ from saefty.models.sae.gated import GatedSAE
56
+
57
+ state = torch.load("sae_final.pt", map_location="cpu")
58
+ sae = GatedSAE(d_model=2048, d_sae=16384)
59
+ sae.load_state_dict(state["sae_state_dict"])
60
+ ```