基于注意力机制和强化学习的疾病风险预测方法

本发明公开了一种基于注意力机制和强化学习的疾病风险预测方法，采用交叉注意力计算特征和标签之间的相关性大小，采用自注意力计算特征之间相关性大小，并结合互信息筛选特征；交叉注意力网络可以根据不同的标签动态调整特征的重要性，针对每个标签单独计算特征的重要性，从而实现更精细的特征选择，为特征选择提供了新方法和新思路；本发明将特征选择与模型训练耦合，通过外部奖励实时反馈至DQN模型，实现了端到端的优化。最终将筛选出的特征用于疾病风险预测，达到了采用少量体检特征就能较好的预测疾病的效果。