zbirobin commited on
Commit
45fc6ba
·
verified ·
1 Parent(s): 477794f

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +136 -0
README.md ADDED
@@ -0,0 +1,136 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - ecology
6
+ - multimodal
7
+ - missing-modality
8
+ - masking
9
+ - species-distribution-modeling
10
+ - image-classification
11
+ library_name: pytorch
12
+ pipeline_tag: image-classification
13
+ ---
14
+
15
+ # MIAM: Modality Imbalance-Aware Masking for Multimodal Ecological Applications
16
+
17
+ This repository hosts released checkpoints for **MIAM** from the ICLR 2026 paper:
18
+
19
+ - Paper: https://openreview.net/forum?id=oljjAkgZN4
20
+ - Project page: https://zbirobin.github.io/publications/miam/
21
+ - Source code: https://github.com/zbirobin/MIAM
22
+
23
+ MIAM is a dynamic masking strategy for multimodal ecological learning. During training, it adapts masking probabilities using modality-specific performance and learning-speed signals to reduce modality imbalance and improve robustness to missing inputs.
24
+
25
+ ## Model details
26
+
27
+ - **Model family**: Multimodal ecological models trained with MIAM and masking baselines
28
+ - **Modalities covered**:
29
+ - GeoPlant (MaskSDM): satellite + environmental tabular + climate time series
30
+ - TaxaBench-8k: location + environmental tabular + natural image + audio + satellite
31
+ - SatBird: satellite + environmental tabular
32
+ - **Framework**: PyTorch
33
+
34
+ ## Available files
35
+
36
+ ### GeoPlant (MaskSDM)
37
+
38
+ - `geoplant_miam.pt`
39
+ - `geoplant_opm.pt`
40
+ - `geoplant_dropout.pt`
41
+ - `geoplant_constant.pt`
42
+ - `geoplant_dirichlet.pt`
43
+ - `geoplant_uniform.pt`
44
+ - `geoplant_pretraining_miam.pt`
45
+ - `geoplant_pretraining_opm.pt`
46
+ - `geoplant_pretraining_dropout.pt`
47
+ - `geoplant_pretraining_constant.pt`
48
+ - `geoplant_pretraining_dirichlet.pt`
49
+ - `geoplant_pretraining_uniform.pt`
50
+
51
+ ### TaxaBench-8k
52
+
53
+ - `taxabench_miam.pt`
54
+ - `taxabench_dirichlet.pt`
55
+ - `taxabench_dropout.pt`
56
+ - `taxabench_opm.pt`
57
+ - `taxabench_uniform.pt`
58
+ - `taxabench_embeds_loc_env_img_aud_sat.pt`
59
+ - `taxabench_num_species_10.csv`
60
+
61
+ ### SatBird
62
+
63
+ - `satbird_miam.pth`
64
+ - `satbird_opm.pth`
65
+ - `satbird_dropout.pth`
66
+ - `satbird_dirichlet.pth`
67
+
68
+ ## Quick start
69
+
70
+ ### Download with `hf` CLI
71
+
72
+ ```bash
73
+ hf download zbirobin/MIAM geoplant_miam.pt
74
+ hf download zbirobin/MIAM taxabench_miam.pt
75
+ hf download zbirobin/MIAM satbird_miam.pth
76
+ ```
77
+
78
+ ### Download in Python
79
+
80
+ ```python
81
+ from huggingface_hub import hf_hub_download
82
+ import torch
83
+
84
+ ckpt_path = hf_hub_download(repo_id="zbirobin/MIAM", filename="geoplant_miam.pt")
85
+ state = torch.load(ckpt_path, map_location="cpu")
86
+ ```
87
+
88
+ ## Intended use
89
+
90
+ - Research on multimodal ecological modeling
91
+ - Robustness to missing-modality settings
92
+ - Benchmark comparison of masking strategies
93
+ - Modality contribution analysis and interpretability studies
94
+
95
+ ## Out-of-scope use
96
+
97
+ - Safety-critical ecological decisions without domain validation
98
+ - Deployment outside the data distributions represented in GeoPlant, TaxaBench-8k, and SatBird
99
+ - Applications requiring calibrated uncertainty without additional validation
100
+
101
+ ## Training and evaluation data
102
+
103
+ - **GeoPlant**: species distribution modeling benchmark used in the paper
104
+ - **TaxaBench-8k**: multimodal species classification benchmark
105
+ - **SatBird-USA-summer**: bird species distribution benchmark
106
+
107
+ Please follow dataset licenses, terms of use, and any access restrictions from the original providers.
108
+
109
+ ## Limitations
110
+
111
+ - Transferability across regions/taxa/modalities may be limited
112
+
113
+ ## Reproducibility
114
+
115
+ Use the benchmark READMEs in the source repository for exact folder structure, commands, and evaluation scripts:
116
+
117
+ - `maskSDM/README.md`
118
+ - `taxabench/README.md`
119
+ - `satbird/README.md`
120
+
121
+ ## Citation
122
+
123
+ ```bibtex
124
+ @inproceedings{
125
+ zbinden2026miam,
126
+ title={{MIAM}: Modality Imbalance-Aware Masking for Multimodal Ecological Applications},
127
+ author={Robin Zbinden and Wesley Monteith-Finas and Gencer Sumbul and Nina van Tiel and Chiara Vanalli and Devis Tuia},
128
+ booktitle={International Conference on Learning Representations (ICLR)},
129
+ year={2026},
130
+ url={https://openreview.net/forum?id=oljjAkgZN4}
131
+ }
132
+ ```
133
+
134
+ ## Contact
135
+
136
+ For issues and questions, please open a ticket in the source repository.