Skip to content

Repository files navigation

Leveraging Big Multitemporal Multisource Satellite Data and Artificial Intelligence for the Detection of Complex and Invisible Features - the Case of Extensive Irrigation Mapping: Swin UNETR levee detection model

Buławka, N., Orengo, H. A., Ruiz, F. L., Berganzo‐Besga, I., & Gupta, E. (2026). Leveraging Big Multitemporal Multisource Satellite Data and Artificial Intelligence for the Detection of Complex and Invisible Features: The Case of Extensive Irrigation Mapping. Archaeological Prospection, arp.70053. https://doi.org/10.1002/arp.70053

Abstract

The detection of buried or obscured archaeological features remains a central challenge in landscape archaeology, particularly in the irrigated floodplains of Mesopotamia where levees and canals formed the basis of complex agrarian systems. This study presents a deep learning–based approach for the large-scale, automated detection of ancient levees in central Iraq, integrating big multitemporal and multisource satellite datasets with advanced instance segmentation models. Datasets were assembled from multitemporal Landsat 5, Sentinel-1 SAR, Sentinel-2 multispectral imagery, and the TanDEM-X Edited DSM, combined with vegetation and moisture indices, PCA reductions of seasonal variability, and Multi-Scale Relief Model (MSRM) outputs. Training labels were generated through both threshold-based automatic extraction and detailed manual digitisation. Three architectures: U-Net, Attention U-Net, and Swin UNETR, were evaluated on datasets containing 53, 48, and 36 bands. Results demonstrate that Swin UNETR consistently outperformed other models, particularly when trained on the 48-band dataset with manually digitised levees. Unlike wide automatic annotations, which produced irregular and noisy patches, thin manual annotations yielded clearer, more linear predictions. Post-processing further refined the outputs, enabling the model to achieve pixel-level precision of 0.6555 and recall of 0.5107, and vector-level precision of 0.7554 and recall of 0.7743 after additional post-processing. While pixel-level metric scores remain modest, reflecting the irregularity of the archaeological features, the model successfully predicted levee networks across ~31,250 km², extending from the Ba’qubah region to Ad Rumaythah, with potentially correctly detected ca. 10,300 km of levees located outside the dataset. Comparative analysis with independent palaeochannel reconstructions confirmed that the model identified many of the most prominent irrigation features while avoiding misclassification of modern infrastructure. The results of the Monte Carlo simulation indicate a clear relationship between identified levees and archaeological sites of different periods, particularly Old Babylonian, Parthian, Sasanian, and Early Islamic. The results highlight both the challenges and promise of deep learning in archaeological remote sensing. Automated predictions cannot yet replace interpretative digitisation, but they provide reproducible, standardised, and scalable outputs that can accelerate archaeological mapping and support regional-scale analysis. By leveraging multitemporal, multisource datasets and advanced AI architectures, this study demonstrates a pathway towards reconstructing irrigation systems of different historical periods and landscapes. The approach opens new possibilities for documenting, preserving, and interpreting water-management legacies in some of the world’s most significant ancient landscapes.

Levee detection demo

Open demo

Demo of the model includes:

  1. Initiate libraries
  2. Download raw data (utilize Copernicus GLO30 and the trained model
  3. Predict the the levees using Swin UNETR model
  4. Postprocessing

Full project workflow

Model

Setup

The Docker container and virtual envinronment used to run all the scripts is available here. More information how to install Docker is provided on the Docker website.

Pre-processing

The project constisted of two steps. First step was to develop a Deep Learning dataset, which was developed gradually. The final version of the dataset consisted of 20GB. The final steps was the development of the deep learning model.

  1. Calculate the multisource rasters using published Google Earth Engine code
  2. Create annotations
    1. Rasterize the levee network created for the purpose of this work - code
    2. Clip (QGIS 3.34.13)
    3. Adjust pixels in multisource raster and annotations (QGIS 3.34.13)
  3. Devide the images and annotations into tiles of 96x96 pixels - code
  4. Create JSON file with structure of the dataset
    1. Create JSON for one or all rasters - code
    2. Merge JSON files if necessary - code
  5. Perform augmentations using albumentations - code
  6. Create an augmented dataset
    1. Append augmented tiles to a JSON file
    2. Shuffle the tiles within collections used to train, validate and test the model (with a data leakage prevention mechanism) - code
    3. Check if the dataset does not have data leakage between train, validation and testing - code

The dataset structure is provided with following link

Train and validate the model

The research utilizes the Unet, Attention Unet and Swin UNETR developed by MONAI

Model selection part

The selection phase of the training included separate sets od code for training Unet, Attention Unet and Swin UNETR models

  • Unet
  • Attention Unet
  • Swin UNETR

The final model

The final version of the model utilized Swin UNETR

The code used in this stage is available using following link

Predict

The code used for prediction consists of several steps

  1. Calculate the multisource rasters using published Google Earth Engine code
  2. Run the detection code
    1. Select raster
    2. Convert to NPY
    3. Predict levees
    4. Convert predictions to TIF and merge them

Post-processing

  1. Post-processing code include:
    1. Filter by size (the filter excludes small features) - opencv-python connectedComponentsWithStats(); min_area = 350; connectivity=8
    2. Closing - opencv-python Closing; kernel = 7x7

Citation

@article{bulawkaLeveragingBigMultitemporal2026,
	title = {Leveraging {Big} {Multitemporal} {Multisource} {Satellite} {Data} and {Artificial} {Intelligence} for the {Detection} of {Complex} and {Invisible} {Features}: {The} {Case} of {Extensive} {Irrigation} {Mapping}},
	issn = {1075-2196, 1099-0763},
	shorttitle = {Leveraging {Big} {Multitemporal} {Multisource} {Satellite} {Data} and {Artificial} {Intelligence} for the {Detection} of {Complex} and {Invisible} {Features}},
	url = {https://onlinelibrary.wiley.com/doi/10.1002/arp.70053},
	doi = {10.1002/arp.70053},
	language = {en},
	urldate = {2026-06-18},
	journal = {Archaeological Prospection},
	author = {Buławka, Nazarij and Orengo, Hector A. and Ruiz, Felipe Lumbreras and Berganzo‐Besga, Iban and Gupta, Ekta},
	month = jun,
	year = {2026},
	pages = {arp.70053},
}


DOI

Shield: CC BY 4.0

This work is licensed under a Creative Commons Attribution 4.0 International License.

CC BY 4.0

About

The Github repository include the code for training and predicting levees using Unet, Attention Unet and Swin UNETR deep learning models (Buławka et al. 2026)

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages