English

Self-Augmented Mixture-of-Experts for QoS Prediction

Machine Learning 2026-02-24 v3

Abstract

Quality of Service (QoS) prediction is one of the most fundamental problems in service computing and personalized recommendation. In the problem, there is a set of users and services, each associated with a set of descriptive features. Interactions between users and services produce feedback values, typically represented as numerical QoS metrics such as response time or availability. Given the observed feedback for a subset of user-service pairs, the goal is to predict the QoS values for the remaining pairs. A key challenge in QoS prediction is the inherent sparsity of user-service interactions, as only a small subset of feedback values is typically observed. To address this, we propose a self-augmented strategy that leverages a model's own predictions for iterative refinement. In particular, we partially mask the predicted values and feed them back into the model to predict again. Building on this idea, we design a self-augmented mixture-of-experts model, where multiple expert networks iteratively and collaboratively estimate QoS values. We find that the iterative augmentation process naturally aligns with the MoE architecture by enabling inter-expert communication: in the second round, each expert receives the first-round predictions and refines its output accordingly. Experiments on benchmark datasets show that our method outperforms existing baselines and achieves competitive results.

Keywords

Cite

@article{arxiv.2601.11036,
  title  = {Self-Augmented Mixture-of-Experts for QoS Prediction},
  author = {Kecheng Cai and Chao Peng and Chenyang Xu and Xia Chen and Yi Wang and Shuo Shi and Qiyuan Liang},
  journal= {arXiv preprint arXiv:2601.11036},
  year   = {2026}
}

Comments

There was an error in the test dataset leakage, leading to an inaccurate improvement magnitude. However, the method and framework remain valid. The paper and data will be revised and resubmitted

R2 v1 2026-07-01T09:07:08.056Z