NEMESIS: Noise-suppressed Efficient MAE with Enhanced Superpatch Integration Strategy
Abstract
Volumetric CT imaging is essential for clinical diagnosis, yet annotating 3D volumes is expensive and time-consuming, motivating self-supervised learning (SSL) from unlabeled data. However, applying SSL to 3D CT remains challenging due to the high memory cost of full-volume transformers and the anisotropic spatial structure of CT data, which is not well captured by conventional masking strategies. We propose NEMESIS, a masked autoencoder (MAE) framework that operates on local 128x128x128 superpatches, enabling memory-efficient training while preserving anatomical detail. NEMESIS introduces three key components: (i) noise-enhanced reconstruction as a pretext task, (ii) Masked Anatomical Transformer Blocks (MATB) that perform dual-masking through parallel plane-wise and axis-wise token removal, and (iii) NEMESIS Tokens (NT) for cross-scale context aggregation. On the BTCV multi-organ classification benchmark, NEMESIS with a frozen backbone and a linear classifier achieves a mean AUROC of 0.9633, surpassing fully fine-tuned SuPreM (0.9493) and VoCo (0.9387). Under a low-label regime with only 10% of available annotations, it retains an AUROC of 0.9075, demonstrating strong label efficiency. Furthermore, the superpatch-based design reduces computational cost to 31.0 GFLOPs per forward pass, compared to 985.8 GFLOPs for the full-volume baseline, providing a scalable and robust foundation for 3D medical imaging.
Cite
@article{arxiv.2604.01612,
title = {NEMESIS: Noise-suppressed Efficient MAE with Enhanced Superpatch Integration Strategy},
author = {Kyeonghun Kim and Hyeonseok Jung and Youngung Han and Hyunsu Go and Eunseob Choi and Seongbin Park and Junsu Lim and Jiwon Yang and Sumin Lee and Insung Hwang and Ken Ying-Kai Liao and Nam-Joon Kim},
journal= {arXiv preprint arXiv:2604.01612},
year = {2026}
}
Comments
5 pages, 5 figures, 5 tables