Prithvi-Complimentary Adaptive Fusion Encoder (CAFE): unlocking full-potential for flood inundation mapping
Not on the map. Why?
Abstract (arXiv)
Geo-Foundation Models (GFMs), have proven effective in diverse downstream applications, including semantic segmentation, classification, and regression tasks. However, in case of flood mapping using Sen1Flood11 dataset as a downstream task, GFMs struggles to outperform the baseline U-Net, highlighting model's limitation in capturing critical local nuances. To address this, we present the Prithvi-Complementary Adaptive Fusion Encoder (CAFE), which integrate Prithvi GFM pretrained encoder with a parallel CNN residual branch enhanced by Convolutional Attention Modules (CAM). Prithvi-CAFE enables fast and efficient fine-tuning through adapters in Prithvi and performs multi-scale, multi-level fusion with CNN features, capturing critical local details while preserving long-range dependencies. We achieve state-of-the-art results on two comprehensive flood mapping datasets: Sen1Flood11 and FloodPlanet. On Sen1Flood11 test data, Prithvi-CAFE (IoU 83.41) outperforms the original Prithvi (IoU 82.50) and other major GFMs (TerraMind 82.90, DOFA 81.54, spectralGPT: 81.02). The improvement is even more pronounced on the hold-out test site, where Prithvi-CAFE achieves an IoU of 81.37 compared to the baseline U-Net (70.57) and original Prithvi (72.42). On FloodPlanet, Prithvi-CAFE also surpasses the baseline U-Net and other GFMs, achieving an IoU of 64.70 compared to U-Net (60.14), Terramind (62.33), DOFA (59.15) and Prithvi 2.0 (61.91). Our proposed simple yet effective Prithvi-CAFE demonstrates strong potential for improving segmentation tasks where multi-channel and multi-modal data provide complementary information and local details are critical. The code is released on hrefhttps://github.com/Sk-2103/Prithvi-CAFEPrithvi-CAFE Github
Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Builds on 24 of its 36 references that are in the map. First posted on arXiv 5 Jan 2026.
Builds on
24 of 36 references.
References · 24
- 2015Deep Residual Learning for Image Recognitioncited by 51,950
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,656
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,364
- 2023RemoteCLIP: A Vision Language Foundation Model for Remote Sensingcited by 359
- 2022Scale-MAE: A Scale-Aware Masked Autoencoder for Multiscale Geospatial Representation Learningcited by 305
- 2023SpectralGPT: Spectral Remote Sensing Foundation Modelcited by 251
- 2023RSGPT: A Remote Sensing Vision Language Model and Benchmarkcited by 204
- 2023SkySense: A Multi-Modal Remote Sensing Foundation Model Towards Universal Interpretation for Earth Observation Imagerycited by 203
- 2022SatlasPretrain: A Large-Scale Dataset for Remote Sensing Image Understandingcited by 165
- 2024Prithvi-EO-2.0: A Versatile Multi-Temporal Foundation Model for Earth Observation Applicationscited by 143
- 2023DiffusionSat: A Generative Foundation Model for Satellite Imagerycited by 130
- 2025AlphaEarth Foundations: An embedding field model for accurate and efficient global mapping from sparse label datacited by 128
12 more
- 2023GEO-Bench: Toward Foundation Models for Earth Monitoringcited by 123
- 2024SkyEyeGPT: Unifying Remote Sensing Vision-Language Tasks via Instruction Tuning with Large Language Modelcited by 112
- 2024Neural Plasticity-Inspired Multimodal Foundation Model for Earth Observationcited by 102
- 2025TerraMind: Large-Scale Generative Multimodality for Earth Observationcited by 101
- 2024PANGAEA: A Global and Inclusive Benchmark for Geospatial Foundation Modelscited by 66
- 2024Vision Foundation Models in Remote Sensing: A Surveycited by 54
- 2025Galileo: Learning Global & Local Features of Many Remote Sensing Modalitiescited by 42
- 2022TransNorm: Transformer Provides a Strong Spatial Normalization Mechanism for a Deep Segmentation Modelcited by 34
- 2025Panopticon: Advancing Any-Sensor Foundation Models for Earth Observationcited by 34
- 2025TerraTorch: The Geospatial Foundation Models Toolkitcited by 25
- 2023Assessment of a new GeoAI foundation model for flood inundation mappingcited by 14
- 2025Fine-tune Smarter, Not Harder: Parameter-Efficient Fine-Tuning for Geospatial Foundation Modelscited by 9
Built on by
Citing papers per year: 2026: 1.
1 paper.