arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.04035cs.LGcs.AI
收藏

STEMO:基于多目标强化学习的早期时空预测

STEMO: Early Spatio-temporal Forecasting with Multi-Objective Reinforcement Learning

Data61, CSIRO(Data61联邦科学与工业研究组织);RMIT University(皇家墨尔本理工大学);Xidian University(西安电子科技大学);Zhejiang University(浙江大学);University of New South Wales(新南威尔士大学)

Wei Shao, Yufan Kang, Ziyan Peng, Xiao Xiao, Lei Wang, Yuhui Yang, Flora D Salim

AI总结:

针对预测任务中准确性与时效性的平衡难题,提出基于多目标强化学习的早期时空预测模型STEMO,可适配偏好确定预测时机,在三个真实数据集上性能优于现有方法。

摘要:

在预测任务中,准确性和时效性确实往往是相互冲突的目标。过早预测可能会导致更高的误报率,而延迟预测以收集更多信息又可能使其为时过晚、失去效用。在野火、犯罪、交通拥堵等应用场景中,及时预测对于保障生命财产安全至关重要,因此在准确性和时效性之间取得平衡十分关键。本文提出一种基于多目标强化学习的早期时空预测模型STEMO,该模型既能根据给定偏好执行最优策略,也能基于少量样本推断偏好。模型解决了两大核心挑战:一是提升早期预测的准确性,二是提供最优策略以确定每个区域最合适的预测时机。该方法在三个大规模真实世界数据集上表现优异,在早期时空预测任务中性能超越现有方法。

英文摘要:

Accuracy and timeliness are indeed often conflicting goals in prediction tasks. Premature predictions may yield a higher rate of false alarms, whereas delaying predictions to gather more information can render them too late to be useful. In applications such as wildfires, crimes, and traffic jams, timely forecasting are vital for safeguarding human life and property. Consequently, finding a balance between accuracy and timeliness is crucial. In this paper, we propose an early spatio-temporal forecasting model based on Multi-Objective reinforcement learning that can either implement an optimal policy given a preference or infer the preference based on a small number of samples. The model addresses two primary challenges: 1) enhancing the accuracy of early forecasting and 2) providing the optimal policy for determining the most suitable prediction time for each area. Our method demonstrates superior performance on three large-scale real-world datasets, surpassing existing methods in early spatio-temporal forecasting tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑