nkiyohara commited on
Commit
5befb4b
·
verified ·
1 Parent(s): 140a86a

Upload LRA-Image Softmax (RoPE, causal) (test_acc=68.88%)

Browse files
Files changed (2) hide show
  1. README.md +59 -0
  2. lra_image_best.pt +3 -0
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: pytorch
4
+ tags:
5
+ - image-classification
6
+ - cifar-10
7
+ - lra
8
+ - sequence-model
9
+ - interdomain-attention
10
+ datasets:
11
+ - cifar10
12
+ metrics:
13
+ - accuracy
14
+ ---
15
+
16
+ # LRA-Image Softmax (RoPE, causal)
17
+
18
+ Softmax (RoPE, causal) model trained on the Long Range Arena (LRA) sCIFAR-10 benchmark.
19
+
20
+ ## Model Details
21
+
22
+ - **Architecture**: Softmax (RoPE, causal)
23
+ - **Task**: Sequential CIFAR-10 (grayscale, 1024 tokens)
24
+ - **Parameters**: ~4.1M
25
+ - **Position encoding**: RoPE
26
+ - **Causal**: True
27
+ - **Test Accuracy**: 68.88%
28
+
29
+ ## Training
30
+
31
+ - **Protocol**: S4 paper LRA-Image (200 epochs, lr=1e-3, batch=64, warmup=18k steps)
32
+ - **Backbone**: Llama-style (6 layers, d=512, 8 heads, RMSNorm, SwiGLU)
33
+ - **Seed**: 2222
34
+ - **WandB run**: [harrisonzhu/InterdomainAttention/x437k0ed](https://wandb.ai/harrisonzhu/InterdomainAttention/runs/x437k0ed)
35
+
36
+ ## Usage
37
+
38
+ ```python
39
+ import torch
40
+ from model import LlamaLRAImage # requires interdomain-attention repo
41
+
42
+ state_dict = torch.load("lra_image_best.pt", weights_only=True)
43
+ model = LlamaLRAImage(...) # match config
44
+ model.load_state_dict(state_dict["model"])
45
+ ```
46
+
47
+ ## Citation
48
+
49
+ ```bibtex
50
+ @article{interdomain2026,
51
+ title={Interdomain Attention},
52
+ author={...},
53
+ year={2026}
54
+ }
55
+ ```
56
+
57
+ ## License
58
+
59
+ Apache 2.0
lra_image_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78a10b07ef7f2715b800902c09389b217f4041fbc0c7f227dac6a2c9859e2c03
3
+ size 243412475