1 Star 0 Fork 0

Lindsay.Lu丶 / vega

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
README.cn.md 6.97 KB
一键复制 编辑 原始数据 按行查看 历史
张家劲 提交于 2020-06-29 21:05 . release 0.9.3

Vega

English

Vega简介

Vega是诺亚方舟实验室自研的AutoML算法工具链,有主要特点:

  1. 完备的AutoML能力:涵盖HPO(超参优化, HyperParameter Optimization)、Data-Augmentation、NAS(网络架构搜索, Network Architecture Search)、Model Compression、Fully Train等关键功能,同时这些功能自身都是高度解耦的,可以根据需要进行配置,构造完整的pipeline。
  2. 业界标杆的自研算法:提供了诺亚方舟实验室自研的业界标杆算法,并提供Model Zoo下载SOTA(State-of-the-art)模型。
  3. 高并发模型训练能力:提供高性能Trainer,加速模型训练和评估。

算法列表

分类 算法 说明 参考
NAS SM-NAS: Structural-to-Modular NAS 两阶段物体检测架构搜索算法 开发中
NAS CARS: Continuous Evolution for Efficient Neural Architecture Search 基于连续进化的多目标高效神经网络结构搜索方法 参考
NAS SR-EA 适用于轻量级网络的自动网络架构搜索方法 参考
NAS ESR-EA: Efficient Residual Dense Block Search for Image Super-resolution 基于网络架构搜索的多目标图像超分方法 参考
NAS Adelaide-EA: SEGMENTATION-Adelaide-EA-NAS 图像分割网络架构搜索算法 参考
NAS SP-NAS: Serial-to-Parallel Backbone Search for Object Detection 面向物体检测及语义分割的高效主干网络架构搜索算法 参考
NAS Auto-Lane: CurveLane-NAS 一种端到端的车道线架构搜索算法 开发中
Model Compression Quant-EA: Quantization based on Evolutionary Algorithm 自动混合比特量化算法,使用进化策略对CNN网络结构每层量化位宽进行搜索 参考
Model Compression Prune-EA 使用进化策略对CNN网络结构进行自动剪枝压缩算法 参考
HPO ASHA: Asynchronous Successive Halving Algorithm 动态连续减半算法 参考
HPO TPE: Tree-structured Parzen Estimator Approach 一种基于树状结构Parzen估计方法的超参优化算法 参考
HPO BO: Bayesian Optimization 贝叶斯优化算法 参考
HPO BOHB: Hyperband with Bayesian Optimization 动态连续减半算法 参考
HPO BOSS: Bayesian Optimization via Sub-Sampling 基于贝叶斯优化框架下的一种针对计算资源受限,需要高效搜索的,具有普适性的超参优化算法 参考
Data Augmentation PBA: Population Based Augmentation: Efficient Learning of Augmentation Policy Schedules 基于PBT优化算法搜索数据增广策略时刻表的数据增广算法 参考
Data Augmentation cyclesr: CycleGAN + SR 底层视觉的无监督风格迁移算法 参考
Fully Train FMD 基于特征图扰动的神经网络训练方法 参考

获取和安装

请获取最新版本,参考安装指导完成安装,若希望在集群中部署Vega,请参考部署指导

使用指导

Vega高度模块化,可通过配置可完成搜索空间、搜索算法、pipeline的构建,运行Vega应用就是加载配置文件,并根据配置来完成AutoML流程。 同时Vega提供了详细的操作示例供大家参考,可参考示例参考,如运行CARS算法示例:

cd examples
python3 ./run_example.py ./nas/cars/cars.yml

在使用Vega前,有必要充分了解配置项的含义,请参考配置指导

注意:

在运行示例前,需要在算法配置文件中配置数据集和预训练模型目录,缺省目录请参考示例参考

开发者指导

Vega框架组件解耦,并采用注册机制来组合各个功能组件,便于扩充功能和算法,可参考开发者指导,了解Vega架构和主要机制。

同时可参考快速入门指导,通过实现一个简单的CNN网络搜索功能,通过实战快速进入Vega应用开发。

在Vega应用的开发中,最先遇到的问题就是,如何引入业务数据集到Vega中,可参考数据集指导

针对不同的算法,可参考算法开发指导,可根据文中提供的示例,一步步的将新算法添加到Vega中。

在Automl的大多数算法中搜索空间和网络是强相关强耦合的,我们尝试统一搜索空间的定义方式,使得同一种搜索空间能够适配不同的搜索算法,我们称之为细粒度搜索空间指导,欢迎大家尝试使用。

当然文档解决不了所有的疑问和问题,若你在使用中遇到任何问题,请及时通过issue反馈,我们会及时答复和解决。

参考列表

对象 参考
用户 安装指导部署指导配置指导示例参考、任务参考(分类检测分割超分)
开发者 API参考开发者指导快速入门指导数据集指导算法开发指导细粒度搜索空间指导

合作和贡献

欢迎大家使用Vega,有任何疑问、求助、修改bug、贡献算法、完善文档,请在社区提交issue,我们会及时回复沟通交流。 欢迎大家加入我们的QQ群: 833345709。

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/lindsaylu/vega.git
git@gitee.com:lindsaylu/vega.git
lindsaylu
vega
vega
master

搜索帮助

344bd9b3 5694891 D2dac590 5694891