将多通道图像分通道预训练,提升医学与遥感图像识别性能
Isolated Channel Vision Transformers: From Single-Channel Pretraining to Multi-Channel Finetuning
- 分通道独立打块预训练,保留多模态信息
- 在细胞显微与卫星图像上提升4-14个百分点
- 适合大规模异构数据预训练,效率高
视觉变换器(ViT)在标准RGB图像任务中表现卓越,但在多通道成像(MCI)数据如医疗和遥感应用中仍面临挑战。由于多通道数据常来自不同模态,直接训练会掩盖互补信息。本文提出一种简单有效的预训练框架——孤立通道ViT(IC-ViT),对各图像通道独立打块,支持多模态多通道任务的预训练。该方法能有效捕捉补丁间及通道间的依赖关系,生成鲁棒特征表示。实验在细胞显微(JUMP-CP、CHAMMI)与卫星影像(So2Sat-LCZ42)任务中表明,相比现有通道自适应方法,IC-ViT性能提升4-14个百分点。其高效训练特性使其成为异构数据大模型预训练的理想选择。代码已开源。
原文摘要 · Abstract (English)
Vision Transformers (ViTs) have achieved remarkable success in standard RGB image processing tasks. However, applying ViTs to multi-channel imaging (MCI) data, e.g., for medical and remote sensing applications, remains a challenge. In particular, MCI data often consist of layers acquired from different modalities. Directly training ViTs on such data can obscure complementary information and impair the performance. In this paper, we introduce a simple yet effective pretraining framework for large-scale MCI datasets. Our method, named Isolated Channel ViT (IC-ViT), patchifies image channels individually and thereby enables pretraining for multimodal multi-channel tasks. We show that this channel-wise patchifying is a key technique for MCI processing. More importantly, one can pretrain the IC-ViT on single channels and finetune it on downstream multi-channel datasets. This pretraining framework captures dependencies between patches as well as channels and produces robust feature representation. Experiments on various tasks and benchmarks, including JUMP-CP and CHAMMI for cell microscopy imaging, and So2Sat-LCZ42 for satellite imaging, show that the proposed IC-ViT delivers 4-14 percentage points of performance improvement over existing channel-adaptive approaches. Further, its efficient training makes it a suitable candidate for large-scale pretraining of foundation models on heterogeneous data. Our code is available at https://github.com/shermanlian/IC-ViT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。