Document Type

Conference Paper

Publication Date

2024

DOI

10.1117/12.3009710

Publication Title

Medical Imaging 2024: Clinical and Biomedical Imaging, Proc. of SPIE 12930

Volume

12930

Pages

129300J

Conference Name

SPIE Medical Imaging 2024, 18-23 February 2024, San Diego, California

Abstract

Federated Learning (FL) is a promising machine learning approach for development of data-driven global model using collaborative local models across multiple local institutions. However, the heterogeneity of medical imaging data is one of the challenges within FL. This heterogeneity is caused by the variation in imaging scanner protocols across institutions, which may result in weight shift among local models leading to deterioration in predictive accuracy of global model. The prevailing approaches involve applying different FL averaging techniques to enhance the performance of the global model, ignoring the distinct imaging features of the local domain. In this work, we address both the local and global model weight shift by introducing multiscale amplitude harmonization of the imaging in the local models utilizing Haar and harmonic wavelets. First, we tackle the local model weight shift by transforming the image feature space into multiscale frequency space using multiscale based harmonization. This aims to achieve harmonized image feature space across local models. Second, based on harmonized image feature space, a weighted regularization term is applied to local models, effectively mitigating weight shifts within these models. This weighted regularization assists in managing global model shifts by aggregating the optimized local models. We evaluate the proposed method using publicly available histopathological dataset MoNuSAC2018, TNBC for nuclei segmentation, and Camelyon17 dataset for tumor tissue classification. The average testing accuracies are 96.55%, and 92.47% for classification of tumor tissue while Dice co-efficients are 84.33%, and 84.46% for segmentation of nuclei with Haar and harmonic multiscale based harmonization, respectively. The comparison results for nuclei segmentation and tumor tissue classification using histopathological data show that our proposed methods perform better than the state-of-the-art FL methods.

Rights

Copyright 2024 Society of Photo‑Optical Instrumentation Engineers (SPIE). One print or electronic copy may be made for personal use only. Systematic reproduction and distribution, duplication of any material in this publication for a fee or for commercial purposes, and modification of the contents of the publication are prohibited.

"SPIE grants to authors (and their employers) of papers, posters, and presentation recordings published in SPIE Proceedings or SPIE Journals on the SPIE Digital Library (hereinafter "publications") the right to post an author-prepared version or an official version (preferred) of the publication on an internal or external server controlled exclusively by the author/employer or the entity funding the research, provided that (a) such posting is noncommercial in nature and the publication is made available to users without charge; (b) an appropriate copyright notice and citation appear with the publication; and (c) a link to SPIE's official online version of the publication is provided using the item's DOI."

Original Publication Citation

Farzana, W., Temtam, A., & Iftekharuddin, K. M. (2024). Wavelet-based harmonization of local and global model shifts in federated learning for histopathological images. In B. S. Gimi & A. Krol (Eds.), Medical Imaging 2024: Clinical and Biomedical Imaging, Proc. of SPIE 12930 (129300J). SPIE. https://doi.org/10.1117/12.3009710

ORCID

0000-0003-1995-2426 (Farzana), 0000-0001-8316-4163 (Iftekharuddin)

Download

Share

COinS