Bayesian Similarity-Weighted Aggregation for Tumor Segmentation | Muslims in ML at NeurIPS'24
Title: Bayesian Similarity-Weighted Aggregation for Federated Brain Tumor Segmentation
Authors: Muhammad Irfan Khan, Suleiman A. Khan, Elina Kontio, Mojtaba Jafaritadi
Abstract: We propose a Bayesian generative approach, Bayesian Similarity-weighted Aggregation (SimAgg), for combining model weights from federated collaborators in brain lesion segmentation. This method effectively adapts to data variability and incorporates probabilistic modeling to handle uncertainty, enhancing robustness in federated learning (FL). Using a novel multi-armed bandit setup, it dynamically selects collaborators to improve aggregation quality. Simulation results on multi-parametric MRI data show that Bayesian SimAgg achieves high Dice scores across tumor regions and converges approximately twice as fast as non-Bayesian methods, providing an effective framework for federated brain tumor segmentation.
Read Full Paper: https://openreview.net/forum?id=VVHkC6U1Je&referrer=%5BAuthor%20Console%5D(%2Fgroup%3Fid%3DNeurIPS.cc%2F2024%2FWorkshop%2FMusIML%2FAuthors%23your-submissions)
For more information: https://neurips.cc/virtual/2024/affinity-event/105018
Muslims in ML Workshop Website: https://www.musiml.org/
Видео Bayesian Similarity-Weighted Aggregation for Tumor Segmentation | Muslims in ML at NeurIPS'24 канала Muslims In ML
Authors: Muhammad Irfan Khan, Suleiman A. Khan, Elina Kontio, Mojtaba Jafaritadi
Abstract: We propose a Bayesian generative approach, Bayesian Similarity-weighted Aggregation (SimAgg), for combining model weights from federated collaborators in brain lesion segmentation. This method effectively adapts to data variability and incorporates probabilistic modeling to handle uncertainty, enhancing robustness in federated learning (FL). Using a novel multi-armed bandit setup, it dynamically selects collaborators to improve aggregation quality. Simulation results on multi-parametric MRI data show that Bayesian SimAgg achieves high Dice scores across tumor regions and converges approximately twice as fast as non-Bayesian methods, providing an effective framework for federated brain tumor segmentation.
Read Full Paper: https://openreview.net/forum?id=VVHkC6U1Je&referrer=%5BAuthor%20Console%5D(%2Fgroup%3Fid%3DNeurIPS.cc%2F2024%2FWorkshop%2FMusIML%2FAuthors%23your-submissions)
For more information: https://neurips.cc/virtual/2024/affinity-event/105018
Muslims in ML Workshop Website: https://www.musiml.org/
Видео Bayesian Similarity-Weighted Aggregation for Tumor Segmentation | Muslims in ML at NeurIPS'24 канала Muslims In ML
Комментарии отсутствуют
Информация о видео
12 декабря 2024 г. 7:20:06
00:09:10
Другие видео канала