🎉 Excited to share that this is our second paper published in the Journal of Artificial Intelligence Research (JAIR)!
📄 Formal Logic Inference Guided Uncertainty Quantification for Personalized Federated Learning
https://lnkd.in/grf_Vda7
Federated Learning (FL) has emerged as a powerful paradigm for training machine learning models while preserving data privacy. Yet, real-world deployments—such as smart grids, traffic forecasting, and IoT systems—must contend with highly heterogeneous client data, making personalization both challenging and computationally expensive.
In this work, we introduce LogiCP, a novel framework that combines formal logic reasoning with uncertainty quantification to enable scalable and personalized federated learning with theoretical guarantees.
Our key idea is to leverage Signal Temporal Logic (STL) to identify meaningful temporal behaviors and group clients based on semantic similarity rather than purely statistical distance. Within each cluster, we apply decentralized Conformal Prediction (CP) to generate distribution-free prediction intervals with mathematical coverage guarantees. LogiCP also supports dynamic client assignment, allowing new clients to join appropriate clusters at runtime without retraining.
Across three real-world applications—traffic forecasting, temperature prediction, and electricity demand forecasting—LogiCP consistently outperforms Bayesian, clustering-based, and conformal prediction baselines, achieving up to a 95% improvement in client-level MSE while maintaining strong scalability.
More broadly, this work demonstrates how formal methods can play a central role in modern AI—not only for verification, but also for guiding learning, improving personalization, and quantifying uncertainty. We hope LogiCP contributes toward building AI systems that are more accurate, scalable, and trustworthy for distributed, real-world applications.
Many thanks to our co-authors (Guocheng He, Ziyan An), collaborators, reviewers, and everyone who contributed to this work. I'm excited to continue exploring the intersection of formal methods, machine learning, and trustworthy AI.
#JAIR #FederatedLearning #MachineLearning #TrustworthyAI #FormalMethods #UncertaintyQuantification #ConformalPrediction #DistributedLearning