Fair and Robust Federated Learning
Research on fairness and robustness in federated learning for distributed systems
In settings characterized by edge nodes that collect heterogeneous and complex data, distributed training offers an effective solution for building collective machine learning models while safeguarding the privacy of local information. My research addresses some fundamental challenges of federated training, operating in non-ideal scenarios where participants have limited computational resources and may introduce noise, inaccurate labels, or exhibit malicious, uncooperative, or evasive behavior.
Selfishness and fairness in federated learning
A core focus of my research involves protecting global FL models from dishonest or non-cooperative participants. In decentralized deployments, “selfish” clients (Augello et al., 2024) may intentionally manipulate local training updates to skew the global model toward their specific local data distribution. This behavior lead to biased global models that fail to generalize across the entire population, resulting in unfair treatment of other participants and potentially degrading overall model performance. To counter this, while recovering useful information from selfish clients, I have worked on robust aggregation strategies to recover the true updates of selfish participants, while also ensuring fairness in the global model by preventing any single participant from disproportionately influencing the outcome (Augello et al., 2026).
On the other hand, if data heterogeneity causes incompatible updates, I have also explored the used of dynamic clustering of participants (Augello et al., 2023) to group clients with similar data distributions, allowing for the creation of specialized sub-models that better capture the diversity of the data while still contributing to the overall federated learning process. This approach is meant to provide each participant with a model that is more aligned with their local data distribution, quickly adapting to changes in the local data, improving both fairness and performance across the network.
Resource-aware federated learning
Battery constraints often lead to client dropouts and uneven participation. To make federated learning practical for hardware-constrained edge devices, I have worked on resource-aware frameworks (Augello et al., 2025) to coordinates battery-conscious nodes using intelligent client and data selection mechanisms, prolonging node lifespan and ensures equitable energy consumption across participating edge devices without sacrificing global model quality.
Hardware dishomogeneity can also limit the effectiveness of federated learning: either the model is too expensive for low-end devices, limiting how many clients can participate, or the model is too small to capture the complexity of the data. To address this, I have explored the use of heterogeneous model architectures (Augello et al., 2025) that allow devices with different computational capabilities to contribute to the training process, enabling a more inclusive and effective federated learning ecosystem. A similar issue arises for the feature extraction process, where some clients might not be able to extract all the features from their local data due to hardware limitations. In this context, I have explored heterogeneous graph neural networks (Augello et al., 2026) that allow clients to extract different sets of features from their local data, while still contributing to the global model training process.
References
2026
2025
- Federated hyperdimensional ensembles for mobile malware detectionIn CEUR Workshop Proceedings - Joint Proceedings of the Thematic Workshops at Ital-IA 2025 colocated with the 5th National Conference on Artificial Intelligence, organized by CINI (Ital-IA 2025) , 2025