arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.12770cs.CVcs.LGeess.IV
收藏

基于注意多任务迁移学习的视觉兴趣预测

Visual Interest Prediction with Attentive Multi-Task Transfer Learning

SUTD(新加坡科技设计大学);Indian Institute of Technology Patna(印度理工学院巴特那分校)

Deepanway Ghosal, Maheshkumar H. Kolekar

摘要:

视觉兴趣和情感预测是计算机视觉领域中一个非常有趣的研究领域。本文提出了一种基于迁移学习和注意力机制的神经网络模型来预测数码照片中的视觉兴趣和情感维度。学习的多维影响是通过一个多任务的学习框架。通过各种实验,我们证明了所提出的方法的有效性。在基准数据集上对我们的模型进行的评估表明,与当前最先进的系统相比,我们的模型有很大的改进。

英文摘要:

Visual interest & affect prediction is a very interesting area of research in the area of computer vision. In this paper, we propose a transfer learning and attention mechanism based neural network model to predict visual interest & affective dimensions in digital photos. Learning the multi-dimensional affects is addressed through a multi-task learning framework. With various experiments we show the effectiveness of the proposed approach. Evaluation of our model on the benchmark dataset shows large improvement over current state-of-the-art systems.

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

↑