arXiv:2505.21862cs.CV2025-05被引 17

直接在未标注临床数据上训练3D医学影像语言模型,提升可扩展性。

Towards Scalable Language-Image Pre-training for 3D Medical Imaging

  • 设计分层注意力机制,适配影像的切片-扫描-研究层级结构
  • 在220K脑MRI和240K头颈CT数据上实现领先性能,提升超10%
  • 适用于真实临床场景,为大规模预训练提供可行路径

当前3D医学影像(如CT、MRI)的语言-图像预训练受限于放射科医生对原始临床研究的手动标注。本文首次在未经标注的研究数据上进行预训练,更贴近实际工作流程,并推动规模化。但此类数据的独特结构对原有模型架构提出挑战,因它们原为2D切片或单个3D扫描设计。为此,我们提出受放射学数据内在层次启发的分层注意力机制,构建了HLIP框架(Hierarchical attention for Language-Image Pre-training)。在220,000例脑MRI研究(共313万切片)和240,000例头颈部CT研究(共144万切片)上训练,HLIP在公开的Pub-Brain-5脑MRI基准上实现+10.5%平衡准确率,在CQ500和RSNA头颈CT基准上分别达+8.3%和+1.7%宏AUC。在现有3D医学多模态预训练基准上也表现优异,如在Rad-ChestCT上预训练于CT-RATE时,宏AUC提升+4.3%。结果表明,通过HLIP,直接在未标注临床数据上预训练是3D医学影像语言-图像预训练的有效且可扩展方向。代码已开源:https://github.com/Zch0414/hlip。

原文摘要 · Abstract (English)

The scalability of current language-image pre-training for 3D medical imaging, such as CT and MRI, is constrained by the need for radiologists to manually curate raw clinical studies. In this work, we pioneer pre-training directly on uncurated studies, which both aligns more closely with the radiologist's workflow and provides a natural path to scalability. However, the unique structure of such data presents new challenges for existing model architectures, which were originally designed for 2D slices or single 3D scans. To address this, we introduce a novel hierarchical attention mechanism inspired by the intrinsic hierarchy of radiology data: slice, scan, and study. We denote our framework as Hierarchical attention for Language-Image Pre-training (HLIP). Trained on 220K studies with 3.13 million scans for brain MRI and 240K studies with 1.44 million scans for head CT, HLIP achieves state-of-the-art performance, e.g., +10.5% balanced ACC on the proposed publicly available brain MRI benchmark Pub-Brain-5; +8.3% and +1.7% macro AUC on head CT benchmarks CQ500 and RSNA, respectively. HLIP also exhibits strong generalizability on existing 3D medical language-image pre-training benchmarks, e.g., +4.3% macro AUC on the Rad-ChestCT benchmark when pre-trained on CT-RATE. These results demonstrate that, with HLIP, directly pre-training on uncurated clinical datasets is a scalable and effective direction for language-image pre-training in 3D medical imaging. The code is available at https://github.com/Zch0414/hlip.

3D医学影像多模态预训练分层注意力可扩展性

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