针对时间图中少数类分类难题,提出新型扩散框架提升少数类识别效果。
MDTE: Minority-Aware Diffusion over Temporal Edge Events for Imbalanced Node Classification

- 通过条件去噪重建稳定的时间边事件表示
- 少数类召回率最高提升23.53个百分点
- 适合处理不平衡节点分类任务的研究者参考
时间图上的类别不平衡节点分类面临挑战:多数类主导的时序传播会逐渐同化少数类表征,而传统节点与邻域信息对少数类缺乏足够判别力。为此,我们提出MDTE,一种少数类感知的扩散框架,通过条件扩散去噪重构稳定且具有判别性的时序边事件表示。具体而言,MDTE引入分布感知选择性传播,结合基于局部异常因子(LOF)的传播过滤与聚类感知低频传播,保留关键邻域依赖的同时缓解有害传播和多数类信息同化。进一步设计多视角判别融合模块,利用特征重建与拓扑预测刻画类别分布差异,提取互补判别信号以指导去噪过程。在五个真实数据集上的实验表明,MDTE在少数类指标上始终表现最优,相比最强基线,少数类召回率提升最高达23.53个百分点,F1提升8.68个百分点,AUPRC提升2.67个百分点。
原文摘要 · Abstract (English)
Class-imbalanced node classification on temporal graphs is challenging because majority-dominated temporal propagation progressively assimilates minority representations, while conventional node and neighborhood information provides insufficient discriminative evidence for minority classes. To address these issues, we propose MDTE, a minority-aware diffusion framework that reconstructs stable and discriminative temporal edge-event representations through conditional diffusion denoising. Specifically, MDTE introduces Distribution-Aware Selective Propagation, which combines Local Outlier Factor (LOF)-based propagation filtering with cluster-aware low-frequency propagation. The module preserves informative neighborhood dependencies while mitigating harmful propagation and majority-class information assimilation. It further develops Multi-View Discriminative Fusion, which exploits feature reconstruction and topology prediction to characterize class-wise differences in distribution learning and extracts complementary discriminability signals to guide denoising. Experiments on five real-world datasets demonstrate that MDTE consistently achieves the best performance on minority-class-oriented metrics, improving minority-class recall by up to 23.53 percentage points, minority-class F1 by 8.68 percentage points, and AUPRC by 2.67 percentage points over the strongest baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。