HyperAIHyperAI

Command Palette

Search for a command to run...

基于种群的增强:高效学习增强策略调度

Daniel Ho; Eric Liang; Ion Stoica; Pieter Abbeel; Xi Chen

摘要

在利用数据增强进行神经网络训练时,一个关键挑战是从大量候选操作中选择有效的增强策略。合理选择的增强策略可以显著提高模型的泛化能力;然而,诸如AutoAugment等最先进的方法对于普通用户来说计算成本过高,难以实现。本文介绍了一种新的数据增强算法——基于群体的数据增强(Population Based Augmentation, PBA),该算法生成的是非静态的增强策略调度,而不是固定的增强策略。我们证明了PBA能够在CIFAR-10、CIFAR-100和SVHN数据集上达到与AutoAugment相当的性能,而整体计算量却减少了三个数量级。在CIFAR-10数据集上,我们实现了1.46%的平均测试错误率,这略优于当前的最先进水平。PBA的代码已开源,并可在https://github.com/arcelien/pba 获取。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供