arXiv:2604.00514cs.CVcs.AI2026-04中稿 · ICEIC 2026被引 1

用3D超块自编码器提升医学影像自监督学习效果

MAESIL: Masked Autoencoder for Enhanced Self-supervised Medical Image Learning

  • 用3D超块替代2D切片,保留体积结构上下文
  • 在三个公开CT数据集上PSNR和SSIM显著优于现有方法
  • 适合需要高质量预训练的3D医学图像任务

三维(3D)医学成像(如计算机断层扫描,CT)深度学习模型的训练受限于标注数据稀缺。尽管自然图像预训练普遍,但存在显著领域偏移,制约性能。自监督学习(SSL)在无标签医学数据上的应用成为有效解决方案,但主流框架常将3D扫描视为独立2D切片的集合,忽略关键的轴向一致性与3D结构上下文。为此,我们提出面向增强自监督医学图像学习的自编码器(MAESIL),一种专为捕捉3D结构信息设计的新框架。核心创新在于‘超块’——基于3D块的输入单元,在保持3D上下文的同时兼顾计算效率。该框架将体积划分为超块,并采用双掩码策略的3D自编码器,学习全面的空间表征。我们在三个不同大型公开CT数据集上验证了该方法。实验结果表明,相较于AE、VAE和VQ-VAE等现有方法,MAESIL在关键重建指标(如PSNR和SSIM)上均有显著提升,确立其为3D医学成像任务中稳健且实用的预训练方案。

原文摘要 · Abstract (English)

Training deep learning models for three-dimensional (3D) medical imaging, such as Computed Tomography (CT), is fundamentally challenged by the scarcity of labeled data. While pre-training on natural images is common, it results in a significant domain shift, limiting performance. Self-Supervised Learning (SSL) on unlabeled medical data has emerged as a powerful solution, but prominent frameworks often fail to exploit the inherent 3D nature of CT scans. These methods typically process 3D scans as a collection of independent 2D slices, an approach that fundamentally discards critical axial coherence and the 3D structural context. To address this limitation, we propose the autoencoder for enhanced self-supervised medical image learning(MAESIL), a novel self-supervised learning framework designed to capture 3D structural information efficiently. The core innovation is the 'superpatch', a 3D chunk-based input unit that balances 3D context preservation with computational efficiency. Our framework partitions the volume into superpatches and employs a 3D masked autoencoder strategy with a dual-masking strategy to learn comprehensive spatial representations. We validated our approach on three diverse large-scale public CT datasets. Our experimental results show that MAESIL demonstrates significant improvements over existing methods such as AE, VAE and VQ-VAE in key reconstruction metrics such as PSNR and SSIM. This establishes MAESIL as a robust and practical pre-training solution for 3D medical imaging tasks.

自监督学习3D医学图像自编码器CT分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。