Skip to content
Open access

Formal Logic Inference Guided Uncertainty Quantification for Personalized Federated Learning

Jul 2026 · Journal of Artificial Intelligence Research · Vol 86 · 0 citations
Computer Science

TL;DR

LogiCP is a novel FL framework that integrates formal logic reasoning with uncertainty quantification (UQ) to support scalable and personalized learning with theoretical guarantees, achieving up to a 95% improvement in client-level MSE while maintaining strong scalability.

Abstract

Federated Learning (FL) enables privacy-preserving model training across heterogeneous distributed systems, such as smartgrid forecasting or traffic-flow prediction from geographically dispersed sensors and devices. A key challenge in such settings is capturing client-specific patterns while addressing data heterogeneity and uncertainty at scale. Existing approaches, including Bayesian Neural Networks (BNNs) and clustering-based methods, struggle with scalability and consistent personalization. We propose LogiCP, a novel FL framework that integrates formal logic reasoning with uncertainty quantification (UQ) to support scalable and personalized learning with theoretical guarantees. LogiCP uses Signal Temporal Logic (STL) to extract temporal patterns and form semantically coherent client clusters, controlling intra-cluster heterogeneity. Within each cluster, LogiCP applies decentralized Conformal Prediction (CP) to produce distribution-free prediction intervals with mathematical guarantees that encompass the real value. LogiCP dynamically assigns clients to clusters at runtime without retraining, improving practicality. Evaluations on three real-world datasets—traffic, temperature, and electricity—show that LogiCP consistently outperforms BNN-, clustering-, and CP-based baselines, achieving up to a 95% improvement in client-level MSE while maintaining strong scalability.

Read PDF

Similar papers

Open access Jul 2026

Privacy-preserving load forecasting in smart grids using federated learning: a comparative analysis of aggregation strategies

Experimental results on real-world energy consumption datasets demonstrate that the proposed FL framework achieves competitive forecasting accuracy while preserving client data privacy, and a rigorous comparative analysis reveals that FedProx and FedTrimmedAvg consistently outperform FedAvg under non-IID conditions.

A. Tibermacine, Ilyes Naidji, Imad Eddine Tibermacine et al. · 1 citation
Conference Jul 2026

A Causal-Driven Hierarchical Decentralised Federated Learning Framework for Resilient Load Forecasting in Distributed Microgrids

Modern microgrids require distributed intelligence and edge computing to handle variable demand and renewable generation, but heterogeneity, communication limits, and privacy hinder centralised forecasting. This paper proposes a causally guided hierarchical decentralised federated learning (H-DFL) framework for resilient short-term load forecasting, integrating a hybrid TCN–BiLSTM with MCMC-based probabilistic causal feature selection. A three-tier architecture enables local training and hierarchical aggregation without raw data sharing, improving scalability and communication efficiency through sparse, interpretable feature selection driven by key factors such as solar and weather dynamics. Experiments on the Ausgrid dataset show improved stability and efficiency over Granger Causality (GC), Dynamic Causal Modeling (DCM), and Markov chain Monte Carlo (MCMC) baselines, with intervention tests confirming robustness under solar, demand, and outage disturbances. Overall, the results demonstrate that combining probabilistic feature sparsity with hierarchical decentralised federated learning to enable scalable, privacy-preserving, and resilient load forecasting for future microgrid systems.

M. Mahi, R. Naha, Alistair Barros · 0 citations
Preprint Aug 2026

Beyond Parameter Space: NTK-Guided Personalized Aggregation for Robust Federated Learning

Local Inference Guided Aggregation for Heterogeneous Training Environments to Yield Enhancement Through Agreement and Regularization (LIGHTYEAR), a federated learning framework that performs update selection in function space using an NTK-based agreement score to characterize predictive behavior and determine a personalized aggregation set for each client.

Mirko Konstantin, S. Zachow, Anirban Mukhopadhyay · 0 citations
#artificial intelligence Preprint Sep 2026

Privacy-Preserving Heterogeneous Multi-LLM Federated Inference for Cognitive Diagnosis

Significant challenges remain in AI-driven educational systems in balancing privacy preservation with accurate cognitive diagnosis. To overcome this, we propose a federated inference framework in which several commercial LLM APIs collaborate without requiring access to raw student data or proprietary model internals. Using multiple federated entities, such as LLaMA-3.3-70B, GPT-4o-mini, and Claude-3-Haiku, our framework builds upon a heterogeneous multi-LLM architecture. The predictions generated by these entities are combined with epsilon-local differential privacy by adding Laplace noise locally to each entity's prediction output before aggregation, while residual-based aggregation mitigates model heterogeneity. Our approach is predicated on an honest-but-curious trust paradigm in which API providers are presumed not to abuse submitted queries, and our differential privacy mechanism shields the published diagnostic results from external inference. We conduct rigorous privacy-utility analysis showing strong privacy guarantees with minimal accuracy loss, and extensive real-world evaluations across three educational benchmarks confirm the framework's practical usability and cross-domain generalizability.

Yagna Manasa Boyapati, Chong Yu, Tian-Yu Jiang et al. · 0 citations
2025

Personalized Federated Conformal Prediction with Localization

Personalized federated learning addresses data heterogeneity across distributed agents but lacks uncertainty quantification that is both agent-specific and instance-specific, which is a critical requirement for risk-sensitive applications. We pro-pose personalized federated conformal prediction (PFCP), a novel framework that combines personalized federated learning with conformal prediction to provide statistically valid agent-personalized prediction sets with instance-localization. By leveraging privacy-preserving knowledge transfer from other source agents, PFCP ensures marginal coverage guarantees for target agents while significantly improving conditional coverage performance on individual test instances, which has been validated by extensive experiments.

Yinjie Min, Chuchen Zhang, Liuhua Peng et al. · 3 citations · ⚡1

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.