解决跨客户端医学多模态数据异构下的专家模型分歧问题
MedMix: Specialization-Consistent Federated Sparse MoEs under Modality Heterogeneity

- 通过模态上下文感知路由实现本地专家选择
- 在严重异构下平均F1表现最优,提升显著
- 适合处理不完整多模态医疗数据的联邦学习场景
联邦多模态医疗AI在客户端和样本层面均面临模态异构:客户端可能系统性缺失特定模态类型,同一客户端内的记录也可能包含不同的部分模态组合。稀疏混合专家(MoE)架构可实现模态自适应计算,但在跨客户端模态异构下使用时存在脆弱性,本地学习的路由策略可能在不同客户端间发散,导致专家专长不一致。不同客户端可能将相同的模态配置分配给不同专家,或让相同索引的专家学习不同的缺失模态配置,标准聚合方式因此造成专家专长错位或覆盖。为此,我们提出MedMix,一种基于语义对齐的联邦多模态稀疏MoE框架,通过模态上下文协调跨客户端的路由与专家专长。客户端侧采用模态上下文感知路由,利用每个令牌的模态身份、位置及不完整性上下文引导专家选择;跨客户端则通过共识引导路由对齐,构建服务器端共享模态模式的共识锚点,对齐各客户端的路由分布。此外,客户端自适应专家聚合利用客户端特有的模态模式原型,匹配并聚合功能相似的专家。在真实世界多模态医疗数据集上的实验表明,MedMix在多种模态异构与不完整设置下均取得最优平均F1,尤其在严重异构条件下提升明显。
原文摘要 · Abstract (English)
Federated multimodal medical AI faces modality heterogeneity at both the client and sample levels: clients may systematically lack access to specific modality types, while individual records within the same client may contain different partial modality subsets. Sparse Mixture-of-Experts (MoE) architectures are a promising remedy for modality-adaptive computation, but their use in federated learning is fragile under cross-client modality heterogeneity, where locally learned routing policies can diverge across clients and drive experts toward incompatible specializations. Different clients may assign the same observed modality configuration to different experts, or train similarly indexed experts on different missing-modality configurations, causing standard aggregation to misalign or overwrite the expert specialization that sparse MoEs are intended to learn. To address this challenge, we propose MedMix, a semantic-alignment framework for federated multimodal sparse MoEs that coordinates cross-client routing and expert specialization using modality context. At the client side, MedMix uses modality-context-aware routing to guide expert selection using each token's modality identity, position, and incompleteness context. Across clients, it uses consensus-guided routing alignment to construct server-side consensus anchors for shared modality patterns and align local routing distributions across clients. Complementing these routing mechanisms, client-adaptive expert aggregation leverages client-specific modality-pattern prototypes to match and aggregate functionally similar experts across clients. Experiments on real-world multimodal medical datasets show that MedMix achieves the best average F1 across diverse modality heterogeneity and modality incompleteness settings, with especially clear gains under severe heterogeneity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。