Federated attention consistent learning models for prostate cancer diagnosis and Gleason grading
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2023
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866910389228273664 |
|---|---|
| author | Kong, Fei Wang, Xiyue Xiang, Jinxi Yang, Sen Wang, Xinran Yue, Meng Zhang, Jun Zhao, Junhan Han, Xiao Dong, Yuhan Zhu, Biyue Wang, Fang Liu, Yueping |
| author_facet | Kong, Fei Wang, Xiyue Xiang, Jinxi Yang, Sen Wang, Xinran Yue, Meng Zhang, Jun Zhao, Junhan Han, Xiao Dong, Yuhan Zhu, Biyue Wang, Fang Liu, Yueping |
| contents | Artificial intelligence (AI) holds significant promise in transforming medical imaging, enhancing diagnostics, and refining treatment strategies. However, the reliance on extensive multicenter datasets for training AI models poses challenges due to privacy concerns. Federated learning provides a solution by facilitating collaborative model training across multiple centers without sharing raw data. This study introduces a federated attention-consistent learning (FACL) framework to address challenges associated with large-scale pathological images and data heterogeneity. FACL enhances model generalization by maximizing attention consistency between local clients and the server model. To ensure privacy and validate robustness, we incorporated differential privacy by introducing noise during parameter transfer. We assessed the effectiveness of FACL in cancer diagnosis and Gleason grading tasks using 19,461 whole-slide images of prostate cancer from multiple centers. In the diagnosis task, FACL achieved an area under the curve (AUC) of 0.9718, outperforming seven centers with an average AUC of 0.9499 when categories are relatively balanced. For the Gleason grading task, FACL attained a Kappa score of 0.8463, surpassing the average Kappa score of 0.7379 from six centers. In conclusion, FACL offers a robust, accurate, and cost-effective AI training model for prostate cancer pathology while maintaining effective data safeguards. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2302_06089 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Federated attention consistent learning models for prostate cancer diagnosis and Gleason grading Kong, Fei Wang, Xiyue Xiang, Jinxi Yang, Sen Wang, Xinran Yue, Meng Zhang, Jun Zhao, Junhan Han, Xiao Dong, Yuhan Zhu, Biyue Wang, Fang Liu, Yueping Computer Vision and Pattern Recognition Machine Learning Quantitative Methods Artificial intelligence (AI) holds significant promise in transforming medical imaging, enhancing diagnostics, and refining treatment strategies. However, the reliance on extensive multicenter datasets for training AI models poses challenges due to privacy concerns. Federated learning provides a solution by facilitating collaborative model training across multiple centers without sharing raw data. This study introduces a federated attention-consistent learning (FACL) framework to address challenges associated with large-scale pathological images and data heterogeneity. FACL enhances model generalization by maximizing attention consistency between local clients and the server model. To ensure privacy and validate robustness, we incorporated differential privacy by introducing noise during parameter transfer. We assessed the effectiveness of FACL in cancer diagnosis and Gleason grading tasks using 19,461 whole-slide images of prostate cancer from multiple centers. In the diagnosis task, FACL achieved an area under the curve (AUC) of 0.9718, outperforming seven centers with an average AUC of 0.9499 when categories are relatively balanced. For the Gleason grading task, FACL attained a Kappa score of 0.8463, surpassing the average Kappa score of 0.7379 from six centers. In conclusion, FACL offers a robust, accurate, and cost-effective AI training model for prostate cancer pathology while maintaining effective data safeguards. |
| title | Federated attention consistent learning models for prostate cancer diagnosis and Gleason grading |
| topic | Computer Vision and Pattern Recognition Machine Learning Quantitative Methods |
| url | https://arxiv.org/abs/2302.06089 |