Intelligent Handling of Noise in Federated Learning with Co-training for Enhanced Diagnostic Precision

Farah Farid Babar, Faisal Jamil, Faiza Fareed Babar

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Federated learning (FL) allows multiple distributed clients to train a model while protecting their data. Medical data, especially brain MRIs, might be misdiagnosed due to capture noise and scanner abnormalities. Existing noise-handling technologies use data transmission, raising communication burdens and privacy risks. To address these challenges, we propose a novel Adaptive Sample Weighting Federated Learning (ASW-FL) approach incorporating co-training into the FL framework. The local and global models in FL have different learning abilities, which we use to our advantage. The two models “teach each other” to ignore noisy labels by exchanging samples with their confident predictions. Our method improved accuracy from 83.05% to 85.20% using various aggregation algorithms on a benchmark dataset of 1300 brain MRIs and our own Biobank UK data. Our methodology for accurate, privacy-preserving medical image analysis is adequate. The proposed model is precise but requires more processing resources, making it more appropriate for powerful servers than personal devices.

Original languageEnglish
Title of host publicationComputational Collective Intelligence
Subtitle of host publication16th International Conference, ICCCI 2024, Proceedings
EditorsNgoc Thanh Nguyen, Bogdan Franczyk, André Ludwig, Manuel Núñez, Jan Treur, Gottfried Vossen, Adrianna Kozierkiewicz
PublisherSpringer, Cham
Pages279-291
Number of pages13
Volume14810
ISBN (Electronic)9783031708169
ISBN (Print)9783031708152
DOIs
Publication statusPublished - 6 Sep 2024
Event16th International Conference on Computational Collective Intelligence - Leipzig, Germany
Duration: 9 Sep 202411 Sep 2024
Conference number: 16

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
PublisherSpringer
Volume14810 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference16th International Conference on Computational Collective Intelligence
Abbreviated titleICCCI 2024
Country/TerritoryGermany
CityLeipzig
Period9/09/2411/09/24

Cite this