Publish openmhc-fedformer-imp

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 FEDformer.pypots filter=lfs diff=lfs merge=lfs -text
+model.pypots filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,11 +1,12 @@
 ---
 license: openrail
 library_name: openmhc
+pipeline_tag: time-series-forecasting
 tags:
-- openmhc
-- wearables
-- time-series
-- imputation
+  - openmhc
+  - wearables
+  - time-series
+  - imputation
 ---
 
 # OpenMHC FEDformer
@@ -31,14 +32,24 @@ Requires the matching optional extras: `pip install 'openmhc[pypots,hf]'`.
 Pin a specific revision with the `@` suffix:
 
 ```python
-FEDformerImputer.from_release("hf://MyHeartCounts/openmhc-fedformer-imp@v1.0")
+FEDformerImputer.from_release("hf://MyHeartCounts/openmhc-fedformer-imp@v1.1")
 ```
 
 ## Provenance
 
-- Origin: W&B artifact `MHC_Dataset/mhc-pypots-fedformer/fedformer:v31`
-- Validation: **val MAE 0.1706 (epoch 12)**
-- Architecture hyperparameters are pinned in `openmhc_manifest.json`.
+- Retrained 2026-06-02 with the `fourier_modes.json` sidecar — training-time
+  stochastic Fourier basis indices (`mode_select="random"`) are now persisted
+  and reloaded at inference so the model uses the same basis it was trained
+  with. The pre-fix v1.0 bundle without the sidecar produced wrong
+  reconstructions; **v1.1 is the version evaluated in the openmhc paper
+  bootstrap**.
+- Architecture hyperparameters are pinned in `openmhc_manifest.json`
+  (`n_steps=1440, n_features=19, n_layers=2, d_model=512, n_heads=8,
+  d_ffn=128, moving_avg_window_size=25, dropout=0.1, modes=32,
+  mode_select="random", variant="Fourier"`).
+- Files: `model.pypots` (PyPOTS checkpoint), `fourier_modes.json` (sidecar
+  for random Fourier indices), `normalization_stats.json` (per-channel
+  z-score stats), `openmhc_manifest.json` (spec v2).
 
 ## License
 

fourier_modes.json

@@ -0,0 +1,36 @@
+{
+  "encoder.encoder.attn_layers.0.attention.attention_operator": [
+    33,
+    39,
+    54,
+    63,
+    65,
+    86,
+    101,
+    120,
+    133,
+    145,
+    198,
+    210,
+    213,
+    254,
+    259,
+    290,
+    292,
+    323,
+    328,
+    340,
+    363,
+    367,
+    390,
+    417,
+    430,
+    446,
+    448,
+    453,
+    478,
+    627,
+    686,
+    706
+  ]
+}

model.pypots

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7b628d413ac6c6c848c9bf7cf4c555d5e322cb036b93bb38c8bce3d0826efb5
+size 20907786

openmhc_manifest.json

@@ -1,7 +1,7 @@
 {
-  "spec_version": 1,
+  "spec_version": 2,
   "kind": "fedformer",
-  "checkpoint": "FEDformer.pypots",
+  "checkpoint": "model.pypots",
   "normalization_stats": "normalization_stats.json",
   "arch": {
     "n_steps": 1440,
@@ -12,11 +12,45 @@
     "d_ffn": 128,
     "moving_avg_window_size": 25,
     "dropout": 0.1,
-    "variant": "Fourier",
     "modes": 32,
-    "mode_select": "random"
+    "mode_select": "random",
+    "variant": "Fourier"
   },
   "provenance": {
-    "wandb_artifact": "MHC_Dataset/mhc-pypots-fedformer/fedformer:v31"
-  }
+    "seed": 42,
+    "training": {
+      "epochs": 50,
+      "batch_size": 128,
+      "patience": 10,
+      "optimizer_lr": 0.002,
+      "weight_decay": 0.0,
+      "clip_grad_norm": null,
+      "device": "cuda",
+      "num_workers": 4
+    },
+    "model_config": {
+      "model_name": "fedformer",
+      "n_steps": 1440,
+      "n_features": 19,
+      "rnn_hidden_size": 128,
+      "n_layers": 2,
+      "d_model": 512,
+      "n_heads": 8,
+      "d_ffn": 128,
+      "dropout": 0.1,
+      "apply_nonstationary_norm": false,
+      "top_k": 5,
+      "n_kernels": 6,
+      "moving_avg_window_size": 25,
+      "individual": false,
+      "ORT_weight": 1.0,
+      "MIT_weight": 1.0,
+      "version": "Fourier",
+      "modes": 32,
+      "mode_select": "random"
+    },
+    "h5_cache_dir": "/scratch/users/schuetzn/openmhc-imputation-train/h5/5c7dd477",
+    "checkpoint_source": "/scratch/users/schuetzn/openmhc-imputation-train/runs/fedformer_20260602_031247/20260602_T031333/FEDformer.pypots"
+  },
+  "fourier_modes": "fourier_modes.json"
 }