arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.13915cs.CV

关注位置很重要:重新思考3D医学自监督学习中的子体积采样

Where to Look Matters: Rethinking Sub-Volume Sampling in 3D Medical Self-supervised Learning

School of Engineering(工程学院);University of Birmingham(伯明翰大学)

Junkai Liu, Le Zhang

AI总结:

该研究针对3D医学自监督学习的子体积采样瓶颈,提出即插即用框架\name,其轻量型变体\name-lite可提升下游迁移性能,增益源于更优的观察与表征。

摘要:

3D医学图像的自监督学习(SSL)在预训练阶段通常依赖随机子体积采样。然而,随机裁剪隐含地将所有空间区域视为同等值得观察的对象,尽管解剖学先验和空间冗余性使得不同区域在表征学习中的价值本就不均等。为此,我们首先通过受控的裁剪质量操作表明,更好的子体积会带来更优的下游迁移性能,这揭示了观察策略是3D医学SSL中一个被忽视的瓶颈。受该发现启发,我们将子体积采样重新视为一个“关注位置”问题。我们提出了\textbf{\name},这是一个即插即用框架,它用由解剖学感知、信息性和多样性引导的离线粗到细探测库取代随机裁剪。我们将\name实例化为两种变体:一种是使用几何和图像统计线索的轻量型\textit{\name-lite}变体,另一种是使用冻结视觉特征的诊断型\textit{\name-feat}变体,以检验轻量型体积固有先验是否足够,或更丰富的语义线索是否能可靠地提供额外价值。在多个3D医学SSL骨干网络和下游任务上进行的大量实验表明,\name-lite相比随机采样始终能提升迁移性能。机制分析进一步将这些增益归因于更符合解剖学且更具信息性的观察、更少的空间冗余、更高效的学习动态以及更结构化的表征。

英文摘要:

Self-supervised learning on 3D medical images commonly relies on random sub-volume sampling during pretraining. However, random cropping implicitly treats all spatial regions as equally worth observing, despite anatomical priors and spatial redundancy that make regions inherently unequal in their value for representation learning. To this end, we first show through controlled crop-quality manipulations that better sub-volumes lead to better downstream transfer, revealing the observation policy as an overlooked bottleneck in 3D medical SSL. Motivated by this finding, we revisit sub-volume sampling as a \emph{where-to-look} problem. We propose \textbf{\method}, a plug-and-play framework that replaces random cropping with an offline coarse-to-fine probe bank guided by \textit{anatomy-awareness}, \textit{informativeness}, and \textit{diversity}. We instantiate \method as a lightweight \lite variant using geometric and image-statistical cues, and a diagnostic \feat variant using frozen visual features, to examine whether lightweight volume-intrinsic priors suffice or richer semantic cues reliably provide additional value. Extensive experiments across multiple 3D medical SSL backbones and downstream tasks demonstrate that \method-\lite consistently improves transfer performance over random sampling. Mechanistic analyses further attribute these gains to more anatomically plausible and informative observations, reduced spatial redundancy, more efficient learning dynamics, and more structured representations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑