用傅里叶相位迁移实现2D与3D模型间权重无缝迁移
Cross-D Conv: Cross-Dimensional Transferable Knowledge Base via Fourier Shifting Operation
- 在傅里叶域学习相位偏移,实现2D/3D卷积权重转换
- 在RadImagenet和多模态体数据上性能相当或更优
- 适合缺乏3D数据的医疗影像模型预训练场景
在生物医学影像分析中,2D与3D数据间的鸿沟构成重大挑战。3D图像虽更贴近真实应用,但每种模态的数据量少且难以大规模训练;2D样本丰富却信息不完整。本文提出Cross-D Conv操作,通过在傅里叶域学习相位偏移,弥合维度差异,实现2D与3D卷积操作间的无缝权重迁移。该架构利用大量2D训练数据提升3D模型性能,为3D医学模型预训练中的多模态数据稀缺问题提供实用解决方案。在RadImagenet(2D)和多模态体数据集上的实验表明,本方法在特征质量评估上达到相当或更优表现。增强的卷积操作为开发高效分类与分割模型开辟新路径。本工作推动了跨维度与多模态医学影像分析的发展,提供了一种利用2D先验进行3D模型预训练的稳健框架,同时保持2D训练的计算效率。代码已开源:https://github.com/convergedmachine/Cross-D-Conv。
原文摘要 · Abstract (English)
In biomedical imaging analysis, the dichotomy between 2D and 3D data presents a significant challenge. While 3D volumes offer superior real-world applicability, they are less available for each modality and not easy to train in large scale, whereas 2D samples are abundant but less comprehensive. This paper introduces Cross-D Conv operation, a novel approach that bridges the dimensional gap by learning the phase shifting in the Fourier domain. Our method enables seamless weight transfer between 2D and 3D convolution operations, effectively facilitating cross-dimensional learning. The proposed architecture leverages the abundance of 2D training data to enhance 3D model performance, offering a practical solution to the multimodal data scarcity challenge in 3D medical model pretraining. Experimental validation on the RadImagenet (2D) and multimodal volumetric sets demonstrates that our approach achieves comparable or superior performance in feature quality assessment. The enhanced convolution operation presents new opportunities for developing efficient classification and segmentation models in medical imaging. This work represents an advancement in cross-dimensional and multimodal medical image analysis, offering a robust framework for utilizing 2D priors in 3D model pretraining while maintaining computational efficiency of 2D training. The code is available on https://github.com/convergedmachine/Cross-D-Conv.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。