tpys commited on
Commit
dc45ef3
·
1 Parent(s): 1cff2ab

Describe diagnostic feedback without misrepresenting channel flow

Browse files

Clarify that radiation-flux and precipitation channels stay in the fixed recurrent tensor but are zeroed before the next step. Document the zeroing stage in the rollout sequence.

Constraint: Preserve the model's fixed 85-channel recurrent interface

Rejected: Describe diagnostic fields as absent from feedback | the channels remain present even though their values are zeroed

Confidence: high

Scope-risk: narrow

Directive: Distinguish channel presence from value propagation when documenting diagnostic fields

Tested: git diff --cached --check; reviewed the data-details and rollout descriptions together

Not-tested: Inference execution; documentation-only correction

Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -145,8 +145,9 @@ FuXi-2.1 operates on 85 channels per time step: 65 pressure-level channels
145
  (5 variables × 13 levels) and 20 surface channels, plus static forcings supplied as
146
  constant inputs. Most channels are **prognostic**: they are both input and output
147
  channels and are fed back during rollout. Radiation fluxes and total precipitation are
148
- **diagnostic** outputs from a dedicated decoder head; they are predicted but not fed
149
- back as inputs.
 
150
 
151
  **Pressure-level variables** (5 vars × 13 levels = 65 channels):
152
 
@@ -202,8 +203,10 @@ fuxi-2.1/
202
 
203
  1. **Load** pre-normalized input (2 frames at t-6h and t)
204
  2. **Rollout** autoregressively: model outputs next 2-frame state, last frame is the prediction
205
- 3. **Denormalize** each prediction: `output = output * std + mean`, then `expm1` for precipitation
206
- 4. **Save** each step as NetCDF with geographic coordinates
 
 
207
 
208
  The recurrence state stays on GPU throughout the rollout (no CPU round-trip per step).
209
 
 
145
  (5 variables × 13 levels) and 20 surface channels, plus static forcings supplied as
146
  constant inputs. Most channels are **prognostic**: they are both input and output
147
  channels and are fed back during rollout. Radiation fluxes and total precipitation are
148
+ **diagnostic** outputs from a dedicated decoder head. Their channels remain present in
149
+ the recurrent input to preserve the fixed 85-channel layout, but their values are zeroed
150
+ before feedback so accumulated diagnostic fields are not propagated to the next step.
151
 
152
  **Pressure-level variables** (5 vars × 13 levels = 65 channels):
153
 
 
203
 
204
  1. **Load** pre-normalized input (2 frames at t-6h and t)
205
  2. **Rollout** autoregressively: model outputs next 2-frame state, last frame is the prediction
206
+ 3. **Prepare recurrence**: retain the prediction for output, then zero radiation-flux and
207
+ precipitation channels in the recurrent state before the next step
208
+ 4. **Denormalize** each prediction: `output = output * std + mean`, then `expm1` for precipitation
209
+ 5. **Save** each step as NetCDF with geographic coordinates
210
 
211
  The recurrence state stays on GPU throughout the rollout (no CPU round-trip per step).
212