计算机与现代化

• 图像处理 • 上一篇    下一篇

类别不平衡的多任务人脸属性识别

  

  1. (南京航空航天大学计算机科学与技术学院,江苏南京210016)
  • 收稿日期:2017-12-05 出版日期:2018-07-05 发布日期:2018-07-05
  • 作者简介:张文(1992-),男,安徽铜鼓人,南京航空航天大学计算机科学与技术学院硕士研究生,研究方向:计算机视觉,深度学习。
  • 基金资助:
    国家自然科学基金资助项目(61672280,61373060,61732006);江苏省333高层次人才培养工程项目(BRA2017377)

Multi-taskFaceAttributeRecognitionBasedonUnbalancedClasses

  1. (CollegeofComputerScienceandTechnology,NanjingUniversityofAeronauticsandAstronautics,Nanjing210016,China)
  • Received:2017-12-05 Online:2018-07-05 Published:2018-07-05

摘要: 属性的识别对物体的识别起到了比较重要的作用,例如人脸验证和场景识别。提高属性的识别率对后面基于属性特征的应用的正确率有很大的影响。近些年来,有些工作也开始关注于属性的学习,而很多的工作都是基于属性之间独立的假设,但在实际中很多的属性都是强相关的,例如没有胡子和女性,光头和头发的颜色;很多的工作忽略了类别之间的不平衡性,例如光头的样本比例可能只占样本的很小一部分。基于这2个观察,本文提出一种基于多任务的类别不平衡的人脸属性识别网络架构,该网络结构是由Densenet修改而来。该方法比以往的方法效果要好,一定程度上缓解了不平衡问题,且参数少,计算效率更高,在公开人脸属性数据集CelebA和LFWA上的实验验证了该方法的有效性。

关键词: 人脸属性, 深度学习, 多任务学习, 多标号学习

Abstract: Therecognitionofattributesplaysanimportantroleinobjectrecognition,suchasfaceverification,activityrecognitioninvideo.Theimprovementofattributerecognitionwillleadtobetterresultfortheapplicationwhichusestheseattributes.Inrecentyears,therearesomeworksfocusingontheattributelearning.However,attributeshavebeenconsideredtobeindependentinmostworks.Inpractice,weknowthatisnotrightandmanyattributesarestronglyrelated,suchasnobeardandfemale,baldandhaircolor.Inaddition,mostworksneglecttheunbalanceofdifferenceclasssamples,suchasimagesofbaldisaverysmallpartofsamples.Basedonthetwoobservations,weproposeamulti-taskunbalancedfacialattributerecognitionframeworkusingmodifiedDensenet.Toacertaindegrees,theframeworkrelievestheunbalancedproblem.Theproposedmethodoutperformsothermethods,haslessparametersandrunsfaster.Wedemonstratetheeffectivenessofourmethodontwochallengingpubliclyavailabledatasets.

Key words: faceattribute, deeplearning, multi-tasklearning, multi-labellearning

中图分类号: