adammnrr commited on
Commit
c51f43c
·
verified ·
1 Parent(s): 7409e2a

Model card: correct the replication section (run-order split caveat, authors' code vs port)

Browse files
Files changed (1) hide show
  1. README.md +22 -12
README.md CHANGED
@@ -84,20 +84,30 @@ Use that loss to match the reference training.
84
 
85
  ## Replication (braindecode PR #1298)
86
 
87
- Fine-tuning on the 61-word task with the paper's protocol (200 epochs, AdamW,
88
- per-word 80/10/10 split, 6 seeds × 12 subjects, mean ± SE over subjects):
 
 
 
89
 
90
- | | paper (Table 2) | braindecode, sigmoid loss | braindecode, logits |
91
  |---|---|---|---|
92
- | Du-IN (mae) | 62.70 ± 4.69 | 54.51 ± 4.77 | 50.07 ± 5.26 |
93
- | Du-IN (vqvae) | 58.24 ± 4.83 | 49.33 ± 4.42 | 49.31 ± 5.04 |
94
- | scratch | 56.29 ± 5.20 | 48.40 ± 4.98 | 47.32 ± 5.26 |
95
-
96
- The pretraining gain over scratch is reproduced (+6.1 vs +6.4 in the paper,
97
- 12/12 subjects), but absolute accuracies are about 8 points lower. The authors'
98
- own `run_cls.py` on the same released data gives the same numbers as the port
99
- (subject 002: 47.8 % mae vs 83.6 % in the paper), so the gap comes from the
100
- released data and checkpoints, not from the conversion.
 
 
 
 
 
 
 
101
 
102
  ## License
103
 
 
84
 
85
  ## Replication (braindecode PR #1298)
86
 
87
+ Fine-tuning on the 61-word task with the authors' `run_cls.py` protocol (200
88
+ epochs, AdamW, sigmoid cross-entropy, 6 seeds × 12 subjects, paper-time data
89
+ revision `056a87b7`, mean ± SE over subjects). The authors' fine-tuning loop
90
+ was run with both the original model and the braindecode port, starting from
91
+ the same weights:
92
 
93
+ | | paper (Table 2) | authors' code | braindecode port |
94
  |---|---|---|---|
95
+ | Du-IN (mae) | 62.70 ± 4.69 | 60.52 ± 4.58 | 60.16 ± 4.62 |
96
+ | Du-IN (vqvae) | 58.24 ± 4.83 | 55.10 ± 4.51 | 55.00 ± 4.51 |
97
+ | scratch | 56.29 ± 5.20 | 54.06 ± 5.12 | 53.73 ± 5.13 |
98
+
99
+ The port matches the authors' code to within 0.4 points. Training the port on
100
+ plain logits (`cross_entropy(logits)`) instead of the sigmoid loss gives 55.6
101
+ for mae.
102
+
103
+ **Split caveat.** The authors' split concatenates a subject's runs in
104
+ `os.listdir` order and puts the first 80 % of each word's trials in the
105
+ training set. Validation and test therefore come almost entirely from the
106
+ **last run listed**, and that run depends on the filesystem. If it is the
107
+ chronologically last session (`run6`), accuracy drops sharply. On subject 002
108
+ (scratch, seed 42), holding out `run6` gives 43–45 %, while holding out any
109
+ other run gives 71–78 %. A sorted run order underestimates the paper by about
110
+ 8 points on average.
111
 
112
  ## License
113