返回
Federated learning enabled hotel customer classification towards imbalanced data
DOI:10.1016/j.asoc.2024.112028.png)
摘要
En 中文
Hotel customer classification is the basis of customer profiling, which can significantly benefit a hotel by providing more appropriate services for targeted customers. However, imbalanced data distribution from individual hotels cannot support a reliable classification result, and sharing personal information among hotels is not allowed. In this paper, we propose to achieve a privacy-preserved hotel customer classification model via federated learning. A significant challenge is that hotels with different star ratings or distributed in different city regions usually serve specific customer groups, resulting in imbalanced data that degrade classification accuracy. We introduce an attention mechanism and design a client selection strategy to balance global and local performance upon imbalanced data. Due to privacy issues, we evaluate our solution's communication cost and accuracy on public imbalanced datasets and demonstrate the real-world customer classification results. Extensive experiments show that our solution performs better than the COTA method.
Keyword:
Federated learning
Customer classification
Imbalanced data
期刊
IF:
6.6
论文数:
1.4W
被引数:
4.8W
机构
引用论文
Clustered Federated Learning: Model-Agnostic Distributed Multitask Optimization Under Privacy Constraints集群联邦学习: 隐私约束下的模型不可知的分布式多任务优化

