6.4　基于注意力机制的小学入学人数预测

随着我国统计数据的逐步完善，各年度学龄人口数、经济、人口数据逐渐开始累积，本节通过构建包含经济、人口等相关参量的多元时间序列来避免单一变量预测的局限性，同时挖掘学龄人口与各地经济、人口等相关特征和时序关系。与此同时，为了充分利用特征间的关联和时序间的依赖关系，本节在时间序列模型长短期记忆网络的基础上引入注意力（ATTENTION）机制。提出了基于ATTENTION-LSTM的小学入学规模预测模型，利用注意力机制对多元时间序列不同特征、不同时刻的特征值描述其对待预测指标的权重，提取特征在历史时间点与待预测指标之间的关系，提高小学入学规模预测模型的精度。