Upload folder using huggingface_hub
Browse files- README.md +73 -0
- checksums.json +212 -0
- classification/lc_finetune_fold0.ckpt +3 -0
- classification/lc_finetune_fold1.ckpt +3 -0
- classification/lc_finetune_fold2.ckpt +3 -0
- classification/lc_probe_fold0.ckpt +3 -0
- classification/lc_probe_fold1.ckpt +3 -0
- classification/lc_probe_fold2.ckpt +3 -0
- classification/lc_supervised_fold0.ckpt +3 -0
- classification/lc_supervised_fold1.ckpt +3 -0
- classification/lc_supervised_fold2.ckpt +3 -0
- classification/mc_finetune_fold0.ckpt +3 -0
- classification/mc_finetune_fold1.ckpt +3 -0
- classification/mc_finetune_fold2.ckpt +3 -0
- classification/mc_probe_fold0.ckpt +3 -0
- classification/mc_probe_fold1.ckpt +3 -0
- classification/mc_probe_fold2.ckpt +3 -0
- classification/mc_supervised_fold0.ckpt +3 -0
- classification/mc_supervised_fold1.ckpt +3 -0
- classification/mc_supervised_fold2.ckpt +3 -0
- classification/sc_finetune_fold0.ckpt +3 -0
- classification/sc_finetune_fold1.ckpt +3 -0
- classification/sc_finetune_fold2.ckpt +3 -0
- classification/sc_probe_fold0.ckpt +3 -0
- classification/sc_probe_fold1.ckpt +3 -0
- classification/sc_probe_fold2.ckpt +3 -0
- classification/sc_supervised_fold0.ckpt +3 -0
- classification/sc_supervised_fold1.ckpt +3 -0
- classification/sc_supervised_fold2.ckpt +3 -0
- encoders/lc.ckpt +3 -0
- encoders/mc.ckpt +3 -0
- encoders/sc.ckpt +3 -0
- manifest.yaml +282 -0
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
|