简单聊聊 Test Time Augmentation

机器学习与生成对抗网络

共 2429字,需浏览 5分钟

 ·

2021-11-18 23:25

点击上方机器学习与生成对抗网络”,关注星标

获取有趣、好玩的前沿干货!

来源:知乎—IanChi
地址:https://zhuanlan.zhihu.com/p/383005472
TTA (Test Time Augmentation) 这个方法非常简单、有意思,对于提升Acc和Calibration都有不错的效果。
先从AutoML说起,大家近几年打榜提点已经不太满足于手动尝试一些超参数或者模型架构设计,而是将DL的训练建模成一个black-box optimization问题(刨个坑,回头整理下昂贵黑盒优化问题的一些方法),然后利用各种优化算法来搜索最优的超参数配置(Hyperparameter Optimization)和架构(NAS)。后来,发现数据增广暴力试错也太麻烦了,就提出了所谓的AutoAugmentation,以及后面一大堆变种。(NAS和AutoAug的工作近三年太多太多了,太TM费电了)
AutoML公司或者相关云服务提供商的一大愿景是,不管你是什么行业,只要有数据,剩下的事情都可以交给他们,然后返回一个高大上的人工智能模型(可以拿去糊弄领导,呵呵呵呵呵)。
扯完这些之后,开始简单说说TTA。所谓TTA,思想非常简单,就是在评测阶段,给每个输入进行多种数据增广变换,将一个输入变成多个输入,然后再merge起来一起输出,形成一种ensemble的效果,一方面可以提点,另一方面可以提升model calibration (ECE来评价)的效果,实现起来也是非常简单,可以直接用Pretrained model来做,相当环保。
问题来了,是否可以找出一个最优的组合策略?
文献2发表在UAI 2020,提出了一种贪心算法,如下图:
固定增广步数,每一步都从整个增广空间里选一个使得Calibrated LL提升最大的增广方式。这种方法也是非常naive的,一定不是最优的,如paper title所言,是一种简单的baseline。
文献3是文献2的一种改进,发表在NIPS 2020,提出了一种所谓Learning Loss的方法,如下图右图(左图是经典TTA):
本文的insight也非常简单,提出了一种基于loss prediction来select数据增广的方法,这里会多一步,需要训练一个DNN来做loss predictor,具体如下图:
上半部分是一个产生loss predictor label的过程,利用已有的增广方式和预训练好的分类模型构造label,下半部分是学习预测loss,使得上下两个部分的loss之间的相关性更好。
文献1更像是个大作业实验报告,没太看到什么好玩的结论,文献4-6都是TTA的在医学图像领域的应用,虽然问题都是语义分割问题,但TTA的思路本身没太多变化。
有几个简单的问题值得思考:
1、哪几类数据增广方式对于提升Acc有效?为什么?
2、哪几类数据增广方式对于提升ECE有效?为什么?
3、提升Acc和ECE的增广方式是否一致?
4、有没有更优的select policy?如果我们分析出哪些增广方式对Acc或ECE影响更大,是可以极大减少policy space的,同时可以提出结合先验知识的select原则,来提升选择效率。
参考文献:
[1] When and Why Test-Time Augmentation Works, arXiv
[2] Greedy Policy Search: A Simple Baseline for Learnable Test-Time Augmentation, UAI 2020
[3] Learning Loss for Test-Time Augmentation, NIPS 2020
[4] Test-time augmentation for deep learning-based cell segmentation on microscopy images, Nature Research
[5] Test-time augmentation with uncertainty estimation for deep learning-based medical image segmentation, MIDL 2018
[6] Test-time Data Augmentation for Estimation of Heteroscedastic Aleatoric Uncertainty in Deep Neural Networks, MIDL 2018


猜您喜欢:

等你着陆!【GAN生成对抗网络】知识星球!

CVPR 2021专题1:GAN的改进

CVPR 2021 | GAN的说话人驱动、3D人脸论文汇总

CVPR 2021 | 图像转换 今如何?几篇GAN论文

【CVPR 2021】通过GAN提升人脸识别的遗留难题

CVPR 2021生成对抗网络GAN部分论文汇总

经典GAN不得不读:StyleGAN

最新最全20篇!基于 StyleGAN 改进或应用相关论文

超100篇!CVPR 2020最全GAN论文梳理汇总!

附下载 | 《Python进阶》中文版

附下载 | 经典《Think Python》中文版

附下载 | 《Pytorch模型训练实用教程》

附下载 | 最新2020李沐《动手学深度学习》

附下载 | 《可解释的机器学习》中文版

附下载 |《TensorFlow 2.0 深度学习算法实战》

附下载 | 超100篇!CVPR 2020最全GAN论文梳理汇总!

附下载 |《计算机视觉中的数学方法》分享

浏览 34
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报