Feature Noise Resilient for QoS Prediction with Probabilistic Deep Supervision

TL;DR

This paper introduces PDS-Net, a novel probabilistic deep learning framework that effectively identifies and mitigates feature noise to enhance QoS prediction accuracy in web recommendation systems.

Contribution

The paper presents PDS-Net, a dual-branch probabilistic network with a noise recognition loss function, improving robustness against feature noise in QoS prediction models.

Findings

PDS-Net outperforms existing models with an average MAE improvement of around 8.9%.

The framework effectively identifies noisy features and refines their distributions.

Demonstrates robustness and versatility across real-world datasets.

Abstract

Accurate Quality of Service (QoS) prediction is essential for enhancing user satisfaction in web recommendation systems, yet existing prediction models often overlook feature noise, focusing predominantly on label noise. In this paper, we present the Probabilistic Deep Supervision Network (PDS-Net), a robust framework designed to effectively identify and mitigate feature noise, thereby improving QoS prediction accuracy. PDS-Net operates with a dual-branch architecture: the main branch utilizes a decoder network to learn a Gaussian-based prior distribution from known features, while the second branch derives a posterior distribution based on true labels. A key innovation of PDS-Net is its condition-based noise recognition loss function, which enables precise identification of noisy features in objects (users or services). Once noisy features are identified, PDS-Net refines the feature's prior distribution, aligning it with the posterior distribution, and propagates this adjusted distribution to intermediate layers, effectively reducing noise interference. Extensive experiments conducted on two real-world QoS datasets demonstrate that PDS-Net consistently outperforms existing models, achieving an average improvement of 8.91% in MAE on Dataset D1 and 8.32% on Dataset D2 compared to the ate-of-the-art. These results highlight PDS-Net's ability to accurately capture complex user-service relationships and handle feature noise, underscoring its robustness and versatility across diverse QoS prediction environments.