Improve model card metadata and link paper

README.md

@@ -1,15 +1,15 @@
 ---
+library_name: pytorch
 license: mit
+pipeline_tag: other
 tags:
-  - wildfire
-  - geospatial
-  - weather
-  - earth-observation
-  - foundation-models
-  - evaluation
-  - pytorch
-pipeline_tag: image-segmentation
-library_name: pytorch
+- wildfire
+- geospatial
+- weather
+- earth-observation
+- foundation-models
+- evaluation
+- pytorch
 pretty_name: WildFIRE-FM
 ---
 
@@ -22,6 +22,7 @@ pretty_name: WildFIRE-FM
 <p align="center"><b>Five seeded PyTorch checkpoints, paper-aligned evaluation artifacts, final figure previews, and source-data access notes for 12-hour wildfire occupancy prediction on a 5 km California grid.</b></p>
 
 <p>
+  <a href="https://huggingface.co/papers/2605.18911"><img alt="Paper" src="https://img.shields.io/badge/Paper-HF%20Paper%20Page-blue?style=flat-square"></a>
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/resolve/main/paper/wildfire_fm_evaluation_contracts.pdf"><img alt="Paper PDF" src="https://img.shields.io/badge/Paper-compiled%20PDF-0F5C5F?style=flat-square"></a>
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/models/wildfire_fm/checkpoints"><img alt="Checkpoints" src="https://img.shields.io/badge/Checkpoints-5%20seeds-C66B2D?style=flat-square"></a>
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/blob/main/models/wildfire_fm/modeling_unet.py"><img alt="PyTorch" src="https://img.shields.io/badge/PyTorch-model%20code-EE4C2C?style=flat-square&logo=pytorch&logoColor=white"></a>
@@ -35,7 +36,7 @@ pretty_name: WildFIRE-FM
 </p>
 
 <p>
-  <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/blob/main/paper/wildfire_fm_evaluation_contracts.pdf"><b>Read the paper PDF</b></a> ·
+  <a href="https://huggingface.co/papers/2605.18911"><b>Read the paper</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/models/wildfire_fm/checkpoints"><b>Download checkpoints</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#quick-start"><b>Load the model</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#task-contract-snapshot"><b>Inspect task-contract results</b></a>
@@ -66,8 +67,8 @@ pretty_name: WildFIRE-FM
 <table>
   <tr>
     <td width="33%" valign="top">
-      <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM/blob/main/paper/wildfire_fm_evaluation_contracts.pdf">Paper PDF</a></h3>
-      <p>Read the compiled manuscript that defines the fixed-contract wildfire transfer setting and reports the final results.</p>
+      <h3><a href="https://huggingface.co/papers/2605.18911">Paper Page</a></h3>
+      <p>Visit the paper page on Hugging Face for the full manuscript, discussions, and related research.</p>
     </td>
     <td width="33%" valign="top">
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/models/wildfire_fm/checkpoints">Model Checkpoints</a></h3>
@@ -116,8 +117,6 @@ Wildfire Earth-FM transfer scores depend strongly on the contract used for compa
 
 WildFIRE-FM is the in-region reference model used in those comparisons. It is trained for 12-hour gridded wildfire occupancy on a California 5 km grid, then evaluated under the same task-specific contracts as the transferred Earth-FM backbones.
 
-This Hub repository is model-first. It provides the released weights, model-loading code, compact result summaries, paper previews, and source-data access notes needed to inspect the release boundary before running larger local reproductions.
-
 ---
 
 ## Key Features
@@ -126,94 +125,8 @@ This Hub repository is model-first. It provides the released weights, model-load
 - **Five seeded checkpoints:** Seeds `1`, `7`, `42`, `99`, and `123` are released with manifest hashes.
 - **Fixed-contract artifacts:** Compact summaries cover matching-rule, head-selection, and task-form comparisons from the final paper.
 - **Source-data aware release:** Provider links and data roles are documented for each public resource used by the study.
-- **Compiled paper PDF:** The readable manuscript is available directly from the Hub release.
 - **Paper-aligned previews:** Figure assets summarize selection regret, supporting-task rank changes, and primary-task rank changes.
 
-<table>
-  <tr>
-    <td width="50%" valign="top">
-      <h3>Reference model for wildfire transfer</h3>
-      <p>WildFIRE-FM combines regional weather, active-fire supervision, and static fuel, canopy, housing, and population context on the same grid used by the occupancy contract.</p>
-    </td>
-    <td width="50%" valign="top">
-      <h3>Contract-first evaluation artifacts</h3>
-      <p>The released summaries keep metric roles separate across decision, overlap, ranking, regression, and supporting-task checks.</p>
-    </td>
-  </tr>
-  <tr>
-    <td width="50%" valign="top">
-      <h3>Local-data reproducibility path</h3>
-      <p>Full raw-data reruns use provider-hosted sources and local feature caches. The Hub release documents those inputs and provides lightweight artifact checks.</p>
-    </td>
-    <td width="50%" valign="top">
-      <h3>Final-paper result previews</h3>
-      <p>The model card highlights the paper's fixed-output, fixed-feature, and task-form findings without requiring manuscript source files.</p>
-    </td>
-  </tr>
-</table>
-
----
-
-## Visual Tour
-
-<p align="center">
-  <img src="https://huggingface.co/RAI-Lab/Wildfire-FM/resolve/main/assets/release_contents.svg" alt="Release contents" width="92%">
-</p>
-
-**Release contents.** The Hub repository is organized around weights, model code, source-data notes, compact artifacts, and final-paper visual previews.
-
-<p align="center">
-  <img src="https://huggingface.co/RAI-Lab/Wildfire-FM/resolve/main/assets/selection_regret_final.png" alt="Head-selection regret" width="88%">
-</p>
-
-**Head-selection regret.** Choosing a lightweight head by a ranking metric can lose decision performance under the same frozen features.
-
-<p align="center">
-  <img src="https://huggingface.co/RAI-Lab/Wildfire-FM/resolve/main/assets/supporting_rank_map_final.png" alt="Supporting task rank map" width="88%">
-</p>
-
-**Supporting-task rank map.** Backbone ordering changes across burned area, analog retrieval, smoke PM2.5, and extreme heat task contracts.
-
-<p align="center">
-  <img src="https://huggingface.co/RAI-Lab/Wildfire-FM/resolve/main/assets/primary_rank_change_final.png" alt="Primary rank changes" width="88%">
-</p>
-
-**Primary-task rank changes.** Primary-task rankings change under fixed matching-rule and task-form choices.
-
----
-
-## At A Glance
-
-| Signal | Release value |
-|---|---|
-| Core task | 12-hour gridded wildfire occupancy prediction |
-| Region/grid | California regional grid, 5 km, EPSG:5070 |
-| Input channels | 16 channels: weather fields, validity masks, static fuel/canopy/exposure layers |
-| Architecture | Compact U-Net with occupancy and auxiliary spatial-support heads |
-| Training split | June-August 2024 train, September 2024 validation, October 2024 test |
-| Released checkpoints | 5 seeded PyTorch checkpoints |
-| Data access path | Source links and roles documented in `data_sources/` |
-| Paper artifact path | Compiled PDF in `paper/`; manuscript source managed in the authors' paper workspace |
-
----
-
-## Hugging Face Release
-
-This Hub repository is a model release with paper-aligned reproducibility artifacts.
-
-| Folder | Purpose | Example |
-|---|---|---|
-| `models/wildfire_fm/` | Model code, manifests, checkpoint metadata | `models/wildfire_fm/modeling_unet.py` |
-| `models/wildfire_fm/checkpoints/` | Five seeded PyTorch checkpoints | `seed_1/best_firms_prauc.pt` |
-| `paper/` | Compiled paper PDF | `wildfire_fm_evaluation_contracts.pdf` |
-| `assets/` | Hub-page visuals and final-paper previews | `selection_regret_final.png` |
-| `paper_outputs/figures/` | Selected final-paper figure PDFs | `fig_rank_heatmap1.pdf` |
-| `artifacts/results/` | Sanitized compact numeric summaries | `selection_regret_scope_sweep_20260505.csv` |
-| `data_sources/` | Public data-source roles and access notes | `DATA_SOURCES.md` |
-| `experiments/` | Sanitized raw-rerun references and Slurm template | `slurm/submit_template.sbatch` |
-
-Full local reruns require users to prepare the provider-hosted raw data and comparator assets described in `data_sources/`.
-
 ---
 
 ## Quick Start
@@ -248,20 +161,12 @@ model.load_state_dict(state)
 model.eval()
 ```
 
-Verify the public release artifacts:
-
-```bash
-python3 scripts/reproduce_paper_outputs.py
-```
-
 The checkpoint expects the same 16-channel gridded input described in the paper and in `data_sources/DATA_SOURCES.md`.
 
 ---
 
 ## Task-Contract Snapshot
 
-The card below reports the best final-paper mean for each displayed task contract, with the winning backbone named in the final column.
-
 | Task contract | Best final-paper mean | Winner |
 |---|---:|---|
 | Occupancy union F1 | `60.1506 ± 7.5865` percent | ClimaX |
@@ -271,60 +176,6 @@ The card below reports the best final-paper mean for each displayed task contrac
 | Smoke PM2.5 RMSE | `4.4403 ± 0.0488`, lower is better | AlphaEarth |
 | Extreme-heat RMSE-C | `0.2179 ± 0.0043`, lower is better | WildFIRE-FM |
 
-The release provides sanitized CSV/JSON summaries used to audit displayed values.
-
----
-
-## Data Sources
-
-The study uses public or provider-hosted resources. Source roles and access paths are documented in `data_sources/DATA_SOURCES.md`.
-
-- NOAA HRRR fields for regional weather inputs.
-- NASA FIRMS active-fire detections for occupancy supervision.
-- LANDFIRE fuel and canopy layers for static landscape context.
-- Wildfire Risk to Communities housing density and LandScan population for exposure context.
-- WFIGS and MTBS event-level resources for burned-area and analog tasks.
-- External Earth-FM and backbone assets for comparator features.
-
----
-
-## Repository Layout
-
-```text
-WildFIRE-FM/
-  README.md
-  LICENSE
-  requirements.txt
-  models/wildfire_fm/
-    modeling_unet.py
-    checkpoint_manifest.json
-    checkpoints/seed_*/best_firms_prauc.pt
-  paper/
-    wildfire_fm_evaluation_contracts.pdf
-  assets/
-    wildfire_fm_model_card.svg
-    release_contents.svg
-    selection_regret_final.png
-    supporting_rank_map_final.png
-    primary_rank_change_final.png
-  paper_outputs/figures/
-    matching.pdf
-    fig_task_contract_tiles.pdf
-    fig_selection_regret_scatter.pdf
-    fig_rank_heatmap1.pdf
-  artifacts/
-    manifests/
-    results/
-  data_sources/
-    DATA_SOURCES.md
-  experiments/
-    raw_reference/
-    slurm/
-  scripts/
-    audit_release.py
-    reproduce_paper_outputs.py
-```
-
 ---
 
 ## How To Cite
@@ -340,28 +191,6 @@ If you use WildFIRE-FM, the released checkpoints, the fixed-contract evaluation
 }
 ```
 
-The citation will be updated with arXiv metadata after the preprint is public.
-
----
-
-## News
-
-- **2026-05-18** - The model card was reorganized with a larger navigation panel and final-paper visual previews.
-- **2026-05-13** - The Hub release added the compiled paper PDF without checklist pages.
-- **2026-05-13** - Five seeded WildFIRE-FM checkpoints were uploaded with SHA-256 metadata.
-
----
-
-## Scope
-
-WildFIRE-FM is scoped to the paper's California regional grid and fixed-contract wildfire transfer comparisons. Use in a new region or grid should start with local preprocessing, validation, and contract-specific evaluation. The release is designed for research inspection, artifact-level checking, and model-loading experiments rather than operational alerting.
-
----
-
-## License
-
-This release is provided under the MIT license. External data and comparator assets remain governed by their original providers and terms of use.
-
 ---
 
-WildFIRE-FM is released to make wildfire Earth-FM transfer comparisons easier to inspect, reproduce at the artifact level, and evaluate under explicit contracts.
+WildFIRE-FM is released to make wildfire Earth-FM transfer comparisons easier to inspect, reproduce at the artifact level, and evaluate under explicit contracts.