Model card: correct the replication section (run-order split caveat, authors' code vs port)
Browse files
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
|
| 88 |
-
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
-
| | paper (Table 2) |
|
| 91 |
|---|---|---|---|
|
| 92 |
-
| Du-IN (mae) | 62.70 ± 4.69 |
|
| 93 |
-
| Du-IN (vqvae) | 58.24 ± 4.83 |
|
| 94 |
-
| scratch | 56.29 ± 5.20 |
|
| 95 |
-
|
| 96 |
-
The
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|