中文

ARMDN:用于电商零售需求预测的关联与循环混合密度网络

机器学习 2018-03-19 v2 人工智能 机器学习

摘要

准确的需求预测可帮助在线零售组织更好地规划其供应链流程。然而,挑战在于大量关联因素导致需求出现巨大的非平稳偏移,传统时间序列与回归方法难以建模。在本文中,我们提出一种称为 AR-MDN 的神经网络架构,其同时建模关联因素、时间序列趋势以及需求中的方差。我们首先识别若干因果特征,并使用特征嵌入、MLP 与 LSTM 的组合来表示它们。随后我们将输出密度建模为学习得到的高斯分布混合。AR-MDN 可端到端训练,无需额外监督。我们在来自 Flipkart 的涵盖数万产品、时长一年的数据集上进行了实验。与现有替代方法相比,所提架构在预测精度上取得了显著提升。

关键词

引用

@article{arxiv.1803.03800,
  title  = {ARMDN: Associative and Recurrent Mixture Density Networks for eRetail Demand Forecasting},
  author = {Srayanta Mukherjee and Devashish Shankar and Atin Ghosh and Nilam Tathawadekar and Pramod Kompalli and Sunita Sarawagi and Krishnendu Chaudhury},
  journal= {arXiv preprint arXiv:1803.03800},
  year   = {2018}
}