用可解释的注意力机制提升内河船舶轨迹预测准确性
Towards Explainable Deep Learning for Ship Trajectory Prediction in Inland Waterways
- 基于LSTM与学习到的船域参数,实现交互船舶状态的注意力融合
- 5分钟预测下平均位移误差约40米,性能接近同类研究
- 模型设计天然可解释,适合需要可信决策的航运安全场景
在拥挤内河航道中准确预测船舶轨迹对交通安全至关重要。尽管深度学习在复杂场景下已显著提升预测精度,但模型可解释性常被忽视,可能掩盖错误逻辑并削弱可靠性。本研究通过引入训练得到的船域参数,分析基于注意力机制融合交互船舶隐藏状态的LSTM模型。该方法虽已在海上航运中探索,但内河航道遭遇形态更复杂多样,为模型可解释性分析提供了更深入视角。采用标准位移误差统计评估模型变体性能,并分析生成船域值的合理性。结果表明,在5分钟预测时长下,最终平均位移误差约为40米,表现与现有研究相当。虽然船间注意力结构提升了预测精度,但基于学习船域值分配给邻近船舶的权重与预期不符。因此,精度提升并非完全源于轨迹间的因果关联。这一发现凸显了模型通过内在可解释设计带来的解释能力。未来工作将探索该架构用于反事实分析及引入更先进的注意力机制。
原文摘要 · Abstract (English)
Accurate predictions of ship trajectories in crowded environments are essential to ensure safety in inland waterways traffic. Recent advances in deep learning promise increased accuracy even for complex scenarios. While the challenge of ship-to-ship awareness is being addressed with growing success, the explainability of these models is often overlooked, potentially obscuring an inaccurate logic and undermining the confidence in their reliability. This study examines an LSTM-based vessel trajectory prediction model by incorporating trained ship domain parameters that provide insight into the attention-based fusion of the interacting vessels' hidden states. This approach has previously been explored in the field of maritime shipping, yet the variety and complexity of encounters in inland waterways allow for a more profound analysis of the model's interpretability. The prediction performance of the proposed model variants are evaluated using standard displacement error statistics. Additionally, the plausibility of the generated ship domain values is analyzed. With an final displacement error of around 40 meters in a 5-minute prediction horizon, the model performs comparably to similar studies. Though the ship-to-ship attention architecture enhances prediction accuracy, the weights assigned to vessels in encounters using the learnt ship domain values deviate from the expectation. The observed accuracy improvements are thus not entirely driven by a causal relationship between a predicted trajectory and the trajectories of nearby ships. This finding underscores the model's explanatory capabilities through its intrinsically interpretable design. Future work will focus on utilizing the architecture for counterfactual analysis and on the incorporation of more sophisticated attention mechanisms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。