基于注意力多任务迁移学习的视觉兴趣预测
计算机视觉与模式识别
2020-05-28 v2 机器学习
图像与视频处理
摘要
视觉兴趣与情感预测是计算机视觉领域中一个非常有趣的研究方向。本文提出一种基于迁移学习与注意力机制的神经网络模型,以预测数字照片中的视觉兴趣与情感维度。多维情感的学习通过多任务学习框架来解决。通过多种实验,我们展示了所提方法的有效性。在基准数据集上对我们模型的评估显示,相较当前最先进(SOTA)系统有大幅改进。
引用
@article{arxiv.2005.12770,
title = {Visual Interest Prediction with Attentive Multi-Task Transfer Learning},
author = {Deepanway Ghosal and Maheshkumar H. Kolekar},
journal= {arXiv preprint arXiv:2005.12770},
year = {2020}
}