arXiv:2512.00872cs.CVcs.AI2025-12被引 3

用3D CT数据自监督预训练,打造无需微调的通用医学影像模型

TAP-CT: 3D Task-Agnostic Pretraining of Computed Tomography Foundation Models

  • 将ViT和DINOv2改造为适配3D体数据的架构,加入深度感知设计
  • 在10.5万例自建CT数据上预训练,冻结后在多个任务中表现稳健
  • 开源全部模型与代码,为医疗影像研究提供低资源基准

现有医学领域基础模型常需大量微调或依赖高资源解码器,且多数编码器预训练目标偏向特定任务。为此,本文提出针对CT基础模型的3D任务无关预训练方法(TAP-CT):对Vision Transformers(ViTs)和DINOv2进行简单而有效的改造,直接在3D CT体数据上实现可扩展的自监督预训练。通过优化局部嵌入、位置编码及体增广策略,使模型具备深度感知能力,同时保持原架构简洁性。在包含10.5万例体积数据的大规模数据集上预训练后,得到稳定且鲁棒的冻结特征表示,在下游任务中表现出强泛化能力。为促进透明度与可复现性,并为未来医学影像研究建立强大、低资源基线,所有预训练模型、实验配置及下游基准代码将在https://huggingface.co/fomofo/tap-ct-b-3d公开发布。

原文摘要 · Abstract (English)

Existing foundation models (FMs) in the medical domain often require extensive fine-tuning or rely on training resource-intensive decoders, while many existing encoders are pretrained with objectives biased toward specific tasks. This illustrates a need for a strong, task-agnostic foundation model that requires minimal fine-tuning beyond feature extraction. In this work, we introduce a suite of task-agnostic pretraining of CT foundation models (TAP-CT): a simple yet effective adaptation of Vision Transformers (ViTs) and DINOv2 for volumetric data, enabling scalable self-supervised pretraining directly on 3D CT volumes. Our approach incorporates targeted modifications to patch embeddings, positional encodings, and volumetric augmentations, making the architecture depth-aware while preserving the simplicity of the underlying architectures. We show that large-scale 3D pretraining on an extensive in-house CT dataset (105K volumes) yields stable, robust frozen representations that generalize strongly across downstream tasks. To promote transparency and reproducibility, and to establish a powerful, low-resource baseline for future research in medical imaging, we will release all pretrained models, experimental configurations, and downstream benchmark code at https://huggingface.co/fomofo/tap-ct-b-3d.

3D医学影像自监督学习基础模型CT预训练

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