Tweedie head

新的 Tweedie 头提升了稀疏船舶交通预测的 ST-GNN 性能

研究人员开发了一种新的即插即用输出模块——可学习 Tweedie 头，旨在增强时空图神经网络 (ST-GNNs) 在船舶交通流预测方面的能力。该模块专门解决了稀疏和间歇性海事数据带来的挑战，这些数据通常会导致传统的 ST-GNNs 产生过于保守的预测。通过优化 Tweedie 单元偏差和学习节点级方差，新的头部提高了预测精度，尤其是在非零事件方面，这在利用洛杉矶和长滩港口真实 AIS 数据进行的实验中得到了证明。