CSMF: Cascaded Selective Mask Fine-Tuning for Multi-Objective Embedding-Based Retrieval
Hao Deng, Haibo Xing, Kanefumi Matsuyama, Moyu Zhang, Jinxin Hu, Hong Wen, Yu Zhang, Xiaoyi Zeng, Jing Zhang
Abstract
Multi-objective embedding-based retrieval (EBR) has become increasingly critical due to the growing complexity of user behaviors and commercial objectives. While traditional approaches often suffer from data sparsity and limited information sharing between objectives, recent methods utilizing a shared network alongside dedicated sub-networks for each objective partially address these limitations. However, such methods significantly increase the model parameters, leading to an increased retrieval latency and a limited ability to model causal relationships between objectives. To address these challenges, we propose the Cascaded Selective Mask Fine-Tuning (CSMF), a novel method that enhances both retrieval efficiency and serving performance for multi-objective EBR. The CSMF framework selectively masks model parameters to free up independent learning space for each objective, leveraging the cascading relationships between objectives during the sequential fine-tuning. Without increasing network parameters or online retrieval overhead, CSMF computes a linearly weighted fusion score for multiple objective probabilities while supporting flexible adjustment of each objective's weight across various recommendation scenarios. Experimental results on real-world datasets demonstrate the superior performance of CSMF, and online experiments validate its significant practical value.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on5
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- SimCSE: Simple Contrastive Learning of Sentence EmbeddingsTianyu Gao, Xingcheng Yao, Danqi ChenEMNLP 2021 · 2,496 citations
- Gradient Surgery for Multi-Task LearningTianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine et al.NeurIPS 2020 · 2,261 citations
- DoRA: Weight-Decomposed Low-Rank AdaptationShih-Yang Liu, Chien-Yi Wang, Hongxu Yin, Pavlo Molchanov et al.ICML 2024 · 820 citations
- LoRA+: Efficient Low Rank Adaptation of Large ModelsSoufiane Hayou, Nikhil Ghosh, Bin YuICML 2024 · 388 citations
Related papers
- Unleashing the Potential of Multi-Channel Fusion in Retrieval for Personalized RecommendationsJunjie Huang, Jiarui Qin, Jianghao Lin, Ziming Feng et al.WWW 2025 · 9 citations
- AutoDCS: Automated Decision Chain Selection in Deep Recommender SystemsDugang Liu, Shenxian Xian, Yuhao Wu, Chaohua Yang et al.SIGIR 2024 · 8 citations
- Can Small Heads Help? Understanding and Improving Multi-Task GeneralizationYuyan Wang, Zhe Zhao, Bo Dai, Christopher Fifty et al.WWW 2022 · 15 citations
- Everyone's Preference Changes Differently: A Weighted Multi-Interest Model For RetrievalHui Shi, Yupeng Gu, Yitong Zhou, Bo Zhao et al.ICML 2023 · 15 citations
- Sparse Meets Dense: Unified Generative Recommendations with Cascaded Sparse-Dense RepresentationsYuhao Yang, Zhi Ji, Zhaopeng Li, Yi Li et al.NeurIPS 2025 · 90 citations
