Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (4): 751-762    DOI: 10.3785/j.issn.1008-973X.2026.04.007
计算机技术     
多尺度图卷积下的水漂垃圾轨迹预测模型
马龙1(),候永琪1(),吴佰靖1,高丽1,邓建伟2,闫光辉1
1. 兰州交通大学 电子与信息工程学院,甘肃 兰州 730070
2. 甘肃省水利科学研究院,甘肃 兰州 730000
Water-floating garbage trajectory prediction model based on multi-scale graph convolution
Long MA1(),Yongqi HOU1(),Baijing WU1,Li GAO1,Jianwei DENG2,Guanghui YAN1
1. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
2. Gansu Provincial Institute of Water Resources, Lanzhou 730000, China
 全文: PDF(1501 KB)   HTML
摘要:

针对水漂垃圾轨迹预测中单一尺度下时空异质性建模不足,导致预测结果不确定性高的问题,提出多尺度自适应图卷积模型MAGC-Trajectory. 构建自适应门控图卷积模块,将时空先验约束的静态邻接关系与数据驱动的动态拓扑结构进行跨域融合,提升模型对垃圾漂移的时序关系和轨迹波动的捕捉能力;设计多尺度时空交互模块,对空间特征进行时间尺度解耦,并与时序特征加权融合,增强垃圾轨迹时空异质性的表征能力;提出改进非线性学习层,使用可学习的自适应激活函数强化不同尺度时空特征的全局融合,生成具有统一表征的漂移轨迹高阶特征;设计概率预测层,使用均值-方差估计轨迹分布区间,量化预测结果的不确定性,提供更加鲁棒的预测轨迹. 在水漂垃圾轨迹数据集上的实验表明,相较于基准模型,所提模型的MAE、RMSE分别降低了0.000 2、0.000 5. 所提方法能够从决策上助力研究区域的水漂垃圾污染治理工作.

关键词: 水漂垃圾轨迹预测图卷积自适应邻接矩阵时空特征融合概率预测    
Abstract:

In response to the insufficient modeling of spatiotemporal heterogeneity under a single scale in water-floating garbage trajectory prediction, which often leads to high uncertainty in results, a multi-scale adaptive graph convolutional model (MAGC-Trajectory) was proposed. First, an adaptive gated graph convolution module was developed. Static adjacency relationships imposed by spatiotemporal priors were cross-domain integrated with data-driven dynamic topological structures. This enhanced the model’s ability to capture temporal dependencies of garbage drift and fluctuations of its trajectories. Next, a multi-scale spatiotemporal interaction module was designed. Spatial features were decoupled along different temporal scales. They were then fused in a weighted manner with temporal features. This strengthened the representation of spatiotemporal heterogeneity in garbage trajectories. Meanwhile, an improved nonlinear learning layer was introduced. A learnable adaptive activation function was employed to reinforce global fusion of multiscale spatiotemporal features. This resulted in high-order drift trajectory features with a unified representation. Lastly, a probabilistic prediction layer was constructed. Mean-variance estimation was utilized to estimate the trajectories’ distribution interval. This quantified predictive uncertainty and thus provided more robust predicted trajectories. Experiments were conducted on the water-floating garbage trajectory dataset. Results showed that, compared with the baseline model, the proposed approach reduced MAE by 0.000 2 and RMSE by 0.000 5. These improvements provide actionable support for decision-making in mitigating water-floating garbage pollution within the study area.

Key words: water-floating garbage    trajectory prediction    graph convolution    adaptive adjacency matrix    spatiotemporal feature fusion    probabilistic prediction
收稿日期: 2025-04-22 出版日期: 2026-03-19
CLC:  TV 697.3  
基金资助: 国家自然科学基金资助项目(62366028);甘肃省水利科学研究院项目基金(LZJT524289).
作者简介: 马龙(1983—),男,副教授,博士生,从事水利大数据、人工智能研究. orcid.org/0009-0004-3475-3710. E-mail:malong@mail.lzjtu.cn
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
马龙
候永琪
吴佰靖
高丽
邓建伟
闫光辉

引用本文:

马龙,候永琪,吴佰靖,高丽,邓建伟,闫光辉. 多尺度图卷积下的水漂垃圾轨迹预测模型[J]. 浙江大学学报(工学版), 2026, 60(4): 751-762.

Long MA,Yongqi HOU,Baijing WU,Li GAO,Jianwei DENG,Guanghui YAN. Water-floating garbage trajectory prediction model based on multi-scale graph convolution. Journal of ZheJiang University (Engineering Science), 2026, 60(4): 751-762.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.04.007        https://www.zjujournals.com/eng/CN/Y2026/V60/I4/751

图 1  水漂垃圾轨迹多尺度自适应图卷积模型(MAGC-Trajectory)
图 2  多尺度时空特征生成模块结构示意图
图 3  自适应门控图卷积
图 4  时空特征融合层结构示意图
图 5  MLP与KAN结构对比
图 6  鹦鸽嘴水库及黄河兰州段卫星图
轨迹采集时间地点轨迹点数目起点经度/(°E)起点纬度/(°N)终点经度/(°E)终点纬度/(°N)
轨迹111月鹦鸽嘴水库57399.833 099 0038.936 024 0099.850 564 9738.942 732 20
轨迹211月鹦鸽嘴水库1 87999.832 001 9438.934 474 4799.843 727 8138.938 661 67
轨迹311月鹦鸽嘴水库2 24499.831 926 3538.934 483 9099.850 458 2038.943 343 19
轨迹411月鹦鸽嘴水库1 70899.833 256 6738.936 136 9799.869 500 0538.955 534 76
轨迹55月黄河兰州段4 327103.769 908 2536.089 722 37103.817 942 836.070 174 72
轨迹65月黄河兰州段3 234103.740 453 6136.095 214 53103.765 875 036.090 130 10
表 1  水漂垃圾轨迹数据集轨迹点位置及数量信息
时间经度/(°E)纬度/(°N)经向速度/(m·s?1纬向速度/(m·s?1经向加速度/(m·s?2纬向加速度/(m·s?2
2024/11/3099.833 099 0038.936 024 000.118 678 8720.032 987 1590.011 867 8870.003 298 716
2024/11/3099.833 102 8138.936 010 280.133 466 4640.021 879 2470.013 346 6460.002 187 925
2024/11/3099.833 100 2838.935 994 850.124 292 8560.084 214 0350.012 429 2860.008 421 404
$\vdots $$\vdots $$\vdots $$\vdots $$\vdots $$\vdots $$\vdots $
2024/11/3099.850 561 5438.942 724 880.063 335 870.029 671 3620.006 333 5870.002 967 136
表 2  轨迹1数据展示
模型名称2号塑料类轨迹
MAERMSEPICPMPIW
GNN0.000 360 700.000 637 3510.003 2
GNN-MS0.000 236 090.000 312 6710.002 5
GNN-AG0.000 235 990.000 294 6110.002 5
GNN-FN0.000 157 570.000 196 1210.001 8
MAGC-Trajectory0.000 135 890.000 161 0310.001 9
表 3  消融模型指标对比
图 7  轨迹6预测结果对比
模型轨迹3(塑料)轨迹4(塑料)轨迹5(编织物)轨迹6(金属)
MAERMSEMAERMSEMAERMSEMAERMSE
ARIMA0.000 907 770.001 044 910.000 826 460.000 997 000.000 620 720.000 733 710.000 497 540.000 479 94
LSTM0.000 664 180.000 377 460.000 526 430.000 475 120.000 610 720.000 714 910.000 581 390.000 892 69
PSO-GRU0.000 268 660.000 303 530.001 282 100.001 528 820.001 282 580.001 788 720.002 556 010.004 059 56
Crossformer0.000 855 280.001 124 090.001 844 850.003 029 050.002 930 030.005 488 650.000 665 450.001 058 63
PatchTST0.000 235 040.000 339 290.000 357 850.000 382 010.000 227 430.000 325 600.000 268 740.000 431 20
CNN-LSTM0.000 206150.000 920 980.000 416 620.000 685 010.000 706 420.000 874 630.000 415 090.000 584 47
ASTGCN0.000 285 360.000 304 840.000 406 330.000 628 630.000 502 940.000 529 850.000 270 920.000 346 26
DCRNN0.000 196 310.000 242 670.000 426 840.000 605 680.000 322 170.000 288 780.000 151 100.000 212 69
K-GCN-LSTM0.000 471 820.000 542 800.001 911 420.002 420 090.000 279 030.000 199 160.001 057 480.001 440 52
MAGC-Trajectory0.000 135 890.000 161 030.000 285 290.000 331 460.000 241 920.000 251 920.000 097 490.000 131 98
表 4  不同对比算法的MAE与RMSE对比
图 8  水漂垃圾轨迹概率预测结果
1 仇威, 栾华龙, 渠庚, 等 三峡水库应急补水对2022年洪季长江口盐水入侵的影响[J]. 长江科学院院报, 2024, 41 (10): 30- 39
QIU Wei, LUAN Hualong, QU Geng, et al Impact of emergent water supply of the Three Gorges Reservoir on saltwater intrusion in the Changjiang River Estuary in 2022[J]. Journal of Changjiang River Scientific Research Institute, 2024, 41 (10): 30- 39
doi: 10.11988/ckyyb.20240616
2 张云, 王雨, 周绍辉, 等 星载GNSS-R检测太湖水华可行性分析[J]. 北京航空航天大学学报, 2024, 50 (3): 695- 705
ZHANG Yun, WANG Yu, ZHOU Shaohui, et al Analysis on feasibility of detecting water blooms in Taihu Lake with spaceborne GNSS-R[J]. Journal of Beijing University of Aeronautics and Astronautics, 2024, 50 (3): 695- 705
3 TAN Y, CHENG Q, LYU F, et al Hydrological reduction and control effect evaluation of sponge city construction based on one-way coupling model of SWMM-FVCOM: a case in university campus[J]. Journal of Environmental Management, 2024, 349: 119599
doi: 10.1016/j.jenvman.2023.119599
4 CASTÁN-LASCORZ M A, JIMÉNEZ-HERRERA P, TRONCOSO A, et al A new hybrid method for predicting univariate and multivariate time series based on pattern forecasting[J]. Information Sciences, 2022, 586: 611- 627
doi: 10.1016/j.ins.2021.12.001
5 潘金伟, 王乙乔, 钟博, 等 基于统计特征搜索的多元时间序列预测方法[J]. 电子与信息学报, 2024, 46 (8): 3276- 3284
PAN Jinwei, WANG Yiqiao, ZHONG Bo, et al Statistical feature-based search for multivariate time series forecasting[J]. Journal of Electronics and Information Technology, 2024, 46 (8): 3276- 3284
doi: 10.11999/JEIT231264
6 VERDONCK T, BAESENS B, ÓSKARSDÓTTIR M, et al Special issue on feature engineering editorial[J]. Machine Learning, 2024, 113 (7): 3917- 3928
doi: 10.1007/s10994-021-06042-2
7 CATON S, HAAS C Fairness in machine learning: a survey[J]. ACM Computing Surveys, 2024, 56 (7): 1- 38
8 AL-SELWI S M, HASSAN M F, ABDULKADIR S J, et al RNN-LSTM: from applications to modeling techniques and beyond: systematic review[J]. Journal of King Saud University - Computer and Information Sciences, 2024, 36 (5): 102068
doi: 10.1016/j.jksuci.2024.102068
9 刘凇佐, 王虔, 李磊, 等 粒子群优化的门控循环单元网络漂流浮标轨迹预测[J]. 电子与信息学报, 2024, 46 (8): 3295- 3304
LIU Songzuo, WANG Qian, LI Lei, et al Gated recurrent unit network of particle swarm optimization for drifting buoy trajectory prediction[J]. Journal of Electronics and Information Technology, 2024, 46 (8): 3295- 3304
doi: 10.11999/JEIT230945
10 吴跃高, 俞万能, 曾广淼, 等 融合拼接注意力机制的船舶轨迹预测方法[J]. 控制理论与应用, 2025, 42 (9): 1798- 1806
WU Yuegao, YU Wanneng, ZENG Guangmiao, et al Ship trajectory prediction method incorporating concatenated attention mechanism[J]. Control Theory and Applications, 2025, 42 (9): 1798- 1806
11 BAI J, ZHU J, SONG Y, et al A3T-GCN: attention temporal graph convolutional network for traffic forecasting[J]. ISPRS International Journal of Geo-Information, 2021, 10 (7): 485
doi: 10.3390/ijgi10070485
12 SRIRAMULU A, FOURRIER N, BERGMEIR C Adaptive dependency learning graph neural networks[J]. Information Sciences, 2023, 625: 700- 714
doi: 10.1016/j.ins.2022.12.086
13 BAI L, YAO L, LI C, et al Adaptive graph convolutional recurrent network for traffic forecasting[J]. Advances in Neural Information Processing Systems, 2020, 33: 17804- 17815
14 ZHOU Y, ZHENG H, HUANG X, et al Graph neural networks: taxonomy, advances, and trends[J]. ACM Transactions on Intelligent Systems and Technology, 2022, 13 (1): 1- 54
15 XU D, PENG H, TANG Y, et al Hierarchical spatio-temporal graph convolutional neural networks for traffic data imputation[J]. Information Fusion, 2024, 106: 102292
doi: 10.1016/j.inffus.2024.102292
16 LIU R W, LIANG M, NIE J, et al STMGCN: mobile edge computing-empowered vessel trajectory prediction using spatio-temporal multigraph convolutional network[J]. IEEE Transactions on Industrial Informatics, 2022, 18 (11): 7977- 7987
doi: 10.1109/TII.2022.3165886
17 ZHAO J, YAN Z, CHEN X, et al K-GCN-LSTM: a k-hop graph convolutional network and long-short-term memory for ship speed prediction[J]. Physica A: Statistical Mechanics and Its Applications, 2022, 606: 128107
doi: 10.1016/j.physa.2022.128107
18 TANG X, CHEN H, XIANG W, et al Short-term load forecasting using channel and temporal attention based temporal convolutional network[J]. Electric Power Systems Research, 2022, 205: 107761
doi: 10.1016/j.jpgr.2021.107761
19 邵海东, 肖一鸣, 邓乾旺, 等 基于不确定性感知网络的可信机械故障诊断[J]. 机械工程学报, 2024, 60 (12): 194- 206
SHAO Haidong, XIAO Yiming, DENG Qianwang, et al Trustworthy mechanical fault diagnosis using uncertainty-aware network[J]. Journal of Mechanical Engineering, 2024, 60 (12): 194- 206
doi: 10.3901/JME.2024.12.194
20 TANG S, LI B, YU H. ChebNet: efficient and stable constructions of deep neural networks with rectified power units via Chebyshev approximation [EB/OL]. (2024–10–14) [2025–04–20]. https://doi.org/10.1007/s40304-023-00392-0.
21 SHARMA K, LEE Y C, NAMBI S, et al A survey of graph neural networks for social recommender systems[J]. ACM Computing Surveys, 2024, 56 (10): 1- 34
22 BARAKBAYEVA T, DEMIRCI F M Fully automatic CNN design with inception and ResNet blocks[J]. Neural Computing and Applications, 2023, 35 (2): 1569- 1580
doi: 10.1007/s00521-022-07700-9
23 LIU Z, WANG Y, VAIDYA S, et al. KAN: Kolmogorov-arnold networks [EB/OL]. (2025−02−09) [2025−04−20]. https://arxiv.org/abs/2404.19756.
24 LAURINDO L C, MARIANO A J, LUMPKIN R An improved near-surface velocity climatology for the global ocean from drifter observations[J]. Deep Sea Research Part I: Oceanographic Research Papers, 2017, 124: 73- 92
doi: 10.1016/j.dsr.2017.04.009
25 ZHONG W, ZHAI D, XU W, et al Accurate and efficient daily carbon emission forecasting based on improved ARIMA[J]. Applied Energy, 2024, 376: 124232
doi: 10.1016/j.apenergy.2024.124232
26 ZHANG Y, YAN J. Crossformer: transformer utilizing cross-dimension dependency for multivariate time series forecasting [C]// International Conference on Learning Representations. Kigali: [S.n.], 2023: 1–21.
27 BAI L, YAO L, LI C, et al Adaptive graph convolutional recurrent network for traffic forecasting[J]. Advances in Neural Information Processing Systems, 2020, 33: 17804- 17815
[1] 陈文强,冯琳越,王东丹,顾玉磊,赵轩. 融合动态风险图与多变量注意力机制的车辆轨迹预测模型[J]. 浙江大学学报(工学版), 2026, 60(3): 455-467.
[2] 闫光辉,黄霄,常文文. 基于脑电多尺度特征和图神经网络的紧急制动行为识别[J]. 浙江大学学报(工学版), 2026, 60(2): 404-414.
[3] 周思瑶,夏楠,江佳鸿. 姿态引导的双分支换装行人重识别网络[J]. 浙江大学学报(工学版), 2026, 60(1): 71-80.
[4] 李宗民,徐畅,白云,鲜世洋,戎光彩. 面向点云理解的双邻域图卷积方法[J]. 浙江大学学报(工学版), 2025, 59(5): 879-889.
[5] 张梦瑶,周杰,李文婷,赵勇. 结合全局信息和局部信息的三维网格分割框架[J]. 浙江大学学报(工学版), 2025, 59(5): 912-919.
[6] 陈文强,王东丹,朱文英,汪勇杰,王涛. 基于时空图注意力网络的车辆多模态轨迹预测模型[J]. 浙江大学学报(工学版), 2025, 59(3): 443-450.
[7] 肖刚,卢大鹏,郑文博,程振波,张元鸣. 基于时空图注意力网络的多变量时序数据异常检测方法[J]. 浙江大学学报(工学版), 2025, 59(10): 2134-2143.
[8] 安春丽,张碧玲,赵国安,王博,刘岩. 基于FFT-CNN-GCN的电网故障诊断[J]. 浙江大学学报(工学版), 2025, 59(10): 2205-2212.
[9] 吴书晗,王丹,陈远方,贾子钰,张越棋,许萌. 融合注意力的滤波器组双视图图卷积运动想象脑电分类[J]. 浙江大学学报(工学版), 2024, 58(7): 1326-1335.
[10] 李劲业,李永强. 融合知识图谱的时空多图卷积交通流量预测[J]. 浙江大学学报(工学版), 2024, 58(7): 1366-1376.
[11] 邢志伟,朱书杰,李彪. 基于改进图卷积神经网络的航空行李特征感知[J]. 浙江大学学报(工学版), 2024, 58(5): 941-950.
[12] 何勇禧,韩虎,孔博. 基于多依赖图和知识融合的方面级情感分析模型[J]. 浙江大学学报(工学版), 2024, 58(4): 737-747.
[13] 向晓倩,陈璟. 基于双重注意力时空图卷积网络的行人轨迹预测[J]. 浙江大学学报(工学版), 2024, 58(12): 2586-2595.
[14] 王友卫,王炜琦,凤丽洲,朱建明,李洋. 基于广度-深度采样和图卷积网络的谣言检测方法[J]. 浙江大学学报(工学版), 2024, 58(10): 2040-2052.
[15] 申自浩,唐雨雨,王辉,刘沛骞,刘琨. 基于聚类和深度学习的车联网轨迹隐私保护机制[J]. 浙江大学学报(工学版), 2024, 58(1): 20-28.