PIVEN：一种用于带特定值预测的预测区间深度神经网络
机器学习
2021-06-22 v3 机器学习
摘要
提升神经网络在回归任务中的鲁棒性是其应用于多个领域的关键。基于深度学习的方法旨在通过改进其对特定值（即点预测）的预测，或产生量化不确定性的预测区间（PIs）来实现这一目标。我们提出 PIVEN，一种同时产生 PI 和值预测的深度神经网络。我们的损失函数将值预测表示为上下界的函数，从而确保其值落在区间内而不增加模型复杂度。此外，我们的方法对 PI 内的数据分布不做任何假设，使其值预测对各类真实世界问题更为有效。在已知基准上的实验和消融测试表明，与当前最先进的 PI 生成方法相比，我们的方法产生了更紧的不确定性边界，同时保持与最先进值预测方法相当的性能。此外，我们超越先前工作，在评估中纳入了大型图像数据集，其中 PIVEN 与现代神经网络相结合。
引用
@article{arxiv.2006.05139,
title = {PIVEN: A Deep Neural Network for Prediction Intervals with Specific Value Prediction},
author = {Eli Simhayev and Gilad Katz and Lior Rokach},
journal= {arXiv preprint arXiv:2006.05139},
year = {2021}
}
备注
19 pages