arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.01179cs.CVcs.AIcs.LG

冻结骨干网络:一种用于鲁棒医学视觉-语言预训练的参数高效对比方法

Freeze the backbones: A Parameter-Efficient Contrastive Approach to Robust Medical Vision-Language Pre-training

Imperial College London(伦敦帝国理工学院);The Hong Kong University of Science and Technology(香港科技大学)

Jiuming Qin, Che Liu, Sibo Cheng, Yike Guo, Rossella Arcucci

AI总结:

针对医学视觉-语言自监督学习端到端训练计算量大且易丢失先验信息的问题,提出冻结预训练编码器并采用轻量级 Adaptor 模块进行跨模态学习的框架,在大幅减少可训练参数的同时保持竞争力。

摘要:

现代医疗保健通常结合放射图像和文本报告进行诊断,促使使用带有大型预训练模型的视觉-语言自监督学习(VL-SSL)来学习通用的医学视觉表示。然而,大多数现有的 VL-SSL 框架是端到端训练的,计算量大,且可能丢失预训练编码器中嵌入的重要先验信息。为解决这两个问题,我们引入了与骨干网络无关的 Adaptor 框架,通过保持预训练的图像和文本编码器冻结来保留医学知识,并采用轻量级 Adaptor 模块进行跨模态学习。在三个数据集上的医学图像分类和分割任务实验表明,我们的框架在提供有竞争力的性能的同时,与当前预训练方法相比减少了超过 90% 的可训练参数。值得注意的是,当仅用 1% 的数据进行微调时,Adaptor 在医学图像分割中优于几种在全数据集上训练的基于 Transformer 的方法。

英文摘要:

Modern healthcare often utilises radiographic images alongside textual reports for diagnostics, encouraging the use of Vision-Language Self-Supervised Learning (VL-SSL) with large pre-trained models to learn versatile medical vision representations. However, most existing VL-SSL frameworks are trained end-to-end, which is computation-heavy and can lose vital prior information embedded in pre-trained encoders. To address both issues, we introduce the backbone-agnostic Adaptor framework, which preserves medical knowledge in pre-trained image and text encoders by keeping them frozen, and employs a lightweight Adaptor module for cross-modal learning. Experiments on medical image classification and segmentation tasks across three datasets reveal that our framework delivers competitive performance while cutting trainable parameters by over 90% compared to current pre-training approaches. Notably, when fine-tuned with just 1% of data, Adaptor outperforms several Transformer-based methods trained on full datasets in medical image segmentation.

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

↑