Liornis commited on
Commit
ab1b5ca
·
verified ·
1 Parent(s): acdb5bc

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags: [freezing-of-gait, parkinsons, accelerometer, self-supervised, mae, time-series]
4
+ pipeline_tag: time-series-classification
5
+ ---
6
+
7
+ # FORGE — FOG Representation via Generative Encoding
8
+
9
+ Self-supervised spectral-temporal encoders for **Freezing of Gait (FOG)** detection from a
10
+ single lower-back accelerometer. Pretrained by masked autoencoding on ~21M unlabeled home
11
+ windows; evaluated zero-shot on the external FogAtHome cohort.
12
+
13
+ **Headline:** the MC probe reaches clinical-grade agreement with expert video annotation —
14
+ **ICC(%TF) = 0.909 [0.87, 0.94]**, zero-shot, one IMU.
15
+
16
+ ## Released weights
17
+
18
+ ### Pretrained FORGE encoders (the backbones)
19
+ | Context | Window (frames) | File | Params |
20
+ |---|---|---|---|
21
+ | LC | 1000 | `encoders/lc.ckpt` | 14,147,072 |
22
+ | MC | 500 | `encoders/mc.ckpt` | 14,147,072 |
23
+ | SC | 200 | `encoders/sc.ckpt` | 12,918,272 |
24
+
25
+ ### Downstream classification checkpoints (128-patient DeFOG, 3-fold CV)
26
+ | File | Context | Phase | Fold |
27
+ |---|---|---|---|
28
+ | `classification/lc_probe_fold0.ckpt` | lc | probe | 0 |
29
+ | `classification/lc_probe_fold1.ckpt` | lc | probe | 1 |
30
+ | `classification/lc_probe_fold2.ckpt` | lc | probe | 2 |
31
+ | `classification/mc_probe_fold0.ckpt` | mc | probe | 0 |
32
+ | `classification/mc_probe_fold1.ckpt` | mc | probe | 1 |
33
+ | `classification/mc_probe_fold2.ckpt` | mc | probe | 2 |
34
+ | `classification/sc_probe_fold0.ckpt` | sc | probe | 0 |
35
+ | `classification/sc_probe_fold1.ckpt` | sc | probe | 1 |
36
+ | `classification/sc_probe_fold2.ckpt` | sc | probe | 2 |
37
+ | `classification/lc_finetune_fold0.ckpt` | lc | finetune | 0 |
38
+ | `classification/lc_finetune_fold1.ckpt` | lc | finetune | 1 |
39
+ | `classification/lc_finetune_fold2.ckpt` | lc | finetune | 2 |
40
+ | `classification/mc_finetune_fold0.ckpt` | mc | finetune | 0 |
41
+ | `classification/mc_finetune_fold1.ckpt` | mc | finetune | 1 |
42
+ | `classification/mc_finetune_fold2.ckpt` | mc | finetune | 2 |
43
+ | `classification/sc_finetune_fold0.ckpt` | sc | finetune | 0 |
44
+ | `classification/sc_finetune_fold1.ckpt` | sc | finetune | 1 |
45
+ | `classification/sc_finetune_fold2.ckpt` | sc | finetune | 2 |
46
+ | `classification/lc_supervised_fold0.ckpt` | lc | supervised | 0 |
47
+ | `classification/lc_supervised_fold1.ckpt` | lc | supervised | 1 |
48
+ | `classification/lc_supervised_fold2.ckpt` | lc | supervised | 2 |
49
+ | `classification/mc_supervised_fold0.ckpt` | mc | supervised | 0 |
50
+ | `classification/mc_supervised_fold1.ckpt` | mc | supervised | 1 |
51
+ | `classification/mc_supervised_fold2.ckpt` | mc | supervised | 2 |
52
+ | `classification/sc_supervised_fold0.ckpt` | sc | supervised | 0 |
53
+ | `classification/sc_supervised_fold1.ckpt` | sc | supervised | 1 |
54
+ | `classification/sc_supervised_fold2.ckpt` | sc | supervised | 2 |
55
+
56
+ ## Usage
57
+ Checkpoints are slimmed PyTorch Lightning checkpoints (weights + config; optimizer state
58
+ stripped). Each keeps the `state_dict` and the `hyper_parameters["config"]` Pydantic config
59
+ used to rebuild the model — the same fields the evaluation pipeline reads.
60
+
61
+ ```python
62
+ import torch
63
+ ckpt = torch.load("encoders/mc.ckpt", map_location="cpu", weights_only=False)
64
+ state_dict = ckpt["state_dict"] # encoder weights
65
+ config = ckpt["hyper_parameters"]["config"] # Config object to rebuild the model
66
+ meta = ckpt.get("forge_meta") # name / context / phase / fold
67
+ ```
68
+
69
+ Reproduce every paper number with the companion repo's `reproduce-evaluations` skill (see
70
+ `manifest.yaml`, shipped in this repo).
71
+ Data: [Liornis/fog-dataset](https://huggingface.co/datasets/Liornis/fog-dataset).
72
+
73
+ License: **MIT**.
checksums.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "hf_path": "encoders/lc.ckpt",
4
+ "name": "encoder_lc_mae",
5
+ "sha256": "0192f32dc76dfb590c1b71ae0aa8d23d7ce2fe65cf509cdd0cdfb0b437c3cf7b",
6
+ "raw_bytes": 200821965,
7
+ "slim_bytes": 67705421
8
+ },
9
+ {
10
+ "hf_path": "encoders/mc.ckpt",
11
+ "name": "encoder_mc_mae",
12
+ "sha256": "c9d6a47782b76eba0f3294f51d3360566eb23b13402b5a95c9bf4eeab11a9e37",
13
+ "raw_bytes": 200771149,
14
+ "slim_bytes": 67654413
15
+ },
16
+ {
17
+ "hf_path": "encoders/sc.ckpt",
18
+ "name": "encoder_sc_mae",
19
+ "sha256": "32e613c08fbe804ef39f720785c10b78874d9aa8e3eeba1d44e99be6b9fd1733",
20
+ "raw_bytes": 178675149,
21
+ "slim_bytes": 60323213
22
+ },
23
+ {
24
+ "hf_path": "classification/lc_probe_fold0.ckpt",
25
+ "name": "lc_probe_all128_fold0",
26
+ "sha256": "16e2fddd092c9cd4648e097ae733c4336d95a116b4a0e9a669435bf8d571a730",
27
+ "raw_bytes": 103464011,
28
+ "slim_bytes": 72934877
29
+ },
30
+ {
31
+ "hf_path": "classification/lc_probe_fold1.ckpt",
32
+ "name": "lc_probe_all128_fold1",
33
+ "sha256": "e7469a5e63082e0af95de2ea0fc92b336b4c566fe40713f827710b7d91fd0e45",
34
+ "raw_bytes": 103465035,
35
+ "slim_bytes": 72935901
36
+ },
37
+ {
38
+ "hf_path": "classification/lc_probe_fold2.ckpt",
39
+ "name": "lc_probe_all128_fold2",
40
+ "sha256": "c9e648d6cd4d29aa07d9a73109d629b7f1bb03beca369168b87aad8a707811d4",
41
+ "raw_bytes": 103466827,
42
+ "slim_bytes": 72937565
43
+ },
44
+ {
45
+ "hf_path": "classification/mc_probe_fold0.ckpt",
46
+ "name": "mc_probe_all128_fold0",
47
+ "sha256": "ee40afd5b420889fe05ce8b590ebaa7d90cfb084cf0a00e712855a1d0ace8b89",
48
+ "raw_bytes": 103470283,
49
+ "slim_bytes": 72941213
50
+ },
51
+ {
52
+ "hf_path": "classification/mc_probe_fold1.ckpt",
53
+ "name": "mc_probe_all128_fold1",
54
+ "sha256": "1163dc85e6d4008aaea64bcadac18702ef06df1349a0294fc209d4ea63e0fb9d",
55
+ "raw_bytes": 103468107,
56
+ "slim_bytes": 72939165
57
+ },
58
+ {
59
+ "hf_path": "classification/mc_probe_fold2.ckpt",
60
+ "name": "mc_probe_all128_fold2",
61
+ "sha256": "a1bb440346b56b456a5465749bf9bf0ccf900a6db9f44184698b80868ace2f77",
62
+ "raw_bytes": 103476043,
63
+ "slim_bytes": 72946973
64
+ },
65
+ {
66
+ "hf_path": "classification/sc_probe_fold0.ckpt",
67
+ "name": "sc_probe_all128_fold0",
68
+ "sha256": "c5a3929bc6910820b8eb00f5612d3cce08cb78b96141f075f8d78613dabc8b52",
69
+ "raw_bytes": 98568843,
70
+ "slim_bytes": 68039709
71
+ },
72
+ {
73
+ "hf_path": "classification/sc_probe_fold1.ckpt",
74
+ "name": "sc_probe_all128_fold1",
75
+ "sha256": "d0a6fee80746a5a6b3f5b3ed3d60d9c78e95b5e43cef62f0c27824592e7278b4",
76
+ "raw_bytes": 98557195,
77
+ "slim_bytes": 68028061
78
+ },
79
+ {
80
+ "hf_path": "classification/sc_probe_fold2.ckpt",
81
+ "name": "sc_probe_all128_fold2",
82
+ "sha256": "fa901bcf00efdf1240be9850fc8332721bc1243f69ab5e321f6bbfbf87cbae4b",
83
+ "raw_bytes": 98568971,
84
+ "slim_bytes": 68039837
85
+ },
86
+ {
87
+ "hf_path": "classification/lc_finetune_fold0.ckpt",
88
+ "name": "lc_finetune_all128_fold0",
89
+ "sha256": "a7cd3505a120701c34865dbe1f5384c70ef7e20a7d81858698baad7be151cb7a",
90
+ "raw_bytes": 216683859,
91
+ "slim_bytes": 72935339
92
+ },
93
+ {
94
+ "hf_path": "classification/lc_finetune_fold1.ckpt",
95
+ "name": "lc_finetune_all128_fold1",
96
+ "sha256": "bad6363bf0a8faea4881722ed250e3a31357ef9a44618403bc8649d3ccd81408",
97
+ "raw_bytes": 216684883,
98
+ "slim_bytes": 72936363
99
+ },
100
+ {
101
+ "hf_path": "classification/lc_finetune_fold2.ckpt",
102
+ "name": "lc_finetune_all128_fold2",
103
+ "sha256": "45b4437ff36e86d52b58334d0950fb3ce1ecd38c282663a67416457ffaefafed",
104
+ "raw_bytes": 216685139,
105
+ "slim_bytes": 72936491
106
+ },
107
+ {
108
+ "hf_path": "classification/mc_finetune_fold0.ckpt",
109
+ "name": "mc_finetune_all128_fold0",
110
+ "sha256": "030db123e537bbf4423f0bc436fedc44b9f9737e3e0c08687898ee5050c6c604",
111
+ "raw_bytes": 216684051,
112
+ "slim_bytes": 72935467
113
+ },
114
+ {
115
+ "hf_path": "classification/mc_finetune_fold1.ckpt",
116
+ "name": "mc_finetune_all128_fold1",
117
+ "sha256": "b9afb63d06c9dd855ff0f8f5c6bf52957742b1c3fa1312d81adada477aaa1448",
118
+ "raw_bytes": 216688147,
119
+ "slim_bytes": 72939563
120
+ },
121
+ {
122
+ "hf_path": "classification/mc_finetune_fold2.ckpt",
123
+ "name": "mc_finetune_all128_fold2",
124
+ "sha256": "2d31a6b248af1b434df8bf3df23fdc45a095b0e14265045f49dfa412f26dca63",
125
+ "raw_bytes": 216689939,
126
+ "slim_bytes": 72941227
127
+ },
128
+ {
129
+ "hf_path": "classification/sc_finetune_fold0.ckpt",
130
+ "name": "sc_finetune_all128_fold0",
131
+ "sha256": "781b6fe75061711dd5c542971ea093d109016368b018d766c4b61b140fae8d1a",
132
+ "raw_bytes": 201939859,
133
+ "slim_bytes": 68021675
134
+ },
135
+ {
136
+ "hf_path": "classification/sc_finetune_fold1.ckpt",
137
+ "name": "sc_finetune_all128_fold1",
138
+ "sha256": "cd99a4649acb611df19c7de5da45c27c0a840618e890beec8831916834663cc3",
139
+ "raw_bytes": 201940499,
140
+ "slim_bytes": 68022315
141
+ },
142
+ {
143
+ "hf_path": "classification/sc_finetune_fold2.ckpt",
144
+ "name": "sc_finetune_all128_fold2",
145
+ "sha256": "612c0a897a06bb4aabbf0b6e95ce0d47e8694e900cda0e4b92be1898c25957fa",
146
+ "raw_bytes": 201939987,
147
+ "slim_bytes": 68021803
148
+ },
149
+ {
150
+ "hf_path": "classification/lc_supervised_fold0.ckpt",
151
+ "name": "lc_supervised_all128_fold0",
152
+ "sha256": "708866f418fd7a2423c7a1cd10007c6bbeb245f80348da5ad79baf0c84623130",
153
+ "raw_bytes": 216684371,
154
+ "slim_bytes": 72935455
155
+ },
156
+ {
157
+ "hf_path": "classification/lc_supervised_fold1.ckpt",
158
+ "name": "lc_supervised_all128_fold1",
159
+ "sha256": "7574b4353ff7a949da30c8c8d647c93052291f3b792ddbe70943ce23ce0bd58c",
160
+ "raw_bytes": 216685395,
161
+ "slim_bytes": 72936479
162
+ },
163
+ {
164
+ "hf_path": "classification/lc_supervised_fold2.ckpt",
165
+ "name": "lc_supervised_all128_fold2",
166
+ "sha256": "92c74d44b05650c541cf45d95690a41b3d110c935756dada368fea94590210e4",
167
+ "raw_bytes": 216687059,
168
+ "slim_bytes": 72938143
169
+ },
170
+ {
171
+ "hf_path": "classification/mc_supervised_fold0.ckpt",
172
+ "name": "mc_supervised_all128_fold0",
173
+ "sha256": "5fdadbf5c60b492153e861c6c9af92d324940ebd424e8783d3a37b4ed94b7e1b",
174
+ "raw_bytes": 216690643,
175
+ "slim_bytes": 72941727
176
+ },
177
+ {
178
+ "hf_path": "classification/mc_supervised_fold1.ckpt",
179
+ "name": "mc_supervised_all128_fold1",
180
+ "sha256": "5a87f4857c4b203136dbafc7c196169cc8fda1801754a5bdf174df510fb37905",
181
+ "raw_bytes": 216688467,
182
+ "slim_bytes": 72939679
183
+ },
184
+ {
185
+ "hf_path": "classification/mc_supervised_fold2.ckpt",
186
+ "name": "mc_supervised_all128_fold2",
187
+ "sha256": "67e2c5aa05f04781e37b9b08aa5f8f2f956e1c2fc4142c3d4ad4522cae8c66db",
188
+ "raw_bytes": 216696403,
189
+ "slim_bytes": 72947487
190
+ },
191
+ {
192
+ "hf_path": "classification/sc_supervised_fold0.ckpt",
193
+ "name": "sc_supervised_all128_fold0",
194
+ "sha256": "75ab4d25a42beaf6c62b1b9f8044c0bee7d17c4b69a1dafa7417946e79972eec",
195
+ "raw_bytes": 201958739,
196
+ "slim_bytes": 68040223
197
+ },
198
+ {
199
+ "hf_path": "classification/sc_supervised_fold1.ckpt",
200
+ "name": "sc_supervised_all128_fold1",
201
+ "sha256": "ba69eb9c94df4badca96a2f50d963d07c89042cd437b1ceef95f09c5e90d3afb",
202
+ "raw_bytes": 201947091,
203
+ "slim_bytes": 68028575
204
+ },
205
+ {
206
+ "hf_path": "classification/sc_supervised_fold2.ckpt",
207
+ "name": "sc_supervised_all128_fold2",
208
+ "sha256": "2dff36a618e479127c05bc1478a68763a6fd07b0c0223a71f353806ff1f61bf2",
209
+ "raw_bytes": 201958995,
210
+ "slim_bytes": 68040479
211
+ }
212
+ ]
classification/lc_finetune_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7cd3505a120701c34865dbe1f5384c70ef7e20a7d81858698baad7be151cb7a
3
+ size 72935339
classification/lc_finetune_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bad6363bf0a8faea4881722ed250e3a31357ef9a44618403bc8649d3ccd81408
3
+ size 72936363
classification/lc_finetune_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45b4437ff36e86d52b58334d0950fb3ce1ecd38c282663a67416457ffaefafed
3
+ size 72936491
classification/lc_probe_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16e2fddd092c9cd4648e097ae733c4336d95a116b4a0e9a669435bf8d571a730
3
+ size 72934877
classification/lc_probe_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7469a5e63082e0af95de2ea0fc92b336b4c566fe40713f827710b7d91fd0e45
3
+ size 72935901
classification/lc_probe_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9e648d6cd4d29aa07d9a73109d629b7f1bb03beca369168b87aad8a707811d4
3
+ size 72937565
classification/lc_supervised_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:708866f418fd7a2423c7a1cd10007c6bbeb245f80348da5ad79baf0c84623130
3
+ size 72935455
classification/lc_supervised_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7574b4353ff7a949da30c8c8d647c93052291f3b792ddbe70943ce23ce0bd58c
3
+ size 72936479
classification/lc_supervised_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92c74d44b05650c541cf45d95690a41b3d110c935756dada368fea94590210e4
3
+ size 72938143
classification/mc_finetune_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:030db123e537bbf4423f0bc436fedc44b9f9737e3e0c08687898ee5050c6c604
3
+ size 72935467
classification/mc_finetune_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9afb63d06c9dd855ff0f8f5c6bf52957742b1c3fa1312d81adada477aaa1448
3
+ size 72939563
classification/mc_finetune_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d31a6b248af1b434df8bf3df23fdc45a095b0e14265045f49dfa412f26dca63
3
+ size 72941227
classification/mc_probe_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee40afd5b420889fe05ce8b590ebaa7d90cfb084cf0a00e712855a1d0ace8b89
3
+ size 72941213
classification/mc_probe_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1163dc85e6d4008aaea64bcadac18702ef06df1349a0294fc209d4ea63e0fb9d
3
+ size 72939165
classification/mc_probe_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1bb440346b56b456a5465749bf9bf0ccf900a6db9f44184698b80868ace2f77
3
+ size 72946973
classification/mc_supervised_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fdadbf5c60b492153e861c6c9af92d324940ebd424e8783d3a37b4ed94b7e1b
3
+ size 72941727
classification/mc_supervised_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a87f4857c4b203136dbafc7c196169cc8fda1801754a5bdf174df510fb37905
3
+ size 72939679
classification/mc_supervised_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67e2c5aa05f04781e37b9b08aa5f8f2f956e1c2fc4142c3d4ad4522cae8c66db
3
+ size 72947487
classification/sc_finetune_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:781b6fe75061711dd5c542971ea093d109016368b018d766c4b61b140fae8d1a
3
+ size 68021675
classification/sc_finetune_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd99a4649acb611df19c7de5da45c27c0a840618e890beec8831916834663cc3
3
+ size 68022315
classification/sc_finetune_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:612c0a897a06bb4aabbf0b6e95ce0d47e8694e900cda0e4b92be1898c25957fa
3
+ size 68021803
classification/sc_probe_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5a3929bc6910820b8eb00f5612d3cce08cb78b96141f075f8d78613dabc8b52
3
+ size 68039709
classification/sc_probe_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0a6fee80746a5a6b3f5b3ed3d60d9c78e95b5e43cef62f0c27824592e7278b4
3
+ size 68028061
classification/sc_probe_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa901bcf00efdf1240be9850fc8332721bc1243f69ab5e321f6bbfbf87cbae4b
3
+ size 68039837
classification/sc_supervised_fold0.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75ab4d25a42beaf6c62b1b9f8044c0bee7d17c4b69a1dafa7417946e79972eec
3
+ size 68040223
classification/sc_supervised_fold1.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba69eb9c94df4badca96a2f50d963d07c89042cd437b1ceef95f09c5e90d3afb
3
+ size 68028575
classification/sc_supervised_fold2.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2dff36a618e479127c05bc1478a68763a6fd07b0c0223a71f353806ff1f61bf2
3
+ size 68040479
encoders/lc.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0192f32dc76dfb590c1b71ae0aa8d23d7ce2fe65cf509cdd0cdfb0b437c3cf7b
3
+ size 67705421
encoders/mc.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9d6a47782b76eba0f3294f51d3360566eb23b13402b5a95c9bf4eeab11a9e37
3
+ size 67654413
encoders/sc.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32e613c08fbe804ef39f720785c10b78874d9aa8e3eeba1d44e99be6b9fd1733
3
+ size 60323213
manifest.yaml ADDED
@@ -0,0 +1,282 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ meta:
2
+ paper: FORGE (draft v0.3)
3
+ headline_model: mc_probe_all128_ensemble
4
+ threshold_protocol: defog_val_pr11
5
+ license: MIT
6
+ hf_weights_repo: Liornis/forge-fog
7
+ hf_dataset_repo: Liornis/fog-dataset
8
+ encoders:
9
+ lc:
10
+ name: encoder_lc_mae
11
+ hf_path: encoders/lc.ckpt
12
+ local: checkpoints/mae/sleek-monkey-199/last.ckpt
13
+ window_frames: 1000
14
+ vit_depth: 4
15
+ params: 14147072
16
+ role: FORGE backbone (long context, 10s)
17
+ mc:
18
+ name: encoder_mc_mae
19
+ hf_path: encoders/mc.ckpt
20
+ local: checkpoints/mae/mae_medcontext_daily/mae_medcontext_daily/last.ckpt
21
+ window_frames: 500
22
+ vit_depth: 4
23
+ params: 14147072
24
+ role: FORGE backbone (medium context, 5s)
25
+ sc:
26
+ name: encoder_sc_mae
27
+ hf_path: encoders/sc.ckpt
28
+ local: checkpoints/mae/mae_shortcontext_vit4_daily/mae_shortcontext_vit4_daily/last.ckpt
29
+ window_frames: 200
30
+ vit_depth: 4
31
+ params: 12918272
32
+ role: FORGE backbone (short context, 2s)
33
+ classification:
34
+ - name: lc_probe_all128_fold0
35
+ context: lc
36
+ phase: probe
37
+ fold: 0
38
+ hf_path: classification/lc_probe_fold0.ckpt
39
+ local: checkpoints/classification/soft_probe_lc_all128_fold0/last.ckpt
40
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
41
+ headline: false
42
+ - name: lc_probe_all128_fold1
43
+ context: lc
44
+ phase: probe
45
+ fold: 1
46
+ hf_path: classification/lc_probe_fold1.ckpt
47
+ local: checkpoints/classification/soft_probe_lc_all128_fold1/last.ckpt
48
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
49
+ headline: false
50
+ - name: lc_probe_all128_fold2
51
+ context: lc
52
+ phase: probe
53
+ fold: 2
54
+ hf_path: classification/lc_probe_fold2.ckpt
55
+ local: checkpoints/classification/soft_probe_lc_all128_fold2/last.ckpt
56
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
57
+ headline: false
58
+ - name: mc_probe_all128_fold0
59
+ context: mc
60
+ phase: probe
61
+ fold: 0
62
+ hf_path: classification/mc_probe_fold0.ckpt
63
+ local: checkpoints/classification/soft_probe_mc_all128_fold0/last.ckpt
64
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
65
+ headline: true
66
+ - name: mc_probe_all128_fold1
67
+ context: mc
68
+ phase: probe
69
+ fold: 1
70
+ hf_path: classification/mc_probe_fold1.ckpt
71
+ local: checkpoints/classification/soft_probe_mc_all128_fold1/last.ckpt
72
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
73
+ headline: true
74
+ - name: mc_probe_all128_fold2
75
+ context: mc
76
+ phase: probe
77
+ fold: 2
78
+ hf_path: classification/mc_probe_fold2.ckpt
79
+ local: checkpoints/classification/soft_probe_mc_all128_fold2/last.ckpt
80
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
81
+ headline: true
82
+ - name: sc_probe_all128_fold0
83
+ context: sc
84
+ phase: probe
85
+ fold: 0
86
+ hf_path: classification/sc_probe_fold0.ckpt
87
+ local: checkpoints/classification/soft_probe_sc_all128_fold0/last.ckpt
88
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
89
+ headline: false
90
+ - name: sc_probe_all128_fold1
91
+ context: sc
92
+ phase: probe
93
+ fold: 1
94
+ hf_path: classification/sc_probe_fold1.ckpt
95
+ local: checkpoints/classification/soft_probe_sc_all128_fold1/last.ckpt
96
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
97
+ headline: false
98
+ - name: sc_probe_all128_fold2
99
+ context: sc
100
+ phase: probe
101
+ fold: 2
102
+ hf_path: classification/sc_probe_fold2.ckpt
103
+ local: checkpoints/classification/soft_probe_sc_all128_fold2/last.ckpt
104
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
105
+ headline: false
106
+ - name: lc_finetune_all128_fold0
107
+ context: lc
108
+ phase: finetune
109
+ fold: 0
110
+ hf_path: classification/lc_finetune_fold0.ckpt
111
+ local: checkpoints/classification/soft_finetune_lc_all128_fold0/last.ckpt
112
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
113
+ headline: false
114
+ - name: lc_finetune_all128_fold1
115
+ context: lc
116
+ phase: finetune
117
+ fold: 1
118
+ hf_path: classification/lc_finetune_fold1.ckpt
119
+ local: checkpoints/classification/soft_finetune_lc_all128_fold1/last.ckpt
120
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
121
+ headline: false
122
+ - name: lc_finetune_all128_fold2
123
+ context: lc
124
+ phase: finetune
125
+ fold: 2
126
+ hf_path: classification/lc_finetune_fold2.ckpt
127
+ local: checkpoints/classification/soft_finetune_lc_all128_fold2/last.ckpt
128
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
129
+ headline: false
130
+ - name: mc_finetune_all128_fold0
131
+ context: mc
132
+ phase: finetune
133
+ fold: 0
134
+ hf_path: classification/mc_finetune_fold0.ckpt
135
+ local: checkpoints/classification/soft_finetune_mc_all128_fold0/last.ckpt
136
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
137
+ headline: false
138
+ - name: mc_finetune_all128_fold1
139
+ context: mc
140
+ phase: finetune
141
+ fold: 1
142
+ hf_path: classification/mc_finetune_fold1.ckpt
143
+ local: checkpoints/classification/soft_finetune_mc_all128_fold1/last.ckpt
144
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
145
+ headline: false
146
+ - name: mc_finetune_all128_fold2
147
+ context: mc
148
+ phase: finetune
149
+ fold: 2
150
+ hf_path: classification/mc_finetune_fold2.ckpt
151
+ local: checkpoints/classification/soft_finetune_mc_all128_fold2/last.ckpt
152
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
153
+ headline: false
154
+ - name: sc_finetune_all128_fold0
155
+ context: sc
156
+ phase: finetune
157
+ fold: 0
158
+ hf_path: classification/sc_finetune_fold0.ckpt
159
+ local: checkpoints/classification/soft_finetune_sc_all128_fold0/last.ckpt
160
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
161
+ headline: false
162
+ - name: sc_finetune_all128_fold1
163
+ context: sc
164
+ phase: finetune
165
+ fold: 1
166
+ hf_path: classification/sc_finetune_fold1.ckpt
167
+ local: checkpoints/classification/soft_finetune_sc_all128_fold1/last.ckpt
168
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
169
+ headline: false
170
+ - name: sc_finetune_all128_fold2
171
+ context: sc
172
+ phase: finetune
173
+ fold: 2
174
+ hf_path: classification/sc_finetune_fold2.ckpt
175
+ local: checkpoints/classification/soft_finetune_sc_all128_fold2/last.ckpt
176
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
177
+ headline: false
178
+ - name: lc_supervised_all128_fold0
179
+ context: lc
180
+ phase: supervised
181
+ fold: 0
182
+ hf_path: classification/lc_supervised_fold0.ckpt
183
+ local: checkpoints/classification/soft_supervised_lc_all128_fold0/last.ckpt
184
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
185
+ headline: false
186
+ - name: lc_supervised_all128_fold1
187
+ context: lc
188
+ phase: supervised
189
+ fold: 1
190
+ hf_path: classification/lc_supervised_fold1.ckpt
191
+ local: checkpoints/classification/soft_supervised_lc_all128_fold1/last.ckpt
192
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
193
+ headline: false
194
+ - name: lc_supervised_all128_fold2
195
+ context: lc
196
+ phase: supervised
197
+ fold: 2
198
+ hf_path: classification/lc_supervised_fold2.ckpt
199
+ local: checkpoints/classification/soft_supervised_lc_all128_fold2/last.ckpt
200
+ config: classification/spectral_patch_mae_lc_valid_defog_soft
201
+ headline: false
202
+ - name: mc_supervised_all128_fold0
203
+ context: mc
204
+ phase: supervised
205
+ fold: 0
206
+ hf_path: classification/mc_supervised_fold0.ckpt
207
+ local: checkpoints/classification/soft_supervised_mc_all128_fold0/last.ckpt
208
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
209
+ headline: false
210
+ - name: mc_supervised_all128_fold1
211
+ context: mc
212
+ phase: supervised
213
+ fold: 1
214
+ hf_path: classification/mc_supervised_fold1.ckpt
215
+ local: checkpoints/classification/soft_supervised_mc_all128_fold1/last.ckpt
216
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
217
+ headline: false
218
+ - name: mc_supervised_all128_fold2
219
+ context: mc
220
+ phase: supervised
221
+ fold: 2
222
+ hf_path: classification/mc_supervised_fold2.ckpt
223
+ local: checkpoints/classification/soft_supervised_mc_all128_fold2/last.ckpt
224
+ config: classification/spectral_patch_mae_mc_valid_defog_soft
225
+ headline: false
226
+ - name: sc_supervised_all128_fold0
227
+ context: sc
228
+ phase: supervised
229
+ fold: 0
230
+ hf_path: classification/sc_supervised_fold0.ckpt
231
+ local: checkpoints/classification/soft_supervised_sc_all128_fold0/last.ckpt
232
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
233
+ headline: false
234
+ - name: sc_supervised_all128_fold1
235
+ context: sc
236
+ phase: supervised
237
+ fold: 1
238
+ hf_path: classification/sc_supervised_fold1.ckpt
239
+ local: checkpoints/classification/soft_supervised_sc_all128_fold1/last.ckpt
240
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
241
+ headline: false
242
+ - name: sc_supervised_all128_fold2
243
+ context: sc
244
+ phase: supervised
245
+ fold: 2
246
+ hf_path: classification/sc_supervised_fold2.ckpt
247
+ local: checkpoints/classification/soft_supervised_sc_all128_fold2/last.ckpt
248
+ config: classification/spectral_patch_mae_sc_valid_defog_soft
249
+ headline: false
250
+ datasets:
251
+ fogathome:
252
+ hf_subdir: fogathome
253
+ role: external_structured
254
+ prevalence: 0.313
255
+ n_patients: 12
256
+ dailyliving:
257
+ hf_subdir: fogathome_dailyliving
258
+ role: external_naturalistic
259
+ prevalence: 0.011
260
+ n_patients: 11
261
+ kaggle:
262
+ hf_subdir: kaggle_labeled
263
+ role: in_distribution
264
+ results:
265
+ - id: fogathome_icc_tf
266
+ model: mc_probe_all128_ensemble
267
+ value: 0.909
268
+ ci:
269
+ - 0.87
270
+ - 0.94
271
+ figure: fig01
272
+ script: source .venv/bin/activate && python scripts/eval/compute_icc_thresholds.py
273
+ --suffix _all128 --out logs/RESULTS_all128_icc.md
274
+ - id: fogathome_seg_auc
275
+ model: mc_probe_all128_ensemble
276
+ value: 0.908
277
+ script: source .venv/bin/activate && python scripts/eval/eval_comprehensive.py --datasets
278
+ fogathome --contexts mc --models probe_all128 --output logs/comprehensive_eval_all128.csv
279
+ --cache-dir logs/comprehensive_eval_cache
280
+ - id: fogathome_cls_ap
281
+ model: mc_probe_all128_ensemble
282
+ value: 0.823