Issue |
ITM Web Conf.
Volume 70, 2025
2024 2nd International Conference on Data Science, Advanced Algorithm and Intelligent Computing (DAI 2024)
|
|
---|---|---|
Article Number | 03029 | |
Number of page(s) | 8 | |
Section | Image Processing and Computer Vision | |
DOI | https://doi.org/10.1051/itmconf/20257003029 | |
Published online | 23 January 2025 |
Hierarchical Learning: A Hybrid of Federated Learning and Personalization Fine-Tuning
1 Computer Science, University of Wisconsin-Madison, 53715 University Ave, Madison, WI, United States
2 Computer Science, Rensselaer Polytechnic Institute, 12180 People Ave, Troy, NY,, United States
* Corresponding author: Sli857@wisc.edu
Hierarchical Federated Learning (FL) presents a novel approach that combines global model training with localized personalization fine-tuning to enhance the predictive accuracy of decentralized machine learning systems. Traditional FL methods, which allow multiple clients to collaboratively train a global model without sharing raw data, are hindered by issues such as non-independent and identically distributed (non-IID) data, communication overhead, and limited generalization across diverse client datasets. This study proposes a hierarchical model that mitigates these challenges by incorporating a global model, trained using the Federated Averaging (FedAvg) algorithm, and applying client-specific fine-tuning to improve local model performance. The experiment conducted on a movie recommendation system demonstrates that this hierarchical approach significantly reduces the global model’s error while offering personalized improvements on client-specific datasets. Results show an average Root Mean Squared Error (RMSE) reduction of 0.0460 following local personalization. This hybrid approach not only enhances model accuracy but also preserves data privacy and increases scalability, making it a promising solution for decentralized recommendation systems.
© The Authors, published by EDP Sciences, 2025
This is an Open Access article distributed under the terms of the Creative Commons Attribution License 4.0, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Current usage metrics show cumulative count of Article Views (full-text article views including HTML views, PDF and ePub downloads, according to the available data) and Abstracts Views on Vision4Press platform.
Data correspond to usage on the plateform after 2015. The current usage metrics is available 48-96 hours after online publication and is updated daily on week days.
Initial download of the metrics may take a while.