OceanMAE: A Foundation Model for Ocean Remote Sensing
Organizations: Faculty of Electrical Engineering and Computer Science, Technische Universität Berlin, Germany and with the Berlin Institute for the Foundations of Learning and Data (BIFOLD), 10623 Berlin, Germany.
Abstract
Accurate ocean mapping is essential for applications such as bathymetry estimation, seabed characterization, marine litter detection, and ecosystem monitoring. However, ocean remote sensing (RS) remains constrained by limited labeled data and by the reduced transferability of models pre-trained mainly on land-dominated Earth observation imagery. In this paper, we propose OceanMAE, an ocean-specific masked autoencoder that extends standard MAE pre-training by integrating multispectral Sentinel-2 observations with physically meaningful ocean descriptors during self-supervised learning. By incorporating these auxiliary ocean features, OceanMAE is designed to learn more informative and ocean-aware latent representations from large- scale unlabeled data. To transfer these representations to downstream applications, we further employ a modified UNet-based framework for marine segmentation and bathymetry estimation. Pre-trained on the Hydro dataset, OceanMAE is evaluated on MADOS and MARIDA for marine pollutant and debris segmentation, and on MagicBathyNet for bathymetry regression. The experiments show that OceanMAE yields the strongest gains on marine segmentation, while bathymetry benefits are competitive and task-dependent. In addition, an ablation against a standard MAE on MARIDA indicates that incorporating auxiliary ocean descriptors during pre-training improves downstream segmentation quality. These findings highlight the value of physically informed and domain-aligned self-supervised pre- training for ocean RS. Code and weights are publicly available at https://git.tu-berlin.de/joanna.stamer/SSLORS2.
Figures & tables
| Region | Model | MAE | RMSE | StdDev |
|---|---|---|---|---|
| Agia Napa | UNet-bathy [ 4 ] | 0.694 | 1.068 | 0.940 |
| OceanMAE + Random | 0.560 | 0.731 | 0.613 | |
| OceanMAE + FE | 0.536 | 0.714 | 0.594 | |
| SatMAE + FF | 0.623 | 0.812 | 0.721 | |
| FGMAE + FF | 0.651 | 0.868 | 0.712 | |
| SSL4EO + FF | 0.612 | 0.823 | 0.722 |
| Model | PA | mIoU | Macro F1 |
|---|---|---|---|
| UNet [ 3 ] | 0.69 | 0.57 | 0.69 |
| OceanMAE + Random | 0.69 | 0.57 | 0.68 |
| OceanMAE + FE | 0.72 | 0.58 | 0.69 |
| SatMAE + FF | 0.54 | 0.67 | 0.64 |
| FGMAE + FF | 0.55 | 0.68 | 0.66 |
| SSL4EO + FF | 0.51 | 0.65 | 0.64 |
| Model | PA | mIoU | Macro F1 |
|---|---|---|---|
| OceanMAE + FE (Patch-16) | 87.4 | 70.8 | 81.3 |
| OceanMAE + FF (Patch-16) | 86.5 | 67.6 | 79.3 |
| OceanMAE + FE (Patch-4) | 88.7 | 62.4 | 73.8 |
| OceanMAE + FF (Patch-4) | 87.2 | 63.4 | 75.4 |
| Model | PA | mIoU | Macro F1 |
|---|---|---|---|
| Random Forest [ 2 ] | 67.1 | 43.9 | 56.6 |
| UNet [ 2 ] | 82.9 | 51.0 | 63.8 |
| SatMAE | 85.3 | 60.0 | 71.2 |
| FGMAE | 82.6 | 65.1 | 77.1 |
| SSL4EO | 83.0 | 63.6 | 74.8 |
| MariNeXt (SOTA) [ 2 ] | 89.1 | 64.3 | 76.0 |
| Model | PA | mIoU | Macro F1 |
|---|---|---|---|
| MAE + FF | 0.48 | 0.68 | 0.59 |
| OceanMAE + FF | 0.77 | 0.61 | 0.72 |
Remote Sensing
Remote sensing uses satellite, aerial, and ground-based observations to measure and map Earth and other planetary surfaces.
See all 528 papers