A privacy-preserving federated learning framework for generalizable CBCT to synthetic CT translation in head and neck.
Authors
Affiliations (2)
Affiliations (2)
- Institute of Biomedical Engineering, Karlsruhe Institute of Technology, Karlsruhe, Germany.
- Department of Experimental and Clinical Medicine, Magna Graecia University, Catanzaro, Italy.
Abstract
Cone-beam computed tomography (CBCT) has become a widely adopted modality for image-guided radiotherapy (IGRT). However, CBCT is characterized by increased noise, limited soft-tissue contrast, and artifacts. These issues result in unreliable Hounsfield unit (HU) values, which limits electron density estimation for direct dose calculation. These issues have been addressed by deriving synthetic CT (sCT) from CBCT, particularly by adopting deep learning (DL) methods. However, existing DL approaches are hindered by institutional heterogeneity, scanner-dependent variations, and data privacy regulations that prevented multi-center data sharing. To overcome these challenges, we propose a cross-silo federated learning (FL) approach for CBCT-to-sCT synthesis in the head and neck region. This approach extends the original FedSynthCT framework to a different image modality and anatomical region. A conditional generative adversarial network (cGAN) was trained using data from three European medical centers within the SynthRAD2025 public challenge dataset while maintaining data privacy at each institution. A combination of the FedAvg and FedProx aggregation strategies, alongside a standardized preprocessing pipeline, was adopted to federate the DL model. The federated model effectively generalized across participating centers, as evidenced by the mean absolute error (MAE) ranging from <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>64.38</mn> <mo>±</mo> <mn>13.63</mn></math> to <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>85.90</mn> <mo>±</mo> <mn>7.10</mn></math> HU, the structural similarity index (SSIM) ranging from <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>0.88</mn> <mo>±</mo> <mn>0.02</mn></math> to <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>0.92</mn> <mo>±</mo> <mn>0.04</mn></math> , and the peak signal-to-noise ratio (PSNR) ranging from <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>32.86</mn> <mo>±</mo> <mn>0.94</mn></math> to <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>34.91</mn> <mo>±</mo> <mn>1.04</mn></math> dB. Notably, performance on an external validation dataset of 60 patients yielded comparable metrics: a MAE of <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>75.22</mn> <mo>±</mo> <mn>11.81</mn></math> HU, an SSIM of <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>0.90</mn> <mo>±</mo> <mn>0.03</mn></math> and a PSNR of <math xmlns="http://www.w3.org/1998/Math/MathML"><mn>33.52</mn> <mo>±</mo> <mn>2.06</mn></math> , confirming robust cross-center generalization despite differences in imaging protocols and scanner types, without additional training. Furthermore, a visual analysis of the results revealed that the obtained metrics were influenced by registration errors. Our findings demonstrated the technical feasibility of FL for CBCT-to-sCT synthesis task while preserving data privacy, offering a collaborative solution for developing generalizable models across institutions without requiring data sharing or center-specific models.