arXiv:2511.17400cs.CVcs.AI2025-11中稿 · NeurIPS

提出MoE-ViT模型,用稀疏专家机制降低多通道图像的计算开销。

Sparse Mixture-of-Experts for Multi-Channel Imaging: Are All Channel Interactions Required?

  • 每通道作为专家,轻量路由只选关键通道参与注意力计算
  • 在JUMP-CP和So2Sat上减少大量计算量,性能不降反升
  • 适合卫星、细胞成像等多通道图像任务的高效建模

视觉变换器(ViTs)已成为视觉基础模型的核心,但在细胞着色或卫星图像等多通道领域,其优化仍不充分。这类数据的关键挑战在于捕捉通道间的交互信息,而现有方法将各通道独立处理,在注意力模块中引入二次增长的计算量,导致极高FLOPs和训练成本。本文聚焦于跨通道注意力中的效率问题,提出一个基于稀疏混合专家(MoE)思想的MoE-ViT架构,将每个通道视为专家,通过轻量路由仅选择每块(patch)最相关的专家进行注意力计算。在真实数据集JUMP-CP和So2Sat上的实验表明,MoE-ViT在显著降低计算开销的同时,性能未下降甚至提升,成为多通道成像任务中实用且高效的骨干网络。

原文摘要 · Abstract (English)

Vision Transformers ($\text{ViTs}$) have become the backbone of vision foundation models, yet their optimization for multi-channel domains - such as cell painting or satellite imagery - remains underexplored. A key challenge in these domains is capturing interactions between channels, as each channel carries different information. While existing works have shown efficacy by treating each channel independently during tokenization, this approach naturally introduces a major computational bottleneck in the attention block - channel-wise comparisons leads to a quadratic growth in attention, resulting in excessive $\text{FLOPs}$ and high training cost. In this work, we shift focus from efficacy to the overlooked efficiency challenge in cross-channel attention and ask: "Is it necessary to model all channel interactions?". Inspired by the philosophy of Sparse Mixture-of-Experts ($\text{MoE}$), we propose MoE-ViT, a Mixture-of-Experts architecture for multi-channel images in $\text{ViTs}$, which treats each channel as an expert and employs a lightweight router to select only the most relevant experts per patch for attention. Proof-of-concept experiments on real-world datasets - JUMP-CP and So2Sat - demonstrate that $\text{MoE-ViT}$ achieves substantial efficiency gains without sacrificing, and in some cases enhancing, performance, making it a practical and attractive backbone for multi-channel imaging.

多通道图像MoEViT优化高效建模

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