基于多任务学习的机票价格预测模型

Airfare price prediction based on multi-task learning

针对现有机票销售模型忽视不同票价等级需求问题,考虑到机票需求的影响,提出一种多任务学习模型预测机票价格。在机票价格预测中引入辅助任务机票需求预测,建立多任务学习网络,通过共享相关任务在日、周、半月、月等水平上的多尺度需求特征,分析不同周期需求特征的影响。在六千万条记录的真实数据集上的实验结果表明,较之基准算法,该模型在准确率和F1分数方面提高了将近6%,验证了多任务学习模型的有效性。

To solve the problem that the existing ticket pricing strategy ignores the demand for different fare classes,a multi-task learning model for airfare price prediction was proposed to consider the impact of ticket demand.The ticket demand prediction was introduced as an auxiliary task to build a multi-task learning network in airfare price prediction.The impact of different periodic demand characteristics was analyzed by sharing the multi-scale demand characteristics of related tasks at the daily,weekly,semi-monthly,and monthly levels.Multiple experiments were conducted on real-word dataset with sixty millions of records.The results show that the improvements over the baselines in terms of accuracy and F1-score are nearly 6%,verifying the effectiveness of the multi-task learning model.