arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2311.17597cs.CV

持续自监督学习:迈向通用多模态医学数据表示学习

Continual Self-supervised Learning: Towards Universal Multi-modal Medical Data Representation Learning

Northwestern Polytechnical University(西北工业大学);The University of Adelaide(阿德莱德大学)

Yiwen Ye, Yutong Xie, Jianpeng Zhang, Ziyang Chen, Qi Wu, Yong Xia

AI总结:

本文提出MedCoSS,一种面向多模态医学数据的持续自监督学习方法,通过分阶段预训练、k-means采样回放、特征蒸馏和模态内mixup策略平衡模态冲突并防止灾难性遗忘,在九个下游数据集上展现出卓越泛化能力和新模态扩展性。

摘要:

自监督学习是医学图像分析中一种高效的预训练方法。然而,当前研究大多局限于特定模态数据的预训练,耗费大量时间和资源却未能实现跨模态的通用性。一种直接的解决方案是联合所有模态数据进行自监督预训练,但这带来了实际挑战。首先,我们的实验表明,随着模态数量的增加,表示学习会出现冲突。其次,预先收集的多模态数据无法覆盖所有现实世界场景。在本文中,我们从持续学习的视角重新审视通用自监督学习,并提出MedCoSS,一种面向多模态医学数据的持续自监督学习方法。与联合自监督学习不同,MedCoSS将不同模态的数据分配到不同的训练阶段,形成多阶段预训练过程。为了平衡模态冲突并防止灾难性遗忘,我们提出了一种基于回放的持续学习方法。我们引入k-means采样策略来保留先前模态的数据,并在学习新模态时进行回放。我们没有在缓冲数据上执行前置任务,而是对这些数据应用特征蒸馏策略和模态内mixup策略以保留知识。我们在大规模多模态无标注数据集上进行了持续自监督预训练,数据包括临床报告、X光片、CT扫描、MRI扫描和病理图像。实验结果表明,MedCoSS在九个下游数据集上展现出卓越的泛化能力,并且在整合新模态数据方面具有显著的可扩展性。代码和预训练权重可在https://github.com/yeerwen/MedCoSS获取。

英文摘要:

Self-supervised learning is an efficient pre-training method for medical image analysis. However, current research is mostly confined to specific-modality data pre-training, consuming considerable time and resources without achieving universality across different modalities. A straightforward solution is combining all modality data for joint self-supervised pre-training, which poses practical challenges. Firstly, our experiments reveal conflicts in representation learning as the number of modalities increases. Secondly, multi-modal data collected in advance cannot cover all real-world scenarios. In this paper, we reconsider versatile self-supervised learning from the perspective of continual learning and propose MedCoSS, a continuous self-supervised learning approach for multi-modal medical data. Unlike joint self-supervised learning, MedCoSS assigns different modality data to different training stages, forming a multi-stage pre-training process. To balance modal conflicts and prevent catastrophic forgetting, we propose a rehearsal-based continual learning method. We introduce the k-means sampling strategy to retain data from previous modalities and rehearse it when learning new modalities. Instead of executing the pretext task on buffer data, a feature distillation strategy and an intra-modal mixup strategy are applied to these data for knowledge retention. We conduct continuous self-supervised pre-training on a large-scale multi-modal unlabeled dataset, including clinical reports, X-rays, CT scans, MRI scans, and pathological images. Experimental results demonstrate MedCoSS's exceptional generalization ability across nine downstream datasets and its significant scalability in integrating new modality data. Code and pre-trained weight are available at https://github.com/yeerwen/MedCoSS.

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

↑