JJiang02 commited on
Commit
f29270f
·
verified ·
1 Parent(s): ffff367

Link reference data generation pipeline

Browse files
Files changed (1) hide show
  1. README.md +8 -10
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 corpus. These are not the separate DeepMIMO
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** will be released separately with the
116
- simulator configuration, preprocessing, normalization statistics, and split
117
- definitions.
 
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 accompanying original
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