Skip to content

Latest commit

 

History

History

dselect_k

DSelect-k

AI Studio在线运行环境

以下是本例的简要目录结构及说明:

├── data # 数据
    ├── train # 训练数据
        ├── train.pickle
    ├── test # 测试数据
        ├── test.pickle
├── __init__.py
├── README.md # 文档
├── config.yaml # cpu 配置
├── config_bigdata.yaml # gpu 配置
├── net.py # 模型核心组网(动静统一)
├── multiMNIST_reader.py # 数据读取程序
├── static_model.py # 构建静态图
├── dygraph_model.py # 构建动态图

注:在阅读该示例前,建议您先了解以下内容:

paddlerec入门教程

内容

模型简介

MoE(Mixture of Experts) 架构在改善多任务学习 MTL(Multi-Task Learning) 中的参数共享和扩展大容量神经网络方面显示出良好的效果。SOTA 的 MoE 类模型使用一个可训练的稀疏门控来为每个输入实例选择一个专家子集。虽然概念上可行有效,但现有的稀疏门控例如 Top-K, 并不平滑(意味着不可导)。在使用基于梯度的方法进行训练时,平滑度的缺失会导致收敛和统计性能问题。 本文基于二进制编码方法提出了 DSelect-k: a continuously differentiable and sparse gate for MoE, 解决了现有稀疏门控不可导的弊端,可以根据梯度下降类方法进行训练。

上图是 MoE 和 MMoE 的结构图, 本文所提出的 DSelect-k 模型用于从 N 个专家中选择 Top-K 个进行后续任务预测。其主要有两种模式,1)Static; 2) Per-example; 前者不感知输入,所有实例会选择同样的专家子集,后者恰恰相反。

@article{hazimeh2021dselectk,
      title={DSelect-k: Differentiable Selection in the Mixture of Experts with Applications to Multi-Task Learning},
      author={Hussein Hazimeh and Zhe Zhao and Aakanksha Chowdhery and Maheswaran Sathiamoorthy and Yihua Chen and Rahul Mazumder and Lichan Hong and Ed H. Chi},
      year={2021},
      eprint={2106.03760},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

数据准备

训练及测试数据集选用的是 Multi-MNIST 数据集,该数据集是在 Dynamic Routing Between Capsules 首次介绍提出,后续一些 MTL 论文大多沿用该数据集。

The MultiMNIST dataset is generated from MNIST. The training and tests are generated by overlaying a digit on top of another digit from the same set (training or test) but different class. Each digit is shifted up to 4 pixels in each direction resulting in a 36×36 image.

上图是其中一张图片,与经典数据集 MNIST 不同,该图片上包含两位数字,左上和右下,对应两个多分类任务。数据集划分为训练集、验证集、测试集,数量分别为 100000、20000、20000。

在 PaperswithCode 网站上检索到 【NeurlPS 2019】Pareto Multi-Task Learning 公布了该数据集。

运行环境

PaddlePaddle>=2.1

python 2.7/3.5/3.6/3.7

os : windows/linux/macos

快速开始

本文提供了样例数据可以供您快速体验,在任意目录下均可执行。在 dselect_k 模型目录的快速执行命令如下:

# 进入模型目录
# cd models/multitask/dselect_k # 在任意目录均可运行
# 动态图训练
python -u ../../../tools/trainer.py -m config.yaml # 全量数据运行config_bigdata.yaml 
# 动态图预测
python -u ../../../tools/infer.py -m config.yaml 

# 静态图训练
python -u ../../../tools/static_trainer.py -m config.yaml # 全量数据运行config_bigdata.yaml 
# 静态图预测
python -u ../../../tools/static_infer.py -m config.yaml 

效果复现

为了方便使用者能够快速的跑通每一个模型,我们在每个模型下都提供了样例数据。如果需要复现 readme 中的效果,请按如下步骤依次操作即可。 在全量数据下模型的指标如下:

模型 Accuracy1 Accuracy2 batch_size epoch_num Time of each epoch
DSelect-k 0.930460 0.916088 256 100 约 0.5 小时
  1. 确认您当前所在目录为 PaddleRec/models/multitask/dselect_k

  2. 进入 paddlerec/datasets/Multi_MNIST_DselectK 目录下,执行该脚本,会从国内源的服务器上下载数据集,并解压到指定文件夹。

cd ../../../datasets/Multi_MNIST_DselectK
sh run.sh
  1. 切回模型目录,执行命令运行全量数据
# 切回模型目录 PaddleRec/models/multitask/dselect_k
# 动态图训练
python -u ../../../tools/trainer.py -m config_bigdata.yaml # 全量数据运行config_bigdata.yaml 
python -u ../../../tools/infer.py -m config_bigdata.yaml # 全量数据运行config_bigdata.yaml 

进阶使用

FAQ