arXiv:2510.10803cs.LGcs.AI2025-10

通过训练时动态删点,让模型自动找出交通数据中最关键的节点。

PruneGCRN: Minimizing and explaining spatio-temporal problems through node pruning

  • 训练中实时删减图节点,边学边筛核心信息
  • 删点后仍保持更高预测精度,比其他方法保留更多信息
  • 适合需要解释性与简化复杂时空问题的场景

本文针对深度学习模型在处理时空问题时如何剪枝图结构并融入可解释性的挑战,提出一种新方法。不同于传统在模型行为后添加解释,本工作旨在理解问题本身:通过在训练过程中动态移除图节点,构建一种能同时最小化预测误差并选择关键节点的优化剪枝机制。该机制使模型在训练中自主识别最相关的节点子集,从而提取问题的核心要素,提升可分析性。在多个主流交通数据集上的实验表明,相比现有方法,该方法在图规模缩小的同时能保留更多有效信息,验证了剪枝作为简化时空问题、挖掘关键成分的有效工具潜力。

原文摘要 · Abstract (English)

This work addresses the challenge of using a deep learning model to prune graphs and the ability of this method to integrate explainability into spatio-temporal problems through a new approach. Instead of applying explainability to the model's behavior, we seek to gain a better understanding of the problem itself. To this end, we propose a novel model that integrates an optimized pruning mechanism capable of removing nodes from the graph during the training process, rather than doing so as a separate procedure. This integration allows the architecture to learn how to minimize prediction error while selecting the most relevant nodes. Thus, during training, the model searches for the most relevant subset of nodes, obtaining the most important elements of the problem, facilitating its analysis. To evaluate the proposed approach, we used several widely used traffic datasets, comparing the accuracy obtained by pruning with the model and with other methods. The experiments demonstrate that our method is capable of retaining a greater amount of information as the graph reduces in size compared to the other methods used. These results highlight the potential of pruning as a tool for developing models capable of simplifying spatio-temporal problems, thereby obtaining their most important elements.

图神经网络时空建模模型剪枝可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。