Link reference data generation pipeline
Browse files
README.md
CHANGED
|
@@ -19,16 +19,13 @@ weights associated
|
|
| 19 |
with [*CSI-MAE: A Masked Autoencoder-based Channel Foundation Model*](https://arxiv.org/abs/2601.03789).
|
| 20 |
|
| 21 |
- Code: [GREAT-ISAC/CSI-MAE](https://github.com/GREAT-ISAC/CSI-MAE)
|
| 22 |
-
|
| 23 |
-
Released by **[GREAT Wireless AI](https://github.com/GREAT-ISAC)**. The
|
| 24 |
-
`GREAT-ISAC` GitHub handle is retained for continuity with existing papers and
|
| 25 |
-
community links.
|
| 26 |
- Paper: [arXiv:2601.03789](https://arxiv.org/abs/2601.03789)
|
| 27 |
|
| 28 |
## Weight files
|
| 29 |
|
| 30 |
The release contains epoch-300 Base and Large checkpoints pretrained on the
|
| 31 |
-
simulated Sionna/3GPP channel
|
| 32 |
experimental checkpoints. None of the files contains an optimizer, AMP scaler,
|
| 33 |
epoch/resume state, training data, or a downstream task head.
|
| 34 |
|
|
@@ -39,7 +36,7 @@ epoch/resume state, training data, or a downstream task head.
|
|
| 39 |
| `csi-mae-large.safetensors` | Large model-only pre-trained weights | Recommended Large weight for safe standalone loading and feature extraction |
|
| 40 |
| `csi-mae-large.pth` | Large model-only PyTorch checkpoint with a `model` key | Large compatibility weight for existing scripts |
|
| 41 |
|
| 42 |
-
These are foundation-model pre-training weights, not final checkpoints for
|
| 43 |
positioning, channel feedback, or channel extrapolation. The corresponding
|
| 44 |
downstream architecture must be initialized from these weights and then
|
| 45 |
adapted or fine-tuned. Exact SHA-256 values are recorded in `manifest.json`.
|
|
@@ -112,9 +109,10 @@ python load_pretrained.py \
|
|
| 112 |
The Base and Large models were pre-trained on simulated Sionna/3GPP CSI. The
|
| 113 |
separate DeepMIMO experiments are not the source of these published weights.
|
| 114 |
Generated training arrays are not included in this model repository. A
|
| 115 |
-
**reproducible data-generation pipeline**
|
| 116 |
-
|
| 117 |
-
|
|
|
|
| 118 |
|
| 119 |
## Intended use
|
| 120 |
|
|
@@ -137,7 +135,7 @@ definitions.
|
|
| 137 |
|
| 138 |
## License
|
| 139 |
|
| 140 |
-
The original CSI-MAE code, these model weights, and the
|
| 141 |
data-generation scripts are released under the
|
| 142 |
[Creative Commons Attribution-NonCommercial 4.0 International](https://creativecommons.org/licenses/by-nc/4.0/)
|
| 143 |
license (**CC BY-NC 4.0**). Attribution is required and commercial use is not
|
|
|
|
| 19 |
with [*CSI-MAE: A Masked Autoencoder-based Channel Foundation Model*](https://arxiv.org/abs/2601.03789).
|
| 20 |
|
| 21 |
- Code: [GREAT-ISAC/CSI-MAE](https://github.com/GREAT-ISAC/CSI-MAE)
|
| 22 |
+
- Reference data generation: [GREAT-ISAC/Channel-Simulation-Data](https://github.com/GREAT-ISAC/Channel-Simulation-Data)
|
|
|
|
|
|
|
|
|
|
| 23 |
- Paper: [arXiv:2601.03789](https://arxiv.org/abs/2601.03789)
|
| 24 |
|
| 25 |
## Weight files
|
| 26 |
|
| 27 |
The release contains epoch-300 Base and Large checkpoints pretrained on the
|
| 28 |
+
simulated Sionna/3GPP channel data. These are not the separate DeepMIMO
|
| 29 |
experimental checkpoints. None of the files contains an optimizer, AMP scaler,
|
| 30 |
epoch/resume state, training data, or a downstream task head.
|
| 31 |
|
|
|
|
| 36 |
| `csi-mae-large.safetensors` | Large model-only pre-trained weights | Recommended Large weight for safe standalone loading and feature extraction |
|
| 37 |
| `csi-mae-large.pth` | Large model-only PyTorch checkpoint with a `model` key | Large compatibility weight for existing scripts |
|
| 38 |
|
| 39 |
+
These are channel-foundation-model pre-training weights, not final checkpoints for
|
| 40 |
positioning, channel feedback, or channel extrapolation. The corresponding
|
| 41 |
downstream architecture must be initialized from these weights and then
|
| 42 |
adapted or fine-tuned. Exact SHA-256 values are recorded in `manifest.json`.
|
|
|
|
| 109 |
The Base and Large models were pre-trained on simulated Sionna/3GPP CSI. The
|
| 110 |
separate DeepMIMO experiments are not the source of these published weights.
|
| 111 |
Generated training arrays are not included in this model repository. A
|
| 112 |
+
**reproducible, model-compatible reference data-generation pipeline** is
|
| 113 |
+
available in [Channel Simulation Data](https://github.com/GREAT-ISAC/Channel-Simulation-Data).
|
| 114 |
+
Its committed Sionna configuration is a runnable reference example; it does
|
| 115 |
+
not reconstruct the complete checkpoint training data.
|
| 116 |
|
| 117 |
## Intended use
|
| 118 |
|
|
|
|
| 135 |
|
| 136 |
## License
|
| 137 |
|
| 138 |
+
The original CSI-MAE code, these model weights, and the repository-owned
|
| 139 |
data-generation scripts are released under the
|
| 140 |
[Creative Commons Attribution-NonCommercial 4.0 International](https://creativecommons.org/licenses/by-nc/4.0/)
|
| 141 |
license (**CC BY-NC 4.0**). Attribution is required and commercial use is not
|