问答

样本标签极不均衡,这种训练方式可行吗?

作者:admin 2021-04-10 我要评论

用多任务学习进行分类,比如任务A是分类3种风格,如阳光、低调、"其他风格",任务B是分类3种款式,如牛仔裤、西裤、"其他款式"。 任一样本被标记了风格和款式两...

在说正事之前,我要推荐一个福利:你还在原价购买阿里云、腾讯云、华为云服务器吗?那太亏啦!来这里,新购、升级、续费都打折,能够为您省60%的钱呢!2核4G企业级云服务器低至69元/年,点击进去看看吧>>>)

用多任务学习进行分类,比如任务A是分类3种风格,如阳光、低调、"其他风格",任务B是分类3种款式,如牛仔裤、西裤、"其他款式"。
任一样本被标记了风格和款式两个标签,但是95%的样本标签都是 任一风格+其他裤子 或者 任一裤子+其他风格 这样的。讲得可能不是很清楚,下图帮助理解:

image

这样数据集中阳光和低调、牛仔裤和西裤之间数量差不多,但是"其他款式"和"其他风格"占比非常大。

由于这是多任务学习网络,可不可一次只训练一个分支,即把另一个分支每个神经元的权重调成0,或者把另一个分支冻结,来进行训练?

或者还有没有其他更好的办法?

补充:

  1. 上面提到的数据集是个假想的例子,实际数据集中的标签要多得多。
  2. 我有考虑过数据增强和调样本权重。
    一开始我用了数据集中较少标签(只有款式和面料),训练出来的模型的F1值大小排序约为:数据增强>样本调权>数据增强+样本调权。
    然后我打算使用更多标签(款式、面料、贴图、风格),并只用了数据增强,最后模型把他们全部都预测为了其他分类。

版权声明:本文转载自网络,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本站转载出于传播更多优秀技术知识之目的,如有侵权请联系QQ/微信:153890879删除

相关文章
  • nginx响应速度很慢

    nginx响应速度很慢

  • 点击选中的多选框,会在已选那一栏显示

    点击选中的多选框,会在已选那一栏显示

  • PHP 多态的理解

    PHP 多态的理解

  • 关于C语言中static的问题

    关于C语言中static的问题

腾讯云代理商
海外云服务器