mlsanityData to 7 Oct 2026

Structure-Guided Masked Autoencoders for Ultra-High Resolution Scientific Image Understanding

arXiv 2609.30682 ↗First posted 25 Sep 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Self-supervised pre-training with Vision Transformers, including Masked Autoencoders (MAE), is difficult to apply to gigapixel scientific images. Random masking is poorly matched to the structured, multi-scale morphology of scientific data, while uniform tokenization produces prohibitively long sequences that make O(N2) attention impractical. We propose SGMA, a structure-guided masked autoencoding framework for ultra-high-resolution scientific images. SGMA couples two components: a content-adaptive quadtree tokenizer that compresses gigapixel images into a fixed-length sequence, and a structure-conditioned masking process that biases reconstruction toward spatially informative regions. To stabilize this process across scales, we introduce Damped Accumulation (DA), which aggregates signal-dependent responses across the tree into a structure canvas used to guide masking. The resulting pre-training task preserves fine microstructure while remaining compatible with standard ViT encoders and MAE-style reconstruction. Across electron microscopy, whole-slide optical microscopy, and X-ray CT datasets, SGMA consistently outperforms MAE baselines. It achieves 95.68% Dice on the 8K x 8K x 28K SpringXCT dataset, improving over the same-architecture MAE baseline by +13.00 points, and 83.21% Dice on the 32K2 WSI PAIP dataset, improving by +16.84 points, while providing up to a 24.8x inference speedup.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 45 of its 64 references that are in the map. First posted on arXiv 25 Sep 2026.

Builds on

45 of 64 references.

References · 45

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  5. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
  6. 2023Segment Anythingcited by 8,158
  7. 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
  8. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  9. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  10. 2020Longformer: The Long-Document Transformercited by 3,362
  11. 2024SAM 2: Segment Anything in Images and Videoscited by 2,784
  12. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
33 more
  1. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
  2. 2020Big Bird: Transformers for Longer Sequencescited by 1,673
  3. 2020Reformer: The Efficient Transformercited by 1,658
  4. 2020Rethinking Attention with Performerscited by 1,584
  5. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  6. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  7. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
  8. 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
  9. 2021CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classificationcited by 806
  10. 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
  11. 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
  12. 2023Ring Attention with Blockwise Transformers for Near-Infinite Contextcited by 448
  13. 2021Self-Supervised Pre-Training of Swin Transformers for 3D Medical Image Analysiscited by 418
  14. 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 412
  15. 2022Scaling Vision Transformers to Gigapixel Images via Hierarchical Self-Supervised Learningcited by 357
  16. 2023Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistlescited by 287
  17. 2023DeepSpeed Ulysses: System Optimizations for Enabling Training of Extreme Long Sequence Transformer Modelscited by 239
  18. 2021CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentationcited by 227
  19. 2019The state of the art in kidney and kidney tumor segmentation in contrast-enhanced CT imaging: Results of the KiTS19 Challengecited by 188
  20. 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 179
  21. 2025DeepSeek-OCR: Contexts Optical Compressioncited by 174
  22. 2023MEGABYTE: Predicting Million-byte Sequences with Multiscale Transformerscited by 164
  23. 2021Self-attention Does Not Need O(n2) Memorycited by 152
  24. 2022QuadTree Attention for Vision Transformerscited by 129
  25. 2023Specformer: Spectral Graph Neural Networks Meet Transformerscited by 102
  26. 2022Monarch: Expressive Structured Matrices for Efficient and Accurate Trainingcited by 100
  27. 2021SparseBERT: Rethinking the Importance Analysis in Self-attentioncited by 46
  28. 2021Octree Transformer: Autoregressive 3D Shape Generation on Hierarchically Structured Sequencescited by 38
  29. 2024SPFormer: Enhancing Vision Transformer with Superpixel Representationcited by 13
  30. 2025ORBIT-2: Scaling Exascale Vision Foundation Models for Weather and Climate Downscalingcited by 10
  31. 2021Three-level Hierarchical Transformer Networks for Long-sequence and Multiple Clinical Documents Classificationcited by 6
  32. 2022Topology-Preserving Segmentation Network: A Deep Learning Segmentation Framework for Connected Componentcited by 4
  33. 2024Adaptive Patching for High-resolution Image Segmentation with Transformerscited by 4

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026AFFMAE: Scalable Vision Pre-Training for High-Resolution Microscopy Segmentation on Desktop Hardware
  2. 2022A simple, efficient and scalable contrastive masked autoencoder for learning visual representations
  3. 2026Masked Swingers: Harnessing Data Augmentation to Advance Autoencoders for Self-Supervised Learning
  4. 2025Fine-structure Preserved Real-world Image Super-resolution via Transfer VAE Training
  5. 2025UltraImageGen: Efficient Ultra-High-Resolution Image Generation with Hierarchical Local Attention
  6. 2023LMD: Faster Image Reconstruction with Latent Masking Diffusion
  7. 2025HMAR: Efficient Hierarchical Masked Auto-Regressive Image Generation
  8. 2026Efficient One-Step Diffusion Restoration Model with Compact Token Compression and Linear Attention