用语言引导的联邦学习提升手术视频多任务理解效果
SurgFed: Language-guided Multi-Task Federated Learning for Surgical Video Understanding
- 通过语言引导通道选择,增强不同医院的本地模型适应性
- 在五个数据集上实现分割与深度估计的性能超越现有方法
- 适合医疗影像联邦学习、多中心手术分析的研究者使用
机器人辅助微创手术中的手术场景多任务联邦学习(MTFL)至关重要,但因两大挑战尚未充分探索:(1) 组织多样性:本地模型难以适应特定部位组织特征,在异质临床环境中表现差,导致预测不准;(2) 任务多样性:仅依赖梯度聚类的服务器端聚合常因跨站点任务差异产生次优或错误参数更新,造成定位不准。针对此,我们提出SurgFed,一种面向多种手术类型的手术视频分割与深度估计的多任务联邦学习框架。SurgFed采用两项创新设计:语言引导通道选择(LCS)和语言引导超聚合(LHA),以实现跨站点与跨任务的充分探索。技术上,LCS设计轻量级个性化通道选择网络,利用预定义文本输入优化本地模型对特定组织嵌入的学习;LHA则引入分层交叉注意力机制,结合预定义文本建模跨站点任务交互,并指导超网络生成个性化参数更新。大量实验证明,SurgFed在四个手术类型、五个公开数据集上均优于现有最先进方法。代码已开源。
原文摘要 · Abstract (English)
Surgical scene Multi-Task Federated Learning (MTFL) is essential for robot-assisted minimally invasive surgery (RAS) but remains underexplored in surgical video understanding due to two key challenges: (1) Tissue Diversity: Local models struggle to adapt to site-specific tissue features, limiting their effectiveness in heterogeneous clinical environments and leading to poor local predictions. (2) Task Diversity: Server-side aggregation, relying solely on gradient-based clustering, often produces suboptimal or incorrect parameter updates due to inter-site task heterogeneity, resulting in inaccurate localization. In light of these two issues, we propose SurgFed, a multi-task federated learning framework, enabling federated learning for surgical scene segmentation and depth estimation across diverse surgical types. SurgFed is powered by two appealing designs, i.e., Language-guided Channel Selection (LCS) and Language-guided Hyper Aggregation (LHA), to address the challenge of fully exploration on corss-site and cross-task. Technically, the LCS is first designed a lightweight personalized channel selection network that enhances site-specific adaptation using pre-defined text inputs, which optimally the local model learn the specific embeddings. We further introduce the LHA that employs a layer-wise cross-attention mechanism with pre-defined text inputs to model task interactions across sites and guide a hypernetwork for personalized parameter updates. Extensive empirical evidence shows that SurgFed yields improvements over the state-of-the-art methods in five public datasets across four surgical types. The code is available at https://anonymous.4open.science/r/SurgFed-070E/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。