Adaptive Dual-Weighting Framework for Federated Learning via Out-of-Distribution Detection
Not on the map. Why?
Abstract (arXiv)
Federated Learning (FL) enables collaborative model training across large-scale distributed service nodes while preserving data privacy, making it a cornerstone of intelligent service systems in edge-cloud environments. However, in real-world service-oriented deployments, data generated by heterogeneous users, devices, and application scenarios are inherently non-IID. This severe data heterogeneity critically undermines the convergence stability, generalization ability, and ultimately the quality of service delivered by the global model. To address this challenge, we propose FLood, a novel FL framework inspired by out-of-distribution (OOD) detection. FLood dynamically counteracts the adverse effects of heterogeneity through a dual-weighting mechanism that jointly governs local training and global aggregation. At the client level, it adaptively reweights the supervised loss by upweighting pseudo-OOD samples, thereby encouraging more robust learning from distributionally misaligned or challenging data. At the server level, it refines model aggregation by weighting client contributions according to their OOD confidence scores, prioritizing updates from clients with higher in-distribution consistency and enhancing the global model's robustness and convergence stability. Extensive experiments across multiple benchmarks under diverse non-IID settings demonstrate that FLood consistently outperforms state-of-the-art FL methods in both accuracy and generalization. Furthermore, FLood functions as an orthogonal plug-in module: it seamlessly integrates with existing FL algorithms to boost their performance under heterogeneity without modifying their core optimization logic. These properties make FLood a practical and scalable solution for deploying reliable intelligent services in real-world federated environments.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 49 references that are in the map. First posted on arXiv 1 Feb 2026.
Builds on
26 of 49 references.
References · 26
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,773
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,325
- 2020Energy-based Out-of-distribution Detectioncited by 1,296
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 836
- 2021Model-Contrastive Federated Learningcited by 762
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 547
14 more
- 2021ReAct: Out-of-distribution Detection With Rectified Activationscited by 486
- 2019Scaling Out-of-Distribution Detection for Real-World Settingscited by 465
- 2022Extremely Simple Activation Shaping for Out-of-Distribution Detectioncited by 212
- 2022FedDC: Federated Learning with Non-IID Data via Local Drift Decoupling and Correctioncited by 181
- 2022Fine-tuning Global Model via Data-Free Knowledge Distillation for Non-IID Federated Learningcited by 172
- 2023FedDisco: Federated Learning with Discrepancy-Aware Collaborationcited by 66
- 2022Towards Understanding and Mitigating Dimensional Collapse in Heterogeneous Federated Learningcited by 45
- 2022FedBR: Improving Federated Learning on Heterogeneous Data via Local Learning Bias Reductioncited by 35
- 2022FedCross: Towards Accurate Federated Learning via Multi-Model Cross-Aggregationcited by 21
- 2023Is Aggregation the Only Choice? Federated Learning via Layer-wise Model Recombinationcited by 13
- 2025CADRef: Robust Out-of-Distribution Detection via Class-Aware Decoupled Relative Feature Leveragingcited by 8
- 2025Agentic Services Computingcited by 6
- 2021Efficient Federated Learning for AIoT Applications Using Knowledge Distillationcited by 5
- 2025Federated Learning for Commercial Image Sourcescited by 1
Built on by
Not cited yet.