SurgFed: Language-guided Multi-Task Federated Learning for Surgical Video Understanding
Organizations: National University of Singapore, Singapore · Spectral AI Team, The Hong Kong University of Science and Technology (Guangzhou), China · Zhejiang University, China · Nanyang Technological University, Singapore
Abstract
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/.
Figures & tables
| Method | Endovis2017 | Endovis2018 | AutoLaparo | SCARED | StereoMIS | Avg | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| IoU | Dice | IoU | Dice | IoU | Dice | RMSE | RMSE | |||||||
| Inference Only | 44.60 ±0.00 | 54.17 ±0.00 | 23.62 ±0.00 | 63.31 ±0.00 | 71.77 ±0.00 | 10.92 ±0.00 | 80.63 ±0.00 | 87.02 ±0.00 | 1.65 ±0.00 | — | — | — | — | — |
| Local Train | 58.77 ±0.74 | 70.47 ±1.28 | 0.00 ±0.42 | 71.53 ±0.79 | 80.06 ±0.91 | 0.00 ±0.20 | 82.39 ±0.82 | 88.04 ±1.66 | 0.00 ±0.37 | 10.76 ±0.23 | 0.00 ±0.69 | 14.75 ±0.54 | 0.00 ±0.13 | 0.00 ±0.71 |
| FedAvg | 54.59 ±0.53 | 66.38 ±0.79 | 6.46 ±0.71 | 65.06 ±0.29 | 74.21 ±0.03 | 8.18 ±0.43 | 83.19 ±0.76 | 88.59 ±0.25 | 0.80 ±0.35 | 28.61 ±0.71 | 165.75 ±0.64 | 14.77 ±0.38 | 0.17 ±0.74 | 35.95 ±1.56 |
| FedAvg+Cluster | 57.21 ±0.57 | 69.44 ±0.15 | 2.06 ±0.32 | 65.76 ±1.08 | 75.51 ±1.46 | 6.87 ±0.30 | 83.26 ±0.45 | 88.71 ±0.50 | 0.91 ±2.00 | 34.09 ±0.61 | 216.67 ±0.60 | 15.71 ±0.40 | 6.57 ±0.46 | 46.25 ±0.37 |
| FedRep | 58.59 ±0.45 | 70.58 ±1.04 | 0.08 ±1.34 | 70.94 ±0.38 | 79.32 ±0.59 | 0.87 ±0.43 | 83.37 ±0.09 | 88.86 ±0.67 | 1.06 ±1.45 | 16.78 ±0.31 | 55.86 ±0.78 | 14.63 ±0.98 | 0.77 ±0.50 | 10.99 ±0.92 |
| # | LCS | LHA | Text Pro. | Endovis2017 Dice | Endovis2018 Dice | AutoLaparo Dice | SCARED RMSE | StereoMIS RMSE | Avg. |
|---|---|---|---|---|---|---|---|---|---|
| 1 | — | — | — | 66.38(0.00) | 74.21(0.00) | 88.59(0.00) | 28.61(0.00) | 14.77(0.00) | |
| 2 | ✓ | — | — | ||||||
| 3 | — | ✓ | — | 89.01(+0.42) | 8.77(-19.84) | ||||
| 4 | — | ✓ | ✓ | 74.59(+8.21) | |||||
| 5 | ✓ | ✓ | — | 81.44(+7.23) | 14.34(-0.43) | 5.92(+41.87) |
| Indicator | Endo17. | Endo18. | Auto. | SCARED. | Stereo. | |
|---|---|---|---|---|---|---|
| Random | 3.08 | 1.85 | 0.82 | -7.68 | -3.20 | -1.03 |
| One-hot | 2.55 | 1.98 | 0.88 | 12.02 | 0.25 | 3.53 |
| Ours | 5.23 | 2.12 | 1.13 | 18.42 | 2.73 | 5.92 |
| Indicator | Endo17. | Endo18. | Auto. | SCARED. | Stereo. | |
|---|---|---|---|---|---|---|
| Random | 2.48 | -3.99 | 0.95 | -8.08 | 2.31 | -1.26 |
| One-hot | 2.50 | -3.33 | 1.03 | 6.36 | -0.15 | 1.28 |
| Ours | 5.23 | 2.12 | 1.13 | 18.42 | 2.73 | 5.92 |
Surgical Video Understanding
Surgical video understanding uses computer vision and multimodal AI to interpret procedures, instruments, actions, and events over time.
See all 118 papers