Self-Augmented Mixture-of-Experts for QoS Prediction

Kecheng Cai, Chao Peng, Chenyang Xu, Xia Chen, Yi Wang +2 more
1/16/2026
cs.LG

No confident code implementation found for this paper.

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.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{cai2026selfaugmented,
  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},
  year   = {2026},
  eprint = {2601.11036},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.11036v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.11036.svg)](https://codeofpaper.com/paper/2601.11036)

Discussion