中文

面向结构化输出预测的深层神经网络正则化

机器学习 2017-10-31 v6 机器学习

摘要

深层神经网络模型是学习表示的强大框架。通常,它用于通过利用输入 xx 中的规律性来学习关系 xyx \to y。在结构化输出预测问题中,yy 是多维的,且维度之间常存在结构关系。本工作的动机是学习可能存在于输出数据中的输出依赖关系,以提高预测精度。遗憾的是,前馈网络无法利用输出之间的关系。为克服此问题,本文提出一种利用多任务框架为这些特定任务训练神经网络的正则化方案。我们的方案旨在以无监督方式将输出表示 yy 的学习纳入训练过程,同时学习有监督映射函数 xyx \to y。我们在面部关键点检测问题(一种典型的结构化输出任务)上评估了我们的框架。我们在两个公开具有挑战性的数据集(LFPW 和 HELEN)上表明,我们的正则化方案改善了深层神经网络的泛化能力并加速了训练。我们还探索了未标记数据和仅标签数据的使用,显示出结果的进一步提升。我们提供了框架的开源实现(https://github.com/sbelharbi/structured-output-ae)。

关键词

引用

@article{arxiv.1504.07550,
  title  = {Deep Neural Networks Regularization for Structured Output Prediction},
  author = {Soufiane Belharbi and Romain Hérault and Clément Chatelain and Sébastien Adam},
  journal= {arXiv preprint arXiv:1504.07550},
  year   = {2017}
}

备注

Submitted to Neurocomputing, 8 figures