Novel Methods for Catheter and Guidewire Segmentation in X-ray Fluoroscopy under a Federated Learning Setting
Organizations: School of Computer Science and Informatics, University of Liverpool, UK.
Abstract
Endovascular procedures rely on real-time manipulation of thin instruments, catheters and guidewires, under X-ray fluoroscopy guidance, where accurate visual analysis is essential for procedural safety. Learning-based methods are constrained by structural complexity, data scarcity, and privacy regulations precluding centralised training across institutions. This thesis presents a structure-aware federated learning framework for catheter and guidewire analysis, with four contributions evaluated on real-animal and phantom data. A benchmark dataset, CathAction, is introduced for catheterisation analysis, with over 600,000 annotated frames and 40,000 segmentation masks. A shape-sensitive loss transforms masks into signed distance maps compared in a structural feature space, improving Dice coefficient by up to 2.9 points across five backbones. This is extended to federated learning with shape-sensitive loss, preserving geometric consistency under heterogeneous client data and outperforming federated averaging by up to three points in mean intersection-over-union as clients scale from four to eight. Federated learning with projected gradient descent adds adversarial optimisation, raising mean intersection-over-union by over ten points on real-animal data. Finally, a structure-aware diffusion framework synthesises catheter and guidewire video sequences, combining structural supervision with a domain-adaptive reconstruction objective, reducing Frechet video distance over a strong baseline while maintaining visual fidelity. Incorporating synthetic sequences into federated training raises the Dice score from 44 to 51 percent under data scarcity, with gains across four held-out sites. Together, these contributions advance privacy-preserving catheter and guidewire analysis, supporting collaborative training without centralising patient data or large amounts of manual annotation.
Figures & tables
| Category | Representative Loss | Key Property |
| Distribution-Based | Binary Cross-Entropy, Focal Loss | Treat segmentation as independent pixel-wise classification without explicit spatial modelling. |
| Region-Based | Dice, Tversky | Optimise region-level overlap between predicted and ground-truth masks. |
| Boundary-Based | Hausdorff Distance Loss, Boundary Loss | Incorporate boundary distance and contour alignment information. |
| Topology-Aware | clDice | Encourage connectivity preservation in tubular structures. |
| Feature-Level / Shape-Sensitive | Shape-Sensitive Loss ( SSL ) | Compare structural representations in feature space rather than pixel masks. |
| Dataset | Modality | Format | #Frames | Source | Tasks |
| Barbu et al. [ 12 ] | X-ray | Video | 535 | Human | Segmentation |
| Wu et al. [ 95 ] | 3D Echo | Video | 800 | Human | Segmentation |
| Ambrosini et al. [ 3 ] | X-ray | Image | 948 | Human | Segmentation |
| Mastmeyer et al. [ 13 ] | 3D MRI | Image | 101 | Human | Segmentation |
| Yi et al. [ 92 ] | X-ray | Image | 2,540 | Synthetic | Segmentation |
| Nguyen et al. [ 80 ] | X-ray | Image | 25,271 | Phantom | Segmentation |
| Baseline | Venue | Accuracy | Precision | Recall |
| CNN [ 112 ] | CVPR 2018 | 28.98 | 30.14 | 29.76 |
| RNN [ 112 ] | CVPR 2018 | 29.64 | 30.38 | 30.44 |
| RU-LSTM [ 113 ] | CVPR 2019 | 35.08 | 34.29 | 34.77 |
| TempAggRe [ 114 ] | ECCV 2020 | 34.64 | 35.56 | 34.71 |
| Trans-SVNet [ 118 ] | IJCARS 2022 | 29.06 | 19.67 | 20.28 |
| AFFT [ 115 ] | WACV 2023 | 37.91 | 36.87 | 37.63 |
| Baseline | Venue | Accuracy | Precision | Recall |
| TDN-ResNet50 [ 120 ] | CVPR 2021 | 58.34 | 59.12 | 57.22 |
| TDN-ResNet101 [ 120 ] | CVPR 2021 | 62.50 | 61.89 | 62.77 |
| Video Swin Transformer [ 121 ] | CVPR 2022 | 51.67 | 52.14 | 51.24 |
| BEVT [ 122 ] | CVPR 2022 | 49.28 | 50.27 | 49.92 |
| CycleACR [ 123 ] | TPAMI 2025 | 60.42 | 58.15 | 61.24 |
| Baseline | Dice Score | Jaccard Index | mIoU | Accuracy |
| U-Net [ 36 ] | 51.69 | 57.51 | 31.17 | 63.26 |
| TransUNet [ 41 ] | 56.52 | 55.93 | 34.13 | 55.61 |
| SwinUNet [ 135 ] | 61.26 | 59.54 | 39.53 | 76.60 |
| SSL [ 228 ] | 56.95 | 56.87 | 40.87 | 72.24 |
| SegViT [ 137 ] | 63.47 | 54.12 | 42.48 | 68.73 |
| AP | mAP | |||||
| Baseline | Collision | Normal | Mean | Collision | Normal | Mean |
| STEP [ 141 ] | 7.79 | 11.21 | 10.98 | 6.92 | 11.29 | 9.08 |
| YOWO [ 139 ] | 8.32 | 12.18 | 11.73 | 7.46 | 12.28 | 9.92 |
| YOWO-Plus [ 140 ] | 8.92 | 12.23 | 11.77 | 7.86 | 12.48 | 10.28 |
| HIT [ 142 ] | 9.37 | 12.74 | 12.14 | 8.18 | 12.72 | 10.81 |
| Yolov ∗ [ 143 ] | 12.30 | 21.08 | 15.89 | 11.88 | 20.04 | 14.11 |
| Baseline | Venue | Accuracy | Precision | Recall |
| RU-LSTM [ 113 ] | CVPR 2019 | 22.93 | 23.91 | 22.57 |
| TempAggRe [ 114 ] | ECCV 2020 | 17.16 | 18.41 | 18.23 |
| Trans-SVNet [ 118 ] | IJCARS 2022 | 19.06 | 17.67 | 19.58 |
| AFFT [ 115 ] | WACV 2023 | 25.67 | 26.29 | 26.33 |
| Baseline | Venue | Accuracy | Precision | Recall |
| TDN-ResNet50 [ 120 ] | CVPR 2021 | 24.19 | 23.17 | 24.56 |
| TDN-ResNet101 [ 120 ] | CVPR 2021 | 25.62 | 24.52 | 25.68 |
| Video Swin Transformer [ 121 ] | CVPR 2022 | 28.79 | 27.98 | 28.12 |
| BEVT [ 122 ] | CVPR 2022 | 31.22 | 30.48 | 31.79 |
| AP | mAP | |||||
| Baseline | Collision | Normal | Mean | Collision | Normal | Mean |
| STEP [ 141 ] | 1.53 | 2.12 | 1.87 | 1.09 | 1.98 | 1.62 |
| YOWO [ 139 ] | 2.12 | 4.11 | 3.09 | 1.97 | 3.68 | 2.92 |
| YOWO-Plus [ 140 ] | 1.18 | 1.43 | 1.21 | 1.07 | 1.26 | 1.09 |
| HIT [ 142 ] | 1.31 | 1.19 | 1.24 | 1.06 | 1.18 | 1.11 |
| Yolov ∗ [ 143 ] | 7.31 | 8.92 | 8.09 | 6.28 | 7.49 | 7.21 |
| Baseline | Dice Score | Jaccard Index | mIoU | Accuracy |
| UNet [ 36 ] | 26.58 | 31.38 | 12.13 | 46.07 |
| TransUNet [ 41 ] | 16.16 | 24.19 | 17.23 | 33.61 |
| SwinUNet [ 135 ] | 17.41 | 38.14 | 7.52 | 40.79 |
| SSL [ 228 ] | 26.91 | 32.04 | 18.72 | 42.44 |
| SegViT [ 137 ] | 30.74 | 32.22 | 11.46 | 50.00 |
| Baseline | Dice Score | Jaccard Index | mIoU | Accuracy |
| UNet [ 36 ] | 51.93 | 35.07 | 67.31 | 49.21 |
| TransUNet [ 41 ] | 58.39 | 43.64 | 71.59 | 63.66 |
| SwinUNet [ 135 ] | 56.60 | 41.43 | 70.48 | 63.23 |
| SSL [ 228 ] | 58.88 | 45.02 | 72.01 | 71.48 |
| SegViT [ 137 ] | 48.72 | 34.64 | 66.99 | 56.81 |
| Baseline | Dice Score | Jaccard Index | mIoU | Accuracy |
| UNet [ 36 ] | 24.33 | 13.85 | 56.51 | 26.88 |
| TransUNet [ 41 ] | 24.42 | 15.26 | 57.28 | 30.14 |
| SwinUNet [ 135 ] | 25.81 | 15.92 | 57.58 | 34.99 |
| SSL [ 228 ] | 24.97 | 16.04 | 58.75 | 34.10 |
| SegViT [ 137 ] | 26.05 | 14.97 | 57.09 | 26.91 |
| Distribution-Based Loss | |
| Binary Cross-Entropy [ 161 ] | |
| Weighted Cross-Entropy [ 162 ] | |
| Balanced Cross-Entropy [ 163 ] | |
| Focal Loss [ 44 ] | |
| Region-Based Loss | |
| Dice Loss [ 50 ] | |
| Network | Dice | Jaccard | mIoU | Accuracy |
| TransU-net | 54.52 | 43.87 | 61.15 | 78.25 |
| TransU-Net w/ours | 57.16 | 46.04 | 62.66 | 79.57 |
| U-Net | 46.20 | 36.62 | 54.14 | 71.57 |
| U-Net w/ours | 47.95 | 38.79 | 55.05 | 72.89 |
| U-Net++ | 48.77 | 39.22 | 56.17 | 72.30 |
| U-Net++ w/ours | 50.35 | 41.39 | 57.08 | 73.62 |
| Network | Dice | Jaccard | mIoU | Accuracy |
| TransU-net | 40.83 | 33.94 | 46.01 | 66.18 |
| TransU-Net w/ours | 43.71 | 36.65 | 47.21 | 67.58 |
| U-Net | 34.51 | 27.62 | 43.14 | 60.57 |
| U-Net w/ours | 37.13 | 28.56 | 45.83 | 62.29 |
| U-Net++ | 35.24 | 27.43 | 44.72 | 61.76 |
| U-Net++ w/ours | 37.25 | 28.94 | 45.97 | 63.12 |
| Method | Real | Phantom | ||||||
| 25 Rounds | 50 Rounds | 25 Rounds | 50 Rounds | |||||
| mIoU | Dice | mIoU | Dice | mIoU | Dice | mIoU | Dice | |
| FedAVG [ 16 ] | 32.92 | 49.18 | 45.80 | 61.63 | 17.61 | 49.10 | 18.81 | 50.24 |
| FedProx [ 56 ] | 28.17 | 35.36 | 39.70 | 48.73 | 15.17 | 46.29 | 17.19 | 48.68 |
| FedSSL_1 ( Ours ) | 28.32 | 48.25 | 48.82 | 63.48 | 17.13 | 48.66 | 21.03 | 51.36 |
| FedSSL_2 ( Ours ) | 28.50 | 47.59 | 48.95 | 63.78 | 17.08 | 48.62 | 19.73 | 50.85 |
| Method | Real + Phantom | |||
| 25 Rounds | 50 Rounds | |||
| mIoU | Dice | mIoU | Dice | |
| FedAVG [ 16 ] | 41.28 | 56.19 | 41.40 | 58.65 |
| FedProx [ 56 ] | 33.53 | 44.34 | 37.04 | 49.65 |
| FedSSL_1 ( Ours ) | 41.21 | 56.79 | 42.31 | 59.42 |
| FedSSL_2 ( Ours ) | 41.40 | 57.14 | 42.61 | 59.97 |
| Method | Real | Phantom | ||||||
| 25 Rounds | 50 Rounds | 25 Rounds | 50 Rounds | |||||
| mIoU | Dice | mIoU | Dice | mIoU | Dice | mIoU | Dice | |
| FedAVG [ 16 ] | 32.92 | 49.18 | 45.80 | 61.63 | 17.61 | 49.10 | 18.81 | 50.24 |
| FedProx [ 56 ] | 28.17 | 35.36 | 39.70 | 48.73 | 15.17 | 46.29 | 17.19 | 48.68 |
| FedPGD ( Ours ) | 43.86 | 58.05 | 47.67 | 62.90 | 17.71 | 48.40 | 21.12 | 50.82 |
| Method | Real + Phantom | |||
| 25 Rounds | 50 Rounds | |||
| mIoU | Dice | mIoU | Dice | |
| FedAVG [ 16 ] | 41.28 | 56.19 | 41.40 | 58.65 |
| FedProx [ 56 ] | 33.53 | 44.34 | 37.04 | 49.65 |
| FedPGD ( Ours ) | 41.38 | 55.78 | 42.43 | 58.95 |
| Method | #Client = 4 | #Client = 8 | ||||||
| Round | 25 | 25 | 50 | 50 | 25 | 25 | 50 | 50 |
| Metric | mIoU | Dice | mIoU | Dice | mIoU | Dice | mIoU | Dice |
| FedAVG [ 16 ] | 32.92 | 49.18 | 45.80 | 61.63 | 30.32 | 46.81 | 44.18 | 60.21 |
| FedProx [ 56 ] | 28.17 | 35.36 | 39.70 | 48.73 | 28.48 | 45.38 | 42.61 | 54.25 |
| FedSSL_1 ( Ours ) | 34.56 | 51.12 | 48.82 | 63.48 | 31.56 | 48.21 | 47.24 | 61.89 |
| FedSSL_2 ( Ours ) | 35.23 | 51.64 | 48.95 | 63.78 | 31.79 | 48.67 | 47.43 | 62.26 |
| Method | Real | Phantom | ||||
| FVD | FID | IS | FVD | FID | IS | |
| LVDM [ 224 ] | 307.01 | 79.13 | 1.18 | 158.73 | 138.16 | 1.53 |
| StyleGAN-V [ 223 ] | 432.87 | 153.12 | 1.03 | 255.89 | 235.64 | 1.44 |
| Endora [ 216 ] | 325.44 | 46.89 | 1.21 | 160.34 | 86.55 | 1.60 |
| Ours | 201.11 | 47.28 | 1.21 | 144.33 | 68.09 | 1.64 |
| Method | Real | Phantom | ||||
| SSIM | W-SSIM | SD | SSIM | W-SSIM | SD | |
| LVDM [ 224 ] | .8917 | .8911 | .8467 | .9488 | .9475 | .8304 |
| StyleGAN-V [ 223 ] | .8528 | .8468 | .8114 | .9075 | .9008 | .8036 |
| Endora [ 216 ] | .9170 | .9118 | .8722 | .9667 | .9646 | .8578 |
| Ours | .9195 | .9142 | .8738 | .9863 | .9849 | .9134 |
| 2PC | SDM | VAE | LPIPS | FVD | FID | SSIM |
| ✗ | ✗ | ✗ | ✗ | 325.44 | 46.89 | .9170 |
| ✓ | ✗ | ✗ | ✗ | 287.24 | 53.88 | .9191 |
| ✓ | ✓ | ✗ | ✗ | 191.96 | 99.34 | .9137 |
| ✓ | ✓ | ✓ | ✗ | 213.04 | 53.33 | .8835 |
| ✓ | ✓ | ✓ | ✓ | 201.11 | 47.28 | .9195 |
| Training Setup | Dice | IoU(FG) | mIoU | HD95 |
| Real-animal dataset | ||||
| Real (10%) | .3661 | .2529 | .6234 | 23.4963 |
| Real (10%) + Endora [ 216 ] | .4734 | .3339 | .6641 | 21.9784 |
| Real (10%) + Ours | .5188 | .3720 | .6835 | 20.6756 |
| Real (100%) | .6929 | .5460 | .7711 | 3.0879 |
| Phantom dataset | ||||
| Synthetic ratio | Dice | HD95 (px) | mIoU |
| Real-only | .4414 | 18.26 | .3280 |
| 0.5 | .4407 | 17.38 | .3309 |
| 1.0 | .4827 | 21.31 | .3482 |
| 2.0 | .5088 | 19.95 | .3615 |
| 4.0 | .5095 | 21.52 | .3529 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Symbol | Value |
| Learning rate | – | |
| Gradient accumulation steps | – | 3 |
| Training iterations | – | 80k |
| Input resolution | – | |
| Sequence length | – | 16 frames |
| SDM loss weight | 1.0 |
| Parameter | Symbol | Value |
| Input resolution | – | |
| Batch size | – | 4 |
| Training epochs | – | 100 |
| Learning rate | – | 0.01 |
| LR decay (power) | – | 0.9 |
| Momentum | – | 0.9 |
| Parameter | Symbol | Value |
| Number of clients | 4 | |
| Communication rounds | 50 | |
| Local epochs per round | – | 1 |
| Local batch size | – | 8 |
| Local learning rate | – | 0.1 |
| Weight decay | – |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 796 papers