Package Arrival Time Prediction via Knowledge Distillation Graph Neural Network

Lei Zhang

doi:10.1145/3643033

Package Arrival Time Prediction via Knowledge Distillation Graph Neural Network

2024-02-28

0

PRE

AI

DOI：10.1145/3643033

Original request for help

Save

Abstract

En 中文

Accurately estimating packages' arrival time in e-commerce can enhance users' shopping experience and improve the placement rate of products. This problem is often formalized as an Origin-Destination (OD)-based ETA (i.e., estimated time of arrival) prediction task, where the delivery time is estimated mainly based on sender and receiver addresses and other context information. One inherent challenge of the OD-based ETA problem is that the delivery time highly depends on the actual delivery trajectory which is unknown at the time of prediction. In this article, we tackle this challenge by effectively exploiting historical delivery trajectories. We propose a novel Knowledge Distillation Graph neural network-based package ETA prediction (KDG-ETA) model, which uses knowledge distillation in the training phase to distill the knowledge of historical trajectories into OD pair embeddings. In KDG-ETA, a multi-level trajectory graph representation model is proposed to fully exploit trajectory information at the node-level, edge-level, and path-level. Then, the OD representations embedded with trajectory knowledge are combined with context embeddings from feature extraction module for delivery time prediction using an adaptive attention module. KDG-ETA consistently outperforms existing state-of-the-art OD-based ETA prediction methods on three real-world Alibaba datasets, reducing the Mean Absolute Error (MAE) by 3.0%-39.1% as demonstrated in our extensive empirical evaluation.

Keywords:

Package arrival time prediction

graph neural network

trajectory data mining

knowledge distillation

Package Arrival Time Prediction via Knowledge Distillation Graph Neural Network

Abstract

Journal

Organization