{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2603-18083","title":"Probabilistic Federated Learning on Uncertain and Heterogeneous Data with Model Personalization","arxiv_id":"2603.18083","date":"2026-03-18","proceeding":null,"authors":["Ratun Rahman","Dinh C. Nguyen"],"abstract":"Conventional federated learning (FL) frameworks often suffer from training degradation due to data uncertainty and heterogeneity across local clients. Probabilistic approaches such as Bayesian neural networks (BNNs) can mitigate this issue by explicitly modeling uncertainty, but they introduce additional runtime, latency, and bandwidth overhead that has rarely been studied in federated settings. To address these challenges, we propose Meta-BayFL, a personalized probabilistic FL method that combines meta-learning with BNNs to improve training under uncertain and heterogeneous data. The framework is characterized by three main features: (1) BNN-based client models incorporate uncertainty across hidden layers to stabilize training on small and noisy datasets, (2) meta-learning with adaptive learning rates enables personalized updates that enhance local training under non-IID conditions, and (3) a unified probabilistic and personalized design improves the robustness of global model aggregation. We provide a theoretical convergence analysis and characterize the upper bound of the global model over communication rounds. In addition, we evaluate computational costs (runtime, latency, and communication) and discuss the feasibility of deployment on resource-constrained devices such as edge nodes and IoT systems. Extensive experiments on CIFAR-10, CIFAR-100, and Tiny-ImageNet show that Meta-BayFL consistently outperforms state-of-the-art methods, including both standard and personalized FL approaches (e.g., pFedMe, Ditto, FedFomo), with up to 7.42\\% higher test accuracy.","url_abs":"https://arxiv.org/abs/2603.18083","url_pdf":"https://arxiv.org/pdf/2603.18083","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2603.18083","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2603.18083"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/Ratun11/Meta-BayFL","reach":null}],"summary":{"unverified":6},"by_repo_kind":{"found_in_text":{"samples":6,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"c18ec5de2f7b7401","entry":"negative_log_likelihood","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"matabayfl/models.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/matabayfl/models.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c18ec5de2f7b7401"}},{"code_sha256_prefix":"0e70a02a4461a121","entry":"nll","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"DFL/meta_iid_fed_bnn.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/DFL/meta_iid_fed_bnn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"0e70a02a4461a121"}},{"code_sha256_prefix":"103d106d2a55d89b","entry":"train_client_model","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"DFL/FL.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/DFL/FL.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"103d106d2a55d89b"}},{"code_sha256_prefix":"49174cea15af7817","entry":"train_client_model","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"DFL/FL_everyClient.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/DFL/FL_everyClient.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"49174cea15af7817"}},{"code_sha256_prefix":"c8a4c01335813d80","entry":"train_client_model","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"DFL/dfl.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/DFL/dfl.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c8a4c01335813d80"}},{"code_sha256_prefix":"800a6e24e9cb0d8f","entry":"train_client_model","repo":"Ratun11/Meta-BayFL","repo_kind":"found_in_text","path":"DFL/dfl2.py","file_url":"https://github.com/Ratun11/Meta-BayFL/blob/HEAD/DFL/dfl2.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"800a6e24e9cb0d8f"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}