QoSDiff 面向服务可靠性预测的逆扩散表征学习与对抗双向混合注意力交互方法研究

Graduation Project · 2025

面向服务可靠性预测的逆扩散表征学习
与对抗双向混合注意力交互方法研究

本文基于 WS-DREAM 真实 QoS 数据集进行训练与测试，该数据集包含 339 个用户与 5825 个 Web 服务之间的服务调用记录。实验结果表明，在不同稀疏度设置下（Density 2.5%–10%），QoSDiff 相较于 QoSGNN 等基线方法取得了更优的预测性能。

作者:杜冠辰 (2022611013) · 指导教师:许建龙 · 汕头大学 数学与计算机学院

339Users
5825Services
≈ 1.97MInteraction samples
6.5%MAE ↓ vs Baseline

① 交互预测 · Globe View

在地球上选中一个用户(蓝色)和一个服务(橙色),实时对比 QoSDiff 与基线方法在 RT 上的预测。

User Service Selected Pair 点击地球上的节点以选择
Prediction Panel
User—
Service—
Distance—
Ground Truth RT — 来自 WS-DREAM rtMatrix
QoSDiff(本文) — Err —
QoSGNN(Baseline) — Err —
QoSDiff
QoSGNN
True

② 模型架构 · QoSDiff Pipeline

无需显式二分图;DELM 通过单步逆扩散从含噪嵌入恢复隐式拓扑,AAIM 用对抗式双向混合注意力建模高阶交互。

① Input Context User: ID · AS · Country Service: ID · Provider · Country aggregate id + multi-attr emb. ② DELM · Single-Step Reverse Diffusion 不构图;直接对嵌入做 T=1 去噪 x₁ (noisy) ε_θ predict noise Self-Attn + MLP (ε_θ) x₀ = z (clean) x₀ = (x₁ − √(1−α₁)·ε̂) / √α₁ → recovered z_u, z_s ③ AAIM · Adversarial Bi-Directional Hybrid Attention Real T = z_u ‖ z_s Fake F (Gumbel) Generator G MHA U→S user attends svc MHA S→U svc attends user concat · LN · FFN σ → ŷ Discriminator D MLP · BCE real vs fake min-max: L_total = L_MSE + λ · L_adv QoS ŷᵢⱼ RT / TP rescale
① Context Embed 聚合身份与多源上下文(AS / Country / Provider …),得到含噪初始向量 x₁。无显式二分图。
② DELM · 单步逆扩散 将 x₁ 视为理想嵌入经一次前向扩散的结果;ε_θ(Self-Attn + MLP) 预测噪声,一次性解出 x₀ = z。
③ AAIM 真实拼接 T 与 Gumbel-Softmax 伪样本 F 并行进入生成器(双向 MHA U→S / S→U);判别器做真伪监督。
④ 联合训练 L_total = MSE(ŷ, y) + λ·L_adv (λ=0.2),AdamW + 早停;1 attention head, 256-dim 为最佳配置。

③ 实验结果 · Benchmark

WS-DREAM 数据集 · 四种 Density (2.5% / 5% / 7.5% / 10%) 下与 12 种基线对比;指标取自论文表 4-1 / 4-2。最强基线为 QoSGNN。

单位:RT = 秒 (s) · TP = kbps
MAE ↓ (越低越好)RT
RMSE ↓ (越低越好)RT
详细指标表 · RT 12 baselines · QoSDiff 最优
Model D=2.5% D=5% D=7.5% D=10%
MAERMSE MAERMSE MAERMSE MAERMSE

④ 噪声鲁棒性 · Robustness

测试集注入 0%–25% 的身份扰动噪声(论文表 4-5 / 4-6),QoSDiff 相对 QoSGNN 退化幅度始终更低。

MAE vs Noise RatioRT · D=10%
性能退化率 Deg. (%)越低越鲁棒
抗噪小结 —

QoSDiff 将表征学习与显式图拓扑解耦:不存在可被污染的边, 因此身份扰动只会影响单个输入向量,不会沿消息传递放大。 结合 AAIM 的对抗注意力正则,模型在 5%–25% 噪声下退化率始终低于 QoSGNN。