Structure-Guided Masked Autoencoders for Ultra-High Resolution Scientific Image Understanding
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
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023Segment Anythingcited by 8,158
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2024SAM 2: Segment Anything in Images and Videoscited by 2,784
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
33 more
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2020Reformer: The Efficient Transformercited by 1,658
- 2020Rethinking Attention with Performerscited by 1,584
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
- 2021CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classificationcited by 806
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
- 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
- 2023Ring Attention with Blockwise Transformers for Near-Infinite Contextcited by 448
- 2021Self-Supervised Pre-Training of Swin Transformers for 3D Medical Image Analysiscited by 418
- 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 412
- 2022Scaling Vision Transformers to Gigapixel Images via Hierarchical Self-Supervised Learningcited by 357
- 2023Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistlescited by 287
- 2023DeepSpeed Ulysses: System Optimizations for Enabling Training of Extreme Long Sequence Transformer Modelscited by 239
- 2021CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentationcited by 227
- 2019The state of the art in kidney and kidney tumor segmentation in contrast-enhanced CT imaging: Results of the KiTS19 Challengecited by 188
- 2021nnFormer: Interleaved Transformer for Volumetric Segmentationcited by 179
- 2025DeepSeek-OCR: Contexts Optical Compressioncited by 174
- 2023MEGABYTE: Predicting Million-byte Sequences with Multiscale Transformerscited by 164
- 2021Self-attention Does Not Need O(n2) Memorycited by 152
- 2022QuadTree Attention for Vision Transformerscited by 129
- 2023Specformer: Spectral Graph Neural Networks Meet Transformerscited by 102
- 2022Monarch: Expressive Structured Matrices for Efficient and Accurate Trainingcited by 100
- 2021SparseBERT: Rethinking the Importance Analysis in Self-attentioncited by 46
- 2021Octree Transformer: Autoregressive 3D Shape Generation on Hierarchically Structured Sequencescited by 38
- 2024SPFormer: Enhancing Vision Transformer with Superpixel Representationcited by 13
- 2025ORBIT-2: Scaling Exascale Vision Foundation Models for Weather and Climate Downscalingcited by 10
- 2021Three-level Hierarchical Transformer Networks for Long-sequence and Multiple Clinical Documents Classificationcited by 6
- 2022Topology-Preserving Segmentation Network: A Deep Learning Segmentation Framework for Connected Componentcited by 4
- 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
- 2026AFFMAE: Scalable Vision Pre-Training for High-Resolution Microscopy Segmentation on Desktop Hardware
- 2022A simple, efficient and scalable contrastive masked autoencoder for learning visual representations
- 2026Masked Swingers: Harnessing Data Augmentation to Advance Autoencoders for Self-Supervised Learning
- 2025Fine-structure Preserved Real-world Image Super-resolution via Transfer VAE Training
- 2025UltraImageGen: Efficient Ultra-High-Resolution Image Generation with Hierarchical Local Attention
- 2023LMD: Faster Image Reconstruction with Latent Masking Diffusion
- 2025HMAR: Efficient Hierarchical Masked Auto-Regressive Image Generation
- 2026Efficient One-Step Diffusion Restoration Model with Compact Token Compression and Linear Attention