Multi-Party Federated Recommendation Based on Semi-Supervised Learning

• Published in: IEEE transactions on big data Vol. 10; no. 4; pp. 356 - 370
• Main Authors: Liu, Xin, Lv, Jiuluan, Chen, Feng, Wei, Qingjie, He, Hangxuan, Qian, Ying
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 01.08.2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Algorithms; Big Data; Data models; Datasets; Federated learning; Machine learning; Machine learning algorithms; positive and unlabeled learning; Random sampling; Recommendation method; Reliability; Semi-supervised learning; Semisupervised learning; Training; vertical federated learning
• ISSN: 2332-7790; 2372-2096
• DOI: 10.1109/TBDATA.2023.3338009

Leveraging multi-party data to provide recommendations remains a challenge, particularly when the party in need of recommendation services possesses only positive samples while other parties just have unlabeled data. To address UDD-PU learning problem, this paper proposes an algorithm VFPU, Vertical Federated learning with Positive and Unlabeled data. VFPU conducts random sampling repeatedly from the multi-party unlabeled data, treating sampled data as negative ones. It hence forms multiple training datasets with balanced positive and negative samples, and multiple testing datasets with those unsampled data. For each training dataset, VFPU trains a base estimator adapted for the vertical federated learning framework iteratively. We use the trained base estimator to generate forecast scores for each sample in the testing dataset. Based on the sum of scores and their frequency of occurrence in the testing datasets, we calculate the probability of being positive for each unlabeled sample. Those with top probabilities are regarded as reliable positive samples. They are then added to the positive samples and subsequently removed from the unlabeled data. This process of sampling, training, and selecting positive samples is iterated repeatedly. Experimental results demonstrated that VFPU performed comparably to its non-federated counterparts and outperformed other federated semi-supervised learning methods.