FedMM: Federated Collaborative Signal Quantization for Multi-Market CTR Prediction
Jun Zhang, Dugang Liu, Xing Tang, Xiuqiang He, Zhong Ming
Abstract
Online platforms such as Amazon and Netflix serve users across multiple countries and regions, underscoring the importance of multi-market recommendation (MMR). Most MMR methods adopt a pre-training and fine-tuning paradigm, in which a unified model is first trained on centralized, global data and subsequently adapted to specific markets. However, this approach ignores the privacy of market data. While traditional federated learning preserves privacy, it typically aims to obtain a global model by aggregating model parameters and does not account for significant market heterogeneity. Additionally, because ID spaces are disjoint across markets, embedding-based aggregation strategies become ineffective. To overcome these challenges, we propose a federated collaborative signal quantization (FedMM) method for multi-market click-through rate (CTR) prediction. Our core idea leverages a discrete codebook mechanism to achieve privacy-preserving transmission and align disjoint ID spaces. We further employ a hierarchical codebook structure to capture cross-market shared patterns and market-specific characteristics. Specifically, we deploy a residual quantized variational autoencoder (RQ-VAE) with a dual-layer codebook mechanism for each market to quantize collaborative embeddings. The first layer utilizes a global federated codebook, updated via aggregation to capture universally shared collaborative patterns, while the second layer maintains a local codebook to learn market-specific semantics. Finally, the learned discrete codes, which integrate both general and specific collaborative signals, are incorporated into downstream CTR models to enhance prediction accuracy across all markets. Extensive experiments on benchmark datasets demonstrate that FedMM significantly improves recommendation performance with privacy guarantees.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on14
- LightGCN: Simplifying and Powering Graph Convolution Network for RecommendationXiangnan He, Kuan Deng, Xiang Wang, Yan Li et al.SIGIR 2020 · 4,448 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Tackling the Objective Inconsistency Problem in Heterogeneous Federated OptimizationJianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi et al.NeurIPS 2020 · 2,231 citations
- Personalized Cross-Silo Federated Learning on Non-IID DataYutao Huang, Lingyang Chu, Zirui Zhou, Lanjun Wang et al.AAAI 2021 · 816 citations
- Recommender Systems with Generative RetrievalShashank Rajput, Nikhil Mehta, Anima Singh, Raghunandan Hulikal Keshavan et al.NeurIPS 2023 · 474 citations
Related papers
- FeDecider: An LLM-Based Framework for Federated Cross-Domain RecommendationXinrui He, Ting-Wei Li, Tianxin Wei, Xuying Ning et al.WWW 2026 · 2 citations
- Vertical Federated Graph Neural Network for Recommender SystemPeihua Mai, Yan PangICML 2023 · 33 citations
- Efficient-FedRec: Efficient Federated Learning Framework for Privacy-Preserving News RecommendationJingwei Yi, Fangzhao Wu, Chuhan Wu, Ruixuan Liu et al.EMNLP 2021 · 50 citations
- Personalized Federated Collaborative Filtering: A Variational AutoEncoder ApproachZhiwei Li, Guodong Long, Tianyi Zhou, Jing Jiang et al.AAAI 2025 · 22 citations
- FedCIA: Federated Collaborative Information Aggregation for Privacy-Preserving RecommendationMingzhe Han, Dongsheng Li, Jiafeng Xia, Jiahao Liu et al.SIGIR 2025 · 11 citations
