CORTEXA
← Browse
arxivcs.LGcs.AI2026-06-29

Federated Learning with Energy-Based Structured Probabilistic Inference

Dario Fenoglio, Daniil Kirilenko, Martin Gjoreski, Marc Langheinrich

Federated learning typically aggregates client updates using fixed or heuristic weighting rules, which can be suboptimal when clients have heterogeneous data and varying contributions to the global model. We propose a framework that refines client aggregation weights using Conditional Random Fields (CRFs). Our method defines unary potentials for individual clients and pairwise potentials for all client pairs, allowing the server to model both client-specific reliability and interactions between clients. The resulting CRF inference produces aggregation weights that enable better convergence of the global training objective. Experiments show that, under non-IID heterogeneity, our approach consistently improves performance over well-established federated learning baselines.

View free PDFSource page

Related papers

arxivcs.LGcs.AI2026-07-14

Constraint-Aware Aggregation for Federated Reinforcement Learning in Microgrid Energy Coordination

Usman Haider, Karl Mason

Federated Reinforcement Learning (FedRL) enables coordination of distributed energy resources without sharing raw local data, but standard aggregation methods such as FedAvg do not account for system-level constraints, often leading to unsafe global behavior. In this work, we stu…

View free PDFSource page
arxivcs.LGcs.AI2026-07-14

Learning-based Probabilistic Load Forecasting with Post-hoc and In-model Uncertainty

Sarah Al-Shareeda, Gulcihan Ozdemir, Heung Seok Jeon

Smart-building load forecasters are often trained offline on dense, multivariate, high-frequency data, but deployment may provide only hourly, feature-limited inputs. Missing features must then be reconstructed, and their errors can propagate through the model. If this input unce…

View free PDFSource page
arxivcs.LGcs.AIcs.DCstat.ML2026-06-30

Entropy-Regularized Probabilistic Gates for Sparse Model Discovery in Scarce-Data Federated Learning

Krishna Harsha Kovelakuntla Huthasana, Alireza Olama, Andreas Lundell

Federated Learning (FL) is a distributed machine learning (ML) paradigm with collaboration among multiple clients without sharing data. FL is challenging under data heterogeneity and partial client participation. Learning sparse models is useful for communication and computationa…

View free PDFSource page
arxivcs.LGcs.AIcs.DC2026-07-03

Decentralised Federated Learning over Temporal Networks: The Role of Heterogeneities

Arash Badie-Modiri, Chiara Boldrini, Lorenzo Valerio, János Kertész, Márton Karsai

Decentralised federated learning, based on peer-to-peer communication, is increasingly proposed for on-device training of machine learning models, promising a privacy-preserving, communication-efficient training process with no risk of single-point failure. However, the role of s…

View free PDFSource page
arxivcs.LGcs.AI2026-07-21

Trustworthy Privacy-Preserving Multimodal Federated Learning for Personalised Breast Cancer Prediction

Ruth Amey, Muhammad Arifur Rahman, Taha Osman, Nicholas Shopland, Andy Burton, Mufti Mahmud, et al.

Federated learning has emerged as a potential solution to privacy concerns associated with using sensitive health data for training predictive models, particularly in personalised cancer care. This research investigates whether federated learning can support the development of ro…

View free PDFSource page