Redian新闻
>
阿里达摩院开源:半监督学习框架Dash,刷新多项SOTA!

阿里达摩院开源:半监督学习框架Dash,刷新多项SOTA!

公众号新闻

点击下方卡片,关注“CVer”公众号

AI/CV重磅干货,第一时间送达

点击进入—>CV微信技术交流群

本文介绍机器学习顶级国际会议 ICML 2021 接收的 long talk (top 3.02%) 论文 “Dash: Semi-Supervised Learning with Dynamic Thresholding”。


一、论文 & 代码


  • 论文链接:https://proceedings.mlr.press/v139/xu21e/xu21e.pdf
  • 代码地址:https://github.com/idstcv/Dash
  • 技术应用:https://modelscope.cn/models/damo/cv_manual_face-liveness_flrgb/summary


这篇论文创新性地提出用动态阈值(dynamic threshold)的方式筛选无标签样本进行半监督学习(semi-supervised learning,SSL)的方法,我们改造了半监督学习的训练框架,在训练过程中对无标签样本的选择策略进行了改进,通过动态变化的阈值来选择更有效的无标签样本进行训练Dash 是一个通用策略,可以轻松与现有的半监督学习方法集成。实验方面,我们在 CIFAR-10, CIFAR-100, STL-10 和 SVHN 等标准数据集上充分验证了其有效性。理论方面,论文从非凸优化的角度证明了 Dash 算法的收敛性质。


二、研究背景

监督学习(Supervised Learning)


我们知道模型训练的目的其实是学习一个预测函数,在数学上,这可以刻画成一个学习从数据 (X) 到标注 (y) 的映射函数。监督学习就是一种最常用的模型训练方法,其效果的提升依赖于大量的且进行了很好标注的训练数据,也就是所谓的大量带标签数据 ((X,y))。但是标注数据往往需要大量的人力物力等等,因此效果提升的同时也会带来成本过高的问题。在实际应用中经常遇到的情况是有少量标注数据和大量未标注数据,由此引出的半监督学习也越来越引起科学工作者的注意。

半监督学习(Semi-Supervised Learning)


半监督学习同时对少量标注数据和大量未标注数据进行学习,其目的是借助无标签数据来提高模型的精度。比如 self-training 就是一种很常见的半监督学习方法,其具体流程是对于标注数据 (X, y) 学习数据从 X 到 y 的映射,同时利用学习得到的模型对未标注数据 X 预测出一个伪标签,通过对伪标签数据 (X, )进一步进行监督学习来帮助模型进行更好的收敛和精度提高。

核心解决问题


现有的半监督学习框架对无标签数据的利用大致可以分为两种,一是全部参与训练,二是用一个固定的阈值卡出置信度较高的样本进行训练 (比如 FixMatch)。由于半监督学习对未标注数据的利用依赖于当前模型预测的伪标签,所以伪标签的正确与否会给模型的训练带来较大的影响,好的预测结果有助于模型的收敛和对新的模式的学习,差的预测结果则会干扰模型的训练。所以我们认为:不是所有的无标签样本都是必须的!


三、方法

Fixmatch 训练框架


在引出我们的方法 Dash 之前,我们介绍一下 Google 提出的 FixMatch 算法,一种利用固定阈值选择无标签样本的半监督学习方法。FixMatch 训练框架是之前的 SOTA 解决方案。整个学习框架的重点可以归纳为以下几点:

1、对于无标签数据经过弱数据增强(水平翻转、偏移等)得到的样本通过当前的模型得到预测值 
2、对于无标签数据经过强数据增强(RA or CTA)得到的样本通过当前的模型得到预测值 
3、把具有高置信度的弱数据增强的结果,通过 one hot 的方式形成伪标签,然后用 和 X 经过强数据增强得到的预测值 进行模型的训练。

fixmatch 的优点是用弱增强数据进行伪标签的预测,增加了伪标签预测的准确性,并在训练过程中用固定的阈值 0.95(对应 loss 为 0.0513) 选取高置信度(阈值大于等于 0.95,也就是 loss 小于等于 0.0513)的预测样本生成伪标签,进一步稳定了训练过程。

Dash 训练框架

针对全部选择伪标签和用固定阈值选择伪标签的问题,我们创新性地提出用动态阈值来进行样本筛选的策略。即动态阈值 是随 t 衰减的


式中 C=1.0001,是有标签数据在第一个 epoch 之后 loss 的平均值,我们选择那些的无标签样本参与梯度回传。下图展示了不同值下的阈值 的变化曲线。可以看到参数 控制了阈值曲线的下降速率。 的变化曲线类似于模拟训练模型时损失函数下降的趋势。


下图对比了训练过程中的 FixMath 和 Dash 选择的正确样本数和错误样本数随训练进行的变化情况(使用的数据集是 cifar100)。从图中可以很清楚地看到,对比 FixMatch,Dash 可以选取更多正确 label 的样本,同时选择更少的错误 label 的样本,从而最终有助于提高训练模型的精度。


我们的算法可以总结为如下 Algorithm 1。Dash 是一个通用策略,可以轻松与现有的半监督学习方法集成。为了方便,在本文的实验中我们主要将 Dash 与 FixMatch 集成。更多理论证明详见论文。


四、结果

我们在半监督学习常用数据集:CIFAR-10,CIFAR-100,STL-10 和 SVHN 上进行了算法的验证。结果分别如下:



可以看到我们的方法在多个实验设置上都取得了比 SOTA 更好的结果,其中需要说明的是针对 CIFAR-100 400label 的实验,ReMixMatch 用了 data align 的额外 trick 取得了更好的结果,在 Dash 中加入 data align 的 trick 之后可以取得 43.31% 的错误率,低于 ReMixMatch 44.28% 的错误率。

五、应用

实际面向任务域的模型研发过程中,该半监督 Dash 框架经常会被应用到。接下来给大家介绍下我们研发的各个域上的开源免费模型,欢迎大家体验、下载(大部分手机端即可体验):

  • https://modelscope.cn/models/damo/cv_resnet50_face-detection_retinaface/summary
  • https://modelscope.cn/models/damo/cv_resnet101_face-detection_cvpr22papermogface/summary
  • https://modelscope.cn/models/damo/cv_manual_face-detection_tinymog/summary
  • https://modelscope.cn/models/damo/cv_manual_face-detection_ulfd/summary
  • https://modelscope.cn/models/damo/cv_manual_face-detection_mtcnn/summary
  • https://modelscope.cn/models/damo/cv_resnet_face-recognition_facemask/summary
  • https://modelscope.cn/models/damo/cv_ir50_face-recognition_arcface/summary
  • https://modelscope.cn/models/damo/cv_manual_face-liveness_flir/summary
  • https://modelscope.cn/models/damo/cv_manual_face-liveness_flrgb/summary
  • https://modelscope.cn/models/damo/cv_manual_facial-landmark-confidence_flcm/summary
  • https://modelscope.cn/models/damo/cv_vgg19_facial-expression-recognition_fer/summary
  • https://modelscope.cn/models/damo/cv_resnet34_face-attribute-recognition_fairface/summary

点击进入—>CV微信技术交流


CVPR/ECCV 2022论文和代码下载


后台回复:CVPR2022,即可下载CVPR 2022论文和代码开源的论文合集

后台回复:ECCV2022,即可下载ECCV 2022论文和代码开源的论文合集

后台回复:Transformer综述,即可下载最新的3篇Transformer综述PDF


目标检测和Transformer交流群成立


扫描下方二维码,或者添加微信:CVer222,即可添加CVer小助手微信,便可申请加入CVer-目标检测或者Transformer 微信交流群。另外其他垂直方向已涵盖:目标检测、图像分割、目标跟踪、人脸检测&识别、OCR、姿态估计、超分辨率、SLAM、医疗影像、Re-ID、GAN、NAS、深度估计、自动驾驶、强化学习、车道线检测、模型剪枝&压缩、去噪、去雾、去雨、风格迁移、遥感图像、行为识别、视频理解、图像融合、图像检索、论文投稿&交流、PyTorch、TensorFlow和Transformer等。


一定要备注:研究方向+地点+学校/公司+昵称(如目标检测或者Transformer+上海+上交+卡卡),根据格式备注,可更快被通过且邀请进群


▲扫码或加微信号: CVer222,进交流群


CVer学术交流群(知识星球)来了!想要了解最新最快最好的CV/DL/ML论文速递、优质开源项目、学习教程和实战训练等资料,欢迎扫描下方二维码,加入CVer学术交流群,已汇集数千人!


扫码进群

▲点击上方卡片,关注CVer公众号

整理不易,请点赞和在看

微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
阿里版ChatGPT已进入测试!中文聊天截图曝光,达摩院出品杭州内推 | 阿里达摩院城市大脑实验室招聘三维计算方向研究型实习生面对道德绑架AAAI 2023 Oral | 图像质量堪忧干扰视觉识别!达摩院提出RTS:鲁棒性特征建模框架语言模型自己学会用搜索引擎了?Meta AI提出API调用自监督学习方法ToolformerECCV 2022 | 半监督学习的隐秘角落: 无监督的数据选择性标注 USL10行代码搞定图Transformer,图神经网络框架DGL迎来1.0版本北京/杭州内推 | 阿里达摩院开放视觉智能实验室招聘研究型实习生/23届春招预言未来!阿里巴巴达摩院发布2023十大科技趋势前美团联合创始人王慧文 “正在收购” 国产开源深度学习框架OneFlowAAAI 2023|基于多模态标签聚合的视频检索模型TABLE,多项SOTA博后招募 | 阿里达摩院系统Al实验室招募机器学习方向博士后揭秘阿里达摩院类ChatGPT核心团队人才地图|36氪独家NeurIPS 2022 | 基于结构聚类的异质图自监督学习北京市数字人基地与阿里巴巴达摩院人工智能开源社区“魔搭”签署战略合作协议校招 | 阿里巴巴达摩院-语言智能实验室-应用算法-智能司法组阿里达摩院春招 & Research Intern 招聘阿里达摩院2023十大科技趋势重磅发布:Chiplet、生成式AI榜上有名AAAI 2023 | HiCo: 一种简单有效的3D骨架动作表示学习框架AAAI 2023 | 图像质量堪忧干扰视觉识别,达摩院提出更鲁棒框架军旅故事原创系列(11)看电影的灰色地带阿里达摩院决策智能实验室招聘全职/博士后/实习生杭州/北京内推 | 阿里达摩院生物医学信息处理团队招聘24届春招实习生突发!阿里云高层大调整,集团CEO张勇兼任阿里云总裁,张建锋专职达摩院立足工农大众的新秧歌运动掀起ICLR 2023 | MocoSFL: 低成本跨用户联邦自监督学习造娃养娃,才知道有这特殊医疗保险-美洋插队生活实录(十六)西电 NeurIPS 2022 | 基于结构聚类的异质图自监督学习AAAI 2023 Oral | 图像质量堪忧干扰视觉识别,达摩院提出更鲁棒框架用热传导方程来指导自监督学习达摩院开源半监督学习框架Dash,刷新多项SOTA巴西队回家了!大牛动态:阿里达摩院认知智能大牛跳槽字节:加入AI Lab汇报副总裁杨震原阿里达摩院自动驾驶负责人陈俊波离职,进军户外清洁机器人!MLNLP学术Talk第十九期 | 余海洋、黎槟华@阿里达摩院:基于知识的下一代对话:新任务、新数据、新榜单
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。