Skip to content

liwei-cpp/MetaNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

English Version

TODO 列表

  • 引入Reshape operator,改变数据的类别与结构
  • 考虑 training 相关的模块
  • 考虑CNN的相关支持(相对较大的工作)

MetaNN

MetaNN 是一个深度学习框架的骨架,旨在探索 C++ 模板元编程在深度学习中的应用。它不支持多机并发训练。目前来说,它只支持 CPU ,包含了一些简单的构造。但 MetaNN 具有足够的扩展性,能够相对容易地支持诸如 GPU 或 FPGA 这样的设备。就目前来说,它所包含的计算逻辑只是示例性的,其速度并不快,但其框架非常灵活,基于这套框架进行的程序优化可以在保证易用性的同时,极大地提升深度学习系统的性能。

本文会简述 MetaNN 的主体设计思想,核心组件以及优化的可能性。

代码组织与运行环境

MetaNN 是一个 C++ 模板库,与其它模板库类似,它的核心逻辑都包含在头文件中(以 .h 结尾)。除了这些头文件外,MetaNN 还包含了若干测试工程:Tests/DataOpTest,Tests/LayerTest 与 Tests/OtherTests ,它们用于测试 MetaNN 的内部逻辑,同时也是简单的示例,展示了 MetaNN 中的数据与操作,层,以及其它部分的使用方式。它们会调用 MetaNN 中包含的逻辑实现具体的计算。 读者可以阅读这其中的代码从使用者的角度了解 MetaNN 的工作原理。

运行环境

MetaNN 库以及测试项目均使用 CodeLite 进行开发。 测试程序可以在如下的环境上编译并运行:

  • g++ 7.4.0
  • g++ 8.3.0
  • g++ 9.1.0
  • clang++ 8.0.0
  • clang++ 9.0.0 不建议使用较低版本的编译器编译本工程。MetaNN 使用了 C++ 17 中的特性。我并不能保证程序的每一行代码都完全符合 C++ 17 这一标准;进一步,标准中未明确定义的部分可能导致不同编译器有不同的行为;编译器中的 Bug也可能阻止程序的顺利编译。读者可以选择自己的编译器进行试验。这里推荐的运行环境是 Ubuntu & g++。

需要说明的一点是:由于要测试每种编译器的行为是否正确是一个比较麻烦的工作,因此我并不会在每次修改后都进行测试。而是会在积累若干修改后统一的进行测试。对于未来得及测试的情形,我会将没有测试的编译器标记一个问号。

MetaNN 是自成体系的。除了对 C++17 的依赖之外,作者有意没有依赖任何第三方库。这样不仅方便运行环境的搭建,也可以使程序的阅读者不会因对某个库的不熟悉而影响对程序的理解(当然,要想理解这个库,需要对 C++17 以及元编程有一定的了解)。

代码组织

整个框架核心逻辑被包含在 MetaNN 目录中。与 MetaNN 目录同级的是 Tests 目录,其中包含了相应的测试程序。这里仅以 MetaNN 目录为例说明代码的组织形式。

  • _root: 只包含了一个文件 meta_nn.h ,引用这个文件即可使用 MetaNN 所提供的全部功能
  • data: 存放数据的目录,其中包含了深度学习框架可能会使用到的各种数据类型
  • data_copy: 保存了用于复制数据的逻辑。目前该目录中只包含了一个函数,用于将一个 CPU 张量赋予另一个 CPU 张量,后续会根据需要引入更多的数据复制方法。
  • evaluate: 保存了 MetaNN 中的求值逻辑
  • facilities: 包含了一些辅助(元)函数与结构
  • layers: “层”的实现逻辑,层是整个深度学习框架的核心
  • model: 目标是存放一些模型相关的逻辑,如参数初始化,梯度收集等。
  • operation: 包含了操作的实现逻辑。
  • policies: 包含了 Policy 的相关实现。 Policy 可以被视为一种编译期的分支机制,在 MetaNN 中主要用于层的策略定制。我们会在后文讨论层的设计时说明 Policy 所起的作用

接下来,文本将从底向上依次讨论框架中的若干组件。阐述其设计理念并给出一些使用示例。

数据类型

设计理念

  • 基本数据表示: MetaNN 之前的版本并不支持 Tensor ,而是使用诸如矩阵、标量这样的类模板 来表示数据。这样做的目的是为了引入概念相对清晰的数据结构。但随着开发的进行,作者发现使用原有的体系结构对扩展性造成了很大的限制。因此新的版本中屏弃了原有的设计,采用了流行深度学习框架中 Tensor 的表示形式。

  • 支持数据扩展与分类: MetaNN 的数据类型支持若干维度的扩展。除了前文所述的可以引入新的数据类型外,还可以对已有数据类型(实际上是数据模板)所包含的元素类型(如float或者double或者其它的定点类型)进行扩展;对数据类型所支持的计算设备(如CPUGPU)进行扩展。 MetaNN 中的数据类型按照其类别进行划分,我们使用 CategoryTags::Tensor<Dim> 来表示数据的类别。比如 CategoryTags::Tensor<2> 表示矩阵;CategoryTags::Tensor<0> 表示标量。

基本用法

MetaNN 中表示数据的最基本类型是 Tensor。比如,可以使用如下的方式声明一个 Tensor:

Tensor<int, DeviceTags::CPU, 2> matrix(10, 20);

这相当于定义了一个矩阵,包含 10 行,每行 20 列,该对象中的数据元素类型为int,所支持的计算设备为CPU

系统同时为 Tensor引入了若干别名,比如:

template <typename TElem, typename TDevice>
using Matrix = Tensor<TElem, TDevice, 2>;

因此,以下的定义与上面的定义等价:

Matrix<int, DeviceTags::CPU> matrix(10, 20);

MetaNN 还可以定义其它的张量对象:

// 向量,包含100个元素,第37个元素为0.3,其余为0
BiasVector(100, 37, Scalar<int, DeviceTags::CPU>{0.3});

// 一个 10*20 的矩阵,其中元素值均为100
TrivalTensor(Scalar<int, DeviceTags::CPU>{100}, 10, 20);

// 一个 10*20 的矩阵,其中元素值均为0
ZeroTensor<int, DeviceTags::CPU, 2> m3(10, 20)

还可以声明维度更高的张量:

// 张量对象,维度为 5*10*20,其中的元素均为 0
ZeroTensor<CheckElement, CheckDevice, 3>(5, 10, 20);

类别划分

上述声明的类型不同,可以应用于不同的场景。每种类型都采用了不同的方式存储其内部数据,并提供不同的访问接口。MetaNN 将这些类型划分成不同的类别,可以用如下的方式获取特定数据类型所对应的类别:

IsMatrix<Matrix<int, DeviceTags::CPU>>;                    // true
IsMatrix<ZeroTensor<int, DeviceTags::CPU, 3>>;        // false
IsThreeDArray<ZeroTensor<int, DeviceTags::CPU, 3>>;	  // true
IsMatrix<ZeroTensor<int, DeviceTags::CPU, 3>>;	      // false

或者使用如下的方式:

// CategoryTags::Tensor<2> (也即 CategoryTags::Matrix)
DataCategory<Matrix<int, DeviceTags::CPU>>;

// CategoryTags::Tensor<3>
DataCategory<ZeroTensor<int, DeviceTags::CPU, 3>>;

除了自身特有的接口外,具有相同类别的数据类型提供一组通用的接口。比如,所有的数据类型都提供了Shape()接口来返回其形状。但Shape()的返回对象的类型则根据数据类型的不同而不同:矩阵类别的对象需要提供接口返回行数与列数,而三维张量对应的Shape()则会返回三个维度的数值。此外,每个数据类型都要提供求值接口以转换为相应主体类型的对象(我们会在后面讨论求值的过程)。

引入新的类型

我们可以很简单地为 MetaNN 引入新的类型:

template <typename TElem, typename TDevice>
class MyMatrix
{
public:
    using CategoryTag = CategoryTags::Matrix;
    using ElementType = TElem;
    using DeviceType = TDevice;
}

MyMatrix<float, DeviceTags::CPU> ...

注意上述定义的CategoryTag会将MyMatrix的类别设定为矩阵。这样,以下的调用将返回true

IsMatrix<MyMatrix<int, DeviceTags::CPU>>;

自定义的类型与 MetaNN 已经定义的类型地位相同。

动态类型

除了上述类型外, MetaNN 还引入了一个特殊的类型模板: DynamicData<>。它用于对不同的数据类别进行封装,掩盖其底层的具体类型信息,只暴露出该数据类型最核心的特性与接口。比如:

DynamicData<float, DeviceTags::CPU, CategoryTags::Matrix>

它可以被视为一个容器,其中可以放置元素类型为float,设备类型为CPU的矩阵:

vector<DynamicData<float, DeviceTags::CPU, CategoryTags::Matrix>> vec;

Matrix<float, DeviceTags::CPU> m1(100, 37);
vec.push_back(MakeDynamic(m1));

ZeroData<CategoryTags::Matrix, float, DeviceTags::CPU> m2(10, 20)
vec.push_back(MakeDynamic(m2));

MakeDynamic是 MetaNN 提供的一个函数,给定一个具体的数据类型,可以转换为相应的Dynamic版本。

同样的,我们也可以将三维张量置于DynamicData<... CategoryTags::Tensor<3>>的容器中。

DynamicData 用于保存中间变量,它掩盖了其底层所包含的具体的数据类型。便于存储类型不同(类型未知)但类别相同(类别已知)的数据。但 DynamicData 的引入导致了具体信息类型的丢失,对编译期计算会产生一些不利的影响。因此,DynamicData 只会在必要时才被使用。

判等

MetaNN 的每个数据类型都提供了 operator ==operator !=来与其它数据进行比较。但这里的比较仅供后续求值优化所使用。我们并不希望提供一个元素级的比较——即比较张量中的每个元素:一方面,浮点类型的数据无法精确比较;另一方面,这样的比较很耗时。MetaNN 中所提供的是一种快速的比较方案:

  • 如果 A 复制自 B,那么 A 与 B 相等
  • 如果 A 复制自 B,同时 C 与 D 是 A 与 B 求值的结果,那么 C 与 D 相等

除此之外, 数据通常是不相等的(除非底层数据的判等操作成本很低,MetaNN才会考虑使用底层数据来判断是否相等)。

操作

MetaNN 包含了多个操作。每个操作都可以接收一到多个数据对象,返回操作结果。

设计理念

MetaNN 使用表达式模板作为操作的结果。表达式模板对象的内部包含了表示该操作的操作数对象,它并不进行实际的计算。因此基于操作来构造操作结果模板的复杂度非常低。表达式模板实际的求值过程被推迟到显式调用求值时进行——这就为求值过程的优化提供了空间。

一方面,表达式模板类型可以被视为操作结果;另一方面,也可以将其看成一种特殊数据。与 MetaNN 中基本的数据类型相似,操作产生的表达式模板对象也会被划分为矩阵、标量等类别,并提供相应分类所需要支持的接口。这样,我们就可以将某个操作的结果用做另一个操作的操作数了。

MetaNN 是富类型的,并非任意类型都可以做为某个操作的操作数。 MetaNN 设计了相关的机制来防止操作数的误用(当然,我们可以通过扩展的方式使得某种操作支持新的操作数或操作数的组合)。

基本用法

可以通过如下的方式来调用 MetaNN 中的操作并生成相应的表达式模板:

auto rm1 = Matrix<int, DeviceTags::CPU>(4, 5);
auto rm2 = Matrix<int, DeviceTags::CPU>(4, 5);
auto add = rm1 + rm2;

add 就是由两个操作数所形成的表达式模板。其实际类型为:

Operation<OpTags::Add, 
          OperandContainer<Matrix<int, DeviceTags::CPU>, Matrix<int, DeviceTags::CPU>>
          PolicyContainer<>
         >

通常来说,我们在实际的程序设计过程中不需要关心 add 的实际类型(只需要像上例中那样使用 auto 让编译器自动推导,或者使用前文所提供的动态类型对其进行封装即可)。但在这里,还是有必要了解一下 add 对象对应的类型所包含的信息的。

add 对象的实际类型是 Operation 模板实例化的结果。这表明这个类型是一个“操作”。操作的具体内容则由模板参数 OpTags::Add 给出。Operation 的第二个模板参数表示了操作数的类型。Operation 的第三个模板参数则用于保存一些编译期的参数。

模板表达式本身也可以视为一种数据,因此,我们可以这样写:

auto rm1 = Matrix<int, DeviceTags::CPU>(4, 5);
auto rm2 = Matrix<int, DeviceTags::CPU>(4, 5);
auto add = rm1 + rm2;
auto rm3 = ZeroTensor<CategoryTags::Matrix, DeviceTags::CPU, 2>(4, 5);
auto sub = add - rm3;

此时, sub 所对应的类型为:

Operation<OpTags::Substract,
          OperandContainer<
                           Operation<OpTags::Add,
                                     OperandContainer<
                                                      Matrix<int, DeviceTags::CPU>,
                                                      Matrix<int, DeviceTags::CPU>>,
                                     PolicyContainer<>>
                           ZeroData<CategoryTags::Matrix, DeviceTags::CPU>,
          PolicyContainer<>>

本质上,模板表达式形成了一种树的结构。而模板表达式的求值过程,也可以被视为一种树的遍历过程。这个过程可以被优化。我们将在后续讨论求值时讨论相关的优化方法。

模板表达式的分类

模板表达式也可以被视为数据, MetaNN 自动实现了模板表达式的分类。比如:

IsMatrix<Operation<OpTags::Substract,
          OperandContainer<
                           Operation<OpTags::Add,
                                     OperandContainer<
                                                      Matrix<int, DeviceTags::CPU>,
                                                      Matrix<int, DeviceTags::CPU>>,
                                     PolicyContainer<>>
                           ZeroData<CategoryTags::Matrix, DeviceTags::CPU>,
          PolicyContainer<>>>;    // true

这是因为两个矩阵相加的结果还是矩阵,而矩阵与矩阵相减的结果还是矩阵。

各式各样的操作

需要说明的是,并非所有的操作都像加减法这样“一目了然”——其操作数与操作结果的类别相一致。比如, MetaNN 中定义了 ReduceSum 操作,可以将张量中某些维度的数据求和,得到新的张量。此时,操作的输入输出类型就不相同了。 MetaNN 的操作各式各样,也可以相对容易地扩展,可以在扩展的同时指定操作数与操作结果的类别,以及一些其它的信息。

操作数的支持

MetaNN 是富类型的,一个使用 MetaNN 的程序可能会在操作中使用多种类型。对于特定的操作,并非所有的操作数类型都是合法的。 MetaNN 引入了特定的元函数来描述某个操作可以接收的合法的操作数。同样以加法为例,相应的元函数为:

template <typename TOp1, typename TOp2>
constexpr bool IsValidOper<OpTags::Add, TOp1, TOp2>
    = IsValidCategoryTag<DataCategory<TOp1>> && IsValidCategoryTag<DataCategory<TOp2>>;
    
namespace OperAddWithNum
{
template <typename TOp1, typename TOp2>
constexpr bool Valid()
{
    if constexpr (IsValidCategoryTag<DataCategory<TOp1>> && !IsValidCategoryTag<DataCategory<TOp2>>)
    {
        return std::is_constructible_v<typename RemConstRef<TOp1>::ElementType, TOp2>;
    }
    else if constexpr (!IsValidCategoryTag<DataCategory<TOp1>> && IsValidCategoryTag<DataCategory<TOp2>>)
    {
        return std::is_constructible_v<typename RemConstRef<TOp2>::ElementType, TOp1>;
    }
    else
    {
        return false;
    }
}

template <typename TOp1, typename TOp2>
constexpr bool IsValidOper<OpTags::AddWithNum, TOp1, TOp2> = OperAddWithNum::Valid<TOp1, TOp2>();

template <typename TP1, typename TP2,
          std::enable_if_t<IsValidOper<OpTags::Add, TP1, TP2> ||
                           IsValidOper<OpTags::AddWithNum, TP1, TP2>>* = nullptr>
auto operator+ (TP1&& p_m1, TP2&& p_m2)

只有 IsValidOper<OpTags::Add, TP1, TP2>IsValidOper<OpTags::AddWithNum, TP1, TP2> 为真时,类型为 TP1TP2 的两个操作数才能相加。事实上,这就限制了:只有两个张量,或者一个张量与一个数值才能相加。

float a, b;
// 这不会触发 MetaNN 中的操作逻辑
float c = a + b;

Matrix<int, DeviceTags::CPU> rm1(2, 3);
Tensor<int, DeviceTags::CPU, 3> rm2(5, 2, 3);
// 触发 MetaNN 中的操作逻辑,相加结果为一个 5*2*3 的张量。
auto rm3 = rm1 + rm2;

当然,操作所支持的操作数组合也是可以扩展的,可以通过特化 IsValidOper 引入相应的扩展,但要在同时为相应的操作给出明确的定义。

基本层

层是 MetaNN 中相对高级的组件。当前很多深度学习框架都弱化了层的概念,转而将操作与层的概念合并起来。 MetaNN 认为区分层与操作,可以更好地描述不同层面上的抽象。本节讨论基本层。

设计理念

层是 MetaNN 对外的接口。基本层封装了操作,调用具体的操作实现数据的正向、反向传播。每个基本层都能实现特定的功能,比如将两个矩阵相加,或者求矩阵的点乘。虽然基本层的功能相对单一,但其行为细节是可以配制的。比如,我们可以指定某个层是否要在训练的过程中更新其中所包含的参数。层应该了解其具体的行为细节,并根据其进行相应的优化。

通常情况下,层的具体行为极少在程序的执行过程中发生改变。因此,可以在编译期指定相应的信息,以确保尽早地利用这些信息进行最大限度的优化。

每个层都应当支持正向、反向传播。除此之外,层可能需要根据其自身特性提供若干额外的接口函数。比如,如果层中包含了参数矩阵,那么它就需要提供接口来对其所包含的矩阵进行初始化、读取或加载矩阵中的内容等。

这些接口大部分都以模板方法的形式提供。这就确保了可以采用不同类型的参数调用层所提供的方法——这一点对正向、反向传播尤其重要——它与表达式模板一起构成了性能优化的前提。我们会在后面看到这一点。

层在设计与使用上有很多独特之处,可以说,它是 MetaNN 框架中极富特色的一个组成部分。 MetaNN 中的很多子模块的引入都与层相关。本节将讨论基本层及其涉及到的子模块,下一节将在此基础上讨论组合层的一些特色之处。

层的声明

通常来说,层会被声明为一个类模板,比如:

template <typename TInputs, typename TPolicies>
class AddLayer;

其中包含了两个模板参数,前者为输入映射的定义,后者为 Policy 的定义。接下来,我们将一一讨论这两部分内容。

输入/输出端口集合 与 输入映射

每个层都会在其内部指定其能够接收的数据个数与名称,以及输出结果的个数与名称。比如,对于加法层来说:

template <typename TInputs, typename TPolicies>
class AddLayer
{
public:
    using InputPortSet = LayerPortSet<struct LeftOperand, struct RightOperand>;
    using OutputPortSet = LayerPortSet<struct LayerOutput>;
    // ...
};

这表明加法层的输入接收两个参数,分命名为 LeftOperandRightOperand;它会产生一个输出结果,被命名为 LayerOutput

InputPortSetOutputPortSet 称为层的输入/输出端口集合。端口集合中的名称会在正向、反向传播时使用。

输入/输出端口集合只指定了层的输入输出参数个数与名称,并未对层的具体输入类型进行限制,对于仅用于推导(inference)的层来说,这一点就够了。对于用于训练(train)的层来说,还需要给出每个参数的具体类型,这个信息是通过输入映射(层模板的第一个参数)指定的。

可以通过如下的方式来指定层的输入映射:

using CommonInputMap = LayerInMap<LayerKV<LeftOperand, Matrix<CheckElement, CheckDevice>>,
                                  LayerKV<RightOperand, Matrix<CheckElement, CheckDevice>>
                                 >;
AddLayer<CommonInputMap, ...> layer;

其中 CommonInputMap就是一个输入映射,它表明 LeftOperandRightOperand的类型均为 Matrix<CheckElement, CheckDevice>

对于仅用于推导的层来说,输入映射可以为空(使用 NullParameter 表示)。因此,以下的代码相当于构造了一个仅用于推导的层:

AddLayer<NullParameter, ...> layer;

MetaNN 同时提供了辅助函数 MakeTrainLayerMakeInferLayer 来构造用于训练的层与用于推导的层。

Policy 与层的声明

Policy 简述

我们希望能对层的具体行为进行控制,进一步,希望在编译期就指定层的具体行为——从而尽量早地获取信息,提升系统优化的空间。 MetaNN 使用 Policy 机制来实现这一点。

Policy 是一种编译期构造,它指定了函数或方法在使用时的具体行为。一个典型的 Policy 为:使用乘法而非加法进行累积——累积函数可以根据其中的信息改变其缺省的行为。

MetaNN 中所使用的 Policy 是被系统化地组织起来的。其中的每个 Policy 都包含了一个主体类别,次要类别,以及相关的缺省值。主体类别用于 Policy 的分类,次要类别用于 Policy 对象的互斥(我们很快就会讨论到 Policy 对象):

struct GradPolicy
{
    using MajorClass = GradPolicy;

    struct IsUpdateValueCate;
    struct IsFeedbackOutputValueCate;

    static constexpr bool IsUpdate = false;
    static constexpr bool IsFeedbackOutput = false;
};

这段代码定义了两个 Policy : IsUpdateIsFeedbackOutput 。其主体类别为 GradPolicy ——这表示了两个 Policy 与反向传播相关。它们所属的次要类别分别为 IsUpdateValueCateIsFeedbackOutputValueCate (次要类别使用 Policy 的名称作为前缀,这一点是专门设计的;而次要类别名称中的 ValueCate 则表示了该类别所对应的 Policy 是数值 Policy)。它们的缺省值均为 false

MetaNN 不仅支持数值 Policy ,还支持类型 Policy (由于 Policy 是编译期的构造,因此支持类型就并不奇怪了):

struct ParamPolicy
{
    using MajorClass = ParamPolicy;

    struct ParamTypeTypeCate;
    using ParamType = NullParameter;

    struct InitializerTypeCate;
    using Initializer = NullParameter;

    struct ElementTypeTypeCate;
    using  ElementType = float;

    struct DeviceTypeTypeCate;
    using  DeviceType = DeviceTags::CPU;
};

ElementType 是类型 Policy ,其次要类型为 ElementTypeTypeCate ,缺省“值” 为 float

事实上, MetaNN 中的 Policy 还能是其它形式的,比如我们甚至可以定义取值为类模板的 Policy 。但数值、类型与数值 Policy 在 MetaNN 中使用的频率最高, MetaNN 也为这两种 Policy 专门提供了宏以建立 Policy 对象。

Policy 对象

Policy 与 Policy 对象的关系很像类与其对象的关系。 Policy 对象可以被视做某个 Policy 实例化的结果。Policy 对象包含了相应 Policy 的主体与次要类别信息,但其取值可能不是 Policy 的缺省值:

ValuePolicyObj(PUpdate,   GradPolicy, IsUpdate, true);
ValuePolicyObj(PNoUpdate, GradPolicy, IsUpdate, false);

上述代码定义了两个 Policy 对象:PUpdatePNoUpdate 。它们均对应于之前提及的 IsUpdate Policy。但其取值一个为 true ,一个为 false 。类似的,还可以定义取值为类型的 Policy 对象:

TypePolicyObj(PNormVarScale,    VarScaleFillerPolicy, Distribute, Norm);
TypePolicyObj(PUniformVarScale, VarScaleFillerPolicy, Distribute, Uniform);

除了 ValuePolicyObjTypePolicyObj 外, MetaNN 还提供了 ValuePolicyTemplateTypePolicyTemplate 来构造 Policy 对象,比如:

ValuePolicyTemplate(PUpdateIs, GradPolicy, IsUpdate);

将构造一个 policy 对象模板 PUpdateIs,我们可以将 PUpdateIs<true> 或者 PUpdateIs<false> 作为 Policy 对象使用。 接下来,让我们看一下如何使用 Policy 对象来指定层的行为细节。

在层中使用 Policy 对象

MetaNN 中的大部分层都被声明为模板。接收两个模板参数,其中的之一是一个“容器”,可以存放零到多个 Policy 对象:

template <typename TInputMap, typename TPolicies>
class AddLayer;

// layer is an object with interfaces such as feed-backward and feed-forward;
AddLayer<TInputMap, PolicyContainer<PUpdate, PFeedbackOutput>> layer;

上述程序使用 Policy 对象的列表作为模板参数,实例化了 AddLayer 模板。可以利用 MetaNN 所提供的元函数使得定义更加清晰易懂:

using MyLayer = MakeInferLayer<AddLayer, PUpdate, PFeedbackOutput>;
MyLayer layer;

Policy 对象的顺序并不重要。以下两个层的功能是相同的(虽然它们的 C++ 类型并不相同):

using MyLayer1 = MakeInferLayer<AddLayer, PRowVecInput, PUpdate>;
using MyLayer2 = MakeInferLayer<AddLayer, PUpdate, PRowVecInput>;

但并非所有的组合都是有效的,比如下面的声明就是非法的:

using MyLayer1 = MakeInferLayer<AddLayer, PNoUpdate, PUpdate>;

显然,这是因为一个层不能同时具有“更新内部参数”以及“不更新内部参数”两种状态。声明这样的层会导致编译器报错:在同一个层的声明中包含了多个属于相同次要类别的 Policy 对象。

对于特定的层来说,并非所有的 Policy 对象都会对其行为产生影响。比如,AddLayer 并不包含内部参数,因此设置 PUpdate 并不会导致相应的参数更新。此时,相应 Policy 对象设置与否不会对层的行为细节产生影响。也即,下面的两个声明具有相同的行为:

// MyLayer1 = AddLayer<NullParameter, PolicyContainer<PUpdate>>
using MyLayer1 = MakeInferLayer<AddLayer, PUpdate>;

// MyLayer2 = AddLayer<NullParameter, PolicyContainer<>>
using MyLayer2 = MakeInferLayer<AddLayer>;

另一方面,如果某个层需要根据特定的 Policy 来决定其行为,但该 Policy 所对应的对象在层的实例化时并未指定,那么该层将使用该 Policy 所对应的缺省值来确定其行为细节。因此,下面两个声明具有相同的行为细节:

using MyLayer1 = MakeInferLayer<AddLayer, PNoFeedbackOutput>;
using MyLayer2 = MakeInferLayer<AddLayer>;

最后,让我们少许深入一下层的实现,看一下层是如何获取 Policy 所对应的值的:

template <typename TInputMap, typename TPolicies>
class AddLayer {
public:
    static constexpr bool IsFeedbackOutput
    	= PolicySelect<FeedbackPolicy, TPolicies>::IsFeedbackOutput;

实际的实现过程与这里所讨论的有少许出入(因为涉及到组合层,需要进一步处理)。但这里所展示的代码并不影响我们对逻辑的理解。 MetaNN 提供了元函数 PolicySelect 来选择 Policy 。它接收 PolicyContainer 列表以及要查询的 Policy 的主体、次要类别,返回相应的 Policy 值(当然,对于类型 Policy ,则返回相应的类型信息)。

层会在其内部根据 Policy 的取值对其行为进行调整,尽量减少不必要的计算。

正向、反向传播

除了构造函数外,每个层都至少需要提供两个接口,用于进行数据的正向、反向传播。正向、反向传播的输入与输出均是容器。正向传播的输入容器与反向传播的输出容器具有相同的“结构”,在 MetaNN 中称为层的输入容器;正向传播的输出容器与反向传播的输入容器具有相同的“结构”,在 MetaNN 中称为层的输出容器。本节将首先讨论容器的一些特性,之后对正向、反向传播函数的一些特点进行讨论。

容器

以正向传播为例,通常来说,层会接收一到多个数据作为输入,计算后产生一到多个数据作为输出。一些深度学习框架使用类似 std::vector 这样的线性表来输入与输出的数据结构——输入数据用一个 vector 表示,输出数据用另一个 vector 表示。这种方法有两个缺点:首先,无法通过 vertor 数据结构本身来区分传递给层的参数。比如,对于一个计算了两个矩阵相减的层,可以规定传入 vector 的第一个参数为减数,第二个为被减数。但这种规定并没有从程序的角度有所体现,可能会存在误用。

其次的一个问题是 vector 这样的列表只能存储类型相同的元素。层的输入与输出数据类型千变万化,使用 vector 这样的列表无疑限制了输入输出的数据类型。

MetaNN 定义了一种编译期容器解决上述问题。使用该容器的第一步是声明其结构:

struct A; struct B; struct Weight;
struct FParams : public VarTypeDict<A, B, Weight> {};

这里定义了一个容器 VarTypeDict<A, B, Weight> 。其中 A, B, Weight 为关键字,每个关键字可以对应存储一个数据对象。为了简化后续的使用,这里利用派生为该容器引入了一个别名 FParams 。当然,也可以通过如下的方式引入别名:

struct A; struct B; struct Weight;
using FParams = VarTypeDict<A, B, Weight>;

进一步,可以将上述代码进一步简化为:

using FParams = VarTypeDict<struct A, struct B, struct Weight>;

使用派生的方法能够获得一个额外的好处。如果容器中只需要包含一个 Key ,那么我们可以让容器名与 Key 的名称相同:

struct LayerIO : public VarTypeDict<LayerIO> {};

声明了容器的构造后,接下来就可以建立容器对象并向其中添加数据了:

auto data = FParams::Create().Set<A>(true)
                             .Set<B>(std::string("abc"))
                             .Set<Weight>(15);

上述代码并不难理解,它建立了 FParams 容器的实例,并为每个 Key 指定了一个相应的 Value。

这里有几点需要说明:首先,不难看出,每个 Key 所对应的 Value 类型可以是不同的。这似乎有点像一些弱类型语言所提供的支持。但 MetaNN 还是依赖于 C++ 这个强类型的语言进行的构造,考虑下面的代码:

auto data1 = FParams::Create().Set<A>(true)
                              .Set<B>(std::string("abc"))
                              .Set<Weight>(15);

auto data2 = FParams::Create().Set<A>(12)
                              .Set<B>(3.5)
                              .Set<Weight>(std::vector<int>());

data1 = data2;    // error!

data1data2 的类型是不同的:它们所对应数值的类型有所区别。

其次,FParams::Create().Set... 之后得到的结果类型并非 FParams ,而是由 FParams 所推导出的一个类型。用户不需要关注具体的类型,只需要使用 auto 来定义变量,由编译器自动推导即可。

第三,容器中元素的设置顺序并不需要与容器声明时 Key 的顺序相同,我们也可以这样写:

auto data = FParams::Create().Set<B>(std::string("abc"))
                             .Set<A>(true)
                             .Set<Weight>(15);

可以通过如下的方式获取容器中的对象:

auto a = data.template Get<A>();

注意编译器会进行自动推导,为 a 指定正确的数据类型:比如 A 对应的对象为 float 型的,那么在执行上述代码后, a 也就是 float 类型的变量了。

容器与正向、反向传播

MetaNN 中的层接提供了正向、反向传播的接口,这些接口接收容器对象作为参数,返回容器对象作为结果。以 AddLayer 为例:

template <typename TPolicies>
class AddLayer {
    using InputPortSet = LayerPortSet<struct LeftOperand, struct RightOperand>;
    using OutputPortSet = LayerPortSet<struct LayerOutput>;

	// ...
    template <typename TIn>
    auto FeedForward(const TIn& p_in)
    {
        const auto& val1 = p_in.template Get<LeftOperand>();
        const auto& val2 = p_in.template Get<RightOperand>();

        return OutputType::Create().template Set<LayerOutput>(val1 + val2);
    }

    template <typename TGrad>
    auto FeedBackward(TGrad&& p_grad)
    {
        const auto& res = p_grad.template Get<LayerOutput>();
	    return AddLayerInput::Create().template Set<LeftOperand>(res)
    	                              .template Set<RightOperand>(res);
    }
};

通过使用容器,我们避免了前文所述的使用 vector 这样的结构所带来的问题:首先,容器中的每个对象都使用 Key 来检索,这减少了使用错误的可能性。其次,容器中的每个 Key 所对应的对象可以是不同类型的——这就简化了代码编写。

层的正向反向传播并不进行实际计算

考虑如下的代码:

auto i1 = Matrix<float, DeviceTags::CPU>(2, 3);
auto i2 = Matrix<float, DeviceTags::CPU>(2, 3);

auto input = AddLayerInput::Create().Set<LeftOperand>(i1)
                                    .Set<RightOperand>(i2);
auto out = layer.FeedForward(input).Get<LayerOutput>();

此时 out 的类型为:

Operation<OpTags::Add, 
          OperandContainer<Matrix<float, DeviceTags::CPU>, Matrix<float, DeviceTags::CPU>>
          PolicyContainer<>
         >

它是一个模板表达式,而非实际的计算结果。正是由于我们将层的正向反向传播的接口设置成模板的,同时使用容器作为输入输出对象,从而实现了这个效果。对于很多深度学习框架来说,引入层就意味着将计算限制在了层的内部(层的输入与输出是计算好的结果),这就对系统的整体性能优化产生了不利的影响。但 MetaNN 使用了模板与编译期计算,有效地解决了这个问题!在 MetaNN 中,层只是用于代码与模板表达式的组织,最终的计算会留到最后一并进行。

层的其它接口

MetaNN 中的每个层都需要提供正向、反向传播用的接口。除此之外,它们还可能提供如下的接口:

  • Init: 用于初始化层中所包含的参数。
  • LoadWeights: 用于从外部读取参数。
  • SaveWeights: 用于将参数保存到外部。
  • GradCollect: 用于获取梯度信息。
  • NeutralInvariant: 用于断言层处于中性状态。

关于NeutralInvariant

这里解释一下最后一个接口。 MetaNN 的层负责保存正向、反向传播的一些中间结果,以供梯度计算等操作使用。在层对象被使用之前,其中不应当包含任何中间结果。我们将这种状态称为“中性”状态。层在进行完一次正向传播,一次反向传播,获取相应的梯度信息后,也不应该再拥有任何中间结果了。也即,此时层也应当处于中性状态。我们可以在一次正向传播,一次反向传播之后调用 NeutralInvariant 对此进行断言,从而确保程序的正确性。

通用接口

除了正向、反向传播之外,其余的接口都是可选的:并非所有的层都会包含全部的接口。比如,有的层并没有内部参数,因此也就不需要 Init 等接口了。因此给定一个层对象 layer ,调用 layer.Init 可能会出错。为了解决这个问题, MetaNN 引入了如下的通用接口函数(最后两个函数是为了保持一致性而引入的):

  • LayerInit: 对 Init 的封装
  • LayerLoadWeights: 对 LoadWeights 的封装
  • LayerSaveWeights: 对 SaveWeights 的封装
  • LayerGradCollect: 对 GradCollect 的封装
  • LayerNeutralInvariant: 对 NeutralInvariant 的封装
  • LayerFeedForward: 对 FeedForward 的封装
  • LayerFeedBackward: 对 FeedBackward 的封装

比如,可以调用 LayerInit(layer, ...) 来实现初始化。 MetaNN 足够智能,如果 layer 对象没有包含 Init 接口,那么 LayerInit 什么也不干。否则就会调用 layer 对象的 Init 接口以实现初始化。

复合层

复合层实现了典型的组合模式。它对基本层或者其它复合层进行封装,使用相对基本的元素构建复杂的网络结构。

设计理念

与基本层相比,复合层由于其天然的特性,引入了两个新的概念:父子关系与拓扑结构。

复合层是基本层的组合,很自然地,复合层与其所包含的基本层就形成了一种父子关系。对于基本层,我们可以通过 Policy 来调整其行为细节。与之类似,我们也希望使用 Policy 来调整复合层的行为。但由于这种父子关系的存在,因此我们需要引入一种机制,可以方便地指定每个子层的行为细节。

通常来说,复合层中的基本层组成了一个有向无环图的结构。复合层需要引入一种机制来描述这种结构,在给定了结构的基础上,实现自动的正向与反向传播。

MetaNN 中实现了几个简单的复合层:主要是出于展示目的,可以通过其了解到复合层的实现方法。接下来,我们将以 WeightLayer 为例描述复合层的构造。WeightLayer 接收一个输入张量,将其与另一个张量点乘并返回。

描述拓扑结构

建立复合层的第一步是要描述该复合层所包含的子层,以及这些子层的连接关系——即复合层的拓扑结构。首先,我们要为复合层中的每个子层给出唯一的名字。注意这个名字并不是层的类型名——一个复合层中可能包含多个具有相同类型的子层,它们也需要进行区分:

struct ParamSublayer;
struct DotSublayer;

上述代码表明: WeightLayer 包含了两个子层,分别命名为 ParamSublayerDotSublayer

在此基础上,我们可以定义复合层的拓扑结构了:

struct LayerInput; struct LayerOutput;
using Topology = ComposeTopology<Sublayer<ParamSublayer, ParamSourceLayer>,
                                 Sublayer<DotSublayer, DotLayer>,
                                 InConnect<LayerInput, DotSublayer, LeftOperand>,
                                 InternalConnect<ParamSublayer, LayerOutput, DotSublayer, RightOperand>,
                                 OutConnect<DotSublayer, LayerOutput, LayerOutput>>;

ComposeTopology 是 MetaNN 中的一种构造。用于描述复合层的拓扑结构。它是一个模板,可以接收四种类型的模板参数:

  • Sublayer: 将子层的名称与层的类型关联起来。
  • InConnect: 将复合层的输入关联到某个子层的输入上。
  • OutConnect: 将某个子层的输出关联到复合层的输出上。
  • InternalConnect: 将某个子层的输出关联到另一个子层的输入上。

因此,上面 Topology 中语句的含意依次是:

  • WeightLayer 的子层 ParamSublayer 是一个 ParamSourceLayer
  • WeightLayer 的子层 DotSublayer 是一个 DotLayer
  • WeightLayer 的输入容器中有一个 Key 为 LayerInput 的元素,这个元素应当与 DotSublayer 子层输入容器的 LeftOperand 相关联
  • ParamSublayer 子层的输出容器中有一个 Key 为 LayerOutput 的元素,将这个元素关联到 DotSublayer 子层输入容器的 RightOperand
  • DotSublayer 子层的输出容器中 Key 为 LayerOutput 的元素应当关联到 WeightLayer 输出容器中 Key 为 LayerOutput 的元素中

结合 WeightLayer 所实现的功能,这个描述就很容易理解了。需要说明的是,描述中语句的顺序是任意的。比如:下面的描述与上面的描述等价:

using Topology = ComposeTopology<Sublayer<ParamSublayer, ParamSourceLayer>,
                                 InConnect<LayerInput, DotSublayer, LeftOperand>,
                                 InternalConnect<ParamSublayer, LayerOutput, DotSublayer, RightOperand>,
                                 Sublayer<DotSublayer, DotLayer>,
                                 OutConnect<DotSublayer, LayerOutput, LayerOutput>>;

定义复合层

在有了拓扑结构后,复合层的定义就非常简单了。以下是 WeightLayer 的完整定义:

template <typename TInputs, typename TPolicies>
using Base = ComposeKernel<LayerPortSet<LayerInput>, LayerPortSet<LayerOutput>, TInputs, TPolicies, Topology>;

template <typename TInputs, typename TPolicies>
class WeightLayer : public Base<TInputs, TPolicies>
{
    using TBase = Base<TInputs, TPolicies>;

public:
    template <typename... TShapeParams>
    WeightLayer(const std::string& p_name, TShapeParams&&... shapeParams)
        : TBase(TBase::CreateSublayers()
                    .template Set<ParamSublayer>(p_name + "/param", std::forward<TShapeParams>(shapeParams)...)
                    .template Set<DotSublayer>(p_name + "/add"))
    { }
};

ComposeKernel 是 MetaNN 中的一个模板。它实现了几乎所有复合层所需要的功能。这个模板接收5个参数,分别表示了复合层的输入/输出端口集合,输入类型映射,Policy 容器以及拓扑结构。之所以还要在这里进行派生,是因为 ComposeKernel 中没有对构造函数进行定义(我们需要在复合层的构造函数中调用其子层的构造函数)。这是 WeightLayer 定义中唯一需要完成的工作。

复合层的定义并不麻烦,这是因为 ComposeKernel 完成了绝大多数的工作,其中就包括一个编译期实现的拓扑排序的逻辑。正是这个逻辑使得我们可以在仅仅定义了这些信息的基础上实现自动的正向、反向传播。

复合层的使用

复合层的使用与基本层非常相似,唯一的区别是 Policy 的指定。在 MetaNN 中,子层的 Policy 会“继承”自复合层的 Policy ,同时,子层也可以指定其自己的 Policy ,如果复合层与其子层的 Policy 出现冲突,那么以子层的为准:

// use default policies
using RootLayer1 = MakeLayer<WeightLayer>;

// All sublayers have Policy1
using RootLayer2 = MakeLayer<WeightLayer, Policy1Is<true>>;

// ParamSublayer has Policy1Is<false>, 
// DotSublayer has Policy1Is<true>
using RootLayer3 =
    MakeLayer<WeightLayer, Policy1Is<true>,
              SubPolicyContainer<ParamSublayer, Policy1Is<false>>>;

// Similar to RootLayer3
using RootLayer4 =
    MakeLayer<WeightLayer, Policy1Is<false>
              SubPolicyContainer<DotSublayer, Policy1Is<true>>>;

SubPolicyContainer 是 MetaNN 专门引入的容器,用于描述子层与其父亲之间 Policy 的差异。

求值

MetaNN 的一个主体思想就是将求值的过程后移。层的正向、反向传播的结果被表示为表达式模板,并不进行实际的求值。通常来说,在进行完正向/反向传播后,我们会得到若干表达式模板所对应的对象,此时,我们可以进行统一的求值。我们将在本节看到:采用这种方式,使得我们可以在多个维度上对求值过程进行优化。

最简单的求值语句

最简单的求值语句调用方式如下:

auto rm1 = Matrix<int, DeviceTags::CPU>(4, 5);
auto rm2 = Matrix<int, DeviceTags::CPU>(4, 5);
auto add = rm1 + rm2;

auto add_r = Evaluate(add);

Evaluate 会接收一个表达式模板,返回该表达式模板所对应的值。

Evaluate 的实现细节

Evaluate 的内部定义如下:

template <typename TData>
auto Evaluate(const TData& data)
{
    auto evalHandle = data.EvalRegister();
    EvalPlan::Eval();
    return evalHandle.Data();
}

Evaluate 内部实际上是对框架一般性求值过程的一个封装。对于任何一个表达式模板(甚至大部分 MetaNN 中的数据类型),我们都可以调用 EvalRegister 函数来进行求值的注册,注册将返回一个句柄。之后,我们需要调用 EvalPlan::Eval() 对注册的函数进行求值。在求值之后,我们可以通过返回句柄的 Data 方法来获取计算结果。

相比 Evaluate 来说,这种一般性的求值过程可能更有用。考虑如下的代码:

auto rm1 = Matrix<int, DeviceTags::CPU>(4, 5);
auto rm2 = Matrix<int, DeviceTags::CPU>(4, 5);
auto rm3 = Matrix<int, DeviceTags::CPU>(4, 5);
auto add1 = rm1 + rm2;
auto add2 = rm1 + rm3;

auto handle1 = add1.EvalRegister();
auto handle2 = add2.EvalRegister();
EvalPlan<DeviceTags::CPU>::Eval();
auto res1 = handle1.Data();
auto res2 = handle2.Data();

在调用 EvalPlan::Eval 之前,程序注册了两个求值过程,那么在 EvalPlan::Eval 具体求值时,系统就可以进行一定程度上的并行化,从而提升其性能。

而这就涉及到了求值的优化问题了。接下来,我们将讨论几种可能的优化(需要说明的是,现有的 MetaNN 并没有实现后续讨论的大部分优化技术。但 MetaNN 已经把框架搭好了,在此基础上实现这些技术是相对平凡的工作):

求值优化之避免重复计算

进行注册的多个表达式模板之间可能存在某些关系。比如它们的一些子树的相同的。 MetaNN 引入了自动分析机制,可以确保在这种情况下,相同的子树只会被求值一次。同时如果某个子树之前已经完成了求值,则不会被再次求值。

求值优化之针对运算特性的优化

运算中可以引入额外的参数,其中一些编译期可以指定的参数是通过 Policy 引入到运算中的。这些编译期指定的参数会对运算的行为进行调整。如果运算的行为满足某些特性,那么就可以根据其进行优化。可以引入编译期分支,根据运算的特性进行优化。编译期分支不会对运行期性能产生不利影响。

求值优化之同类操作的合并

为了达到较高的计算速度,框架的底层需要调用一些高效计算的库。比如 Intel 的 MKL ,或者 CUDA 等等。这些库提供了一些函数,可以将多个计算合并进行。比如同时计算多组矩阵点乘。使用求值注册的方法,我们可以将多个此类任务合并到一起,在调用 EvalPlan::Eval 时一并完成,从而提升计算速度。

求值优化之不同操作的合并

大部分情况下,对表达式模板的求值可以看图的拓扑排序:要对某个结点求值,就需要先对前驱结点求值。但在一些情况下,我们可以简化这个求值的过程。比如,如果当前结点是求对数,而其子结点是求指数,那么我们可以将求对数与求指数的操作抵消掉。我们还可以引入更复杂的机制,在一定程度上简化求值的过程。

事实上,这是一种非常典型的求值优化手段。 MetaNN 中为此专门引入了 OperSeq_ 模板以实现类似的功能。 OperSeq_ 实现了典型的职责链模式,可以处理表达式模板中“特殊”的情况。以 RowSoftmaxGrad 为例,MetaNN 定义了:

template <>
struct OperSeq_<OpTags::SoftmaxGrad>
{
    using type = OperCalAlgoChain<NSCaseNLLLossGrad::Calculator,
                                  TailCalculator<NSCaseGen::EvalItem,
                                                 NSCaseGen::EvalGroup,
                                                 PolicyContainer<PPassPolicy>>>;
};

它就表示了:首先看看当前结构是否满足 NSCaseNLLLossGrad::Calculator ,如果满足,就按照 NSCaseNLLLossGrad::Calculator 所定义方式计算。否则再尝试一般方式(TailCalculator)。

总结

以上就是 MetaNN 的一个简单的介绍了。与其它的深度学习框架相比, MetaNN 更深层次地使用了模板元编程的技术。正如文章开头所述,它目前只是一个骨架,距实际的应用还有一定的距离。同时使用这套框架也需要一定的模板元编程基础,这可能对很多 C++ 开发者来说都是一块不算熟悉的领域。但它也有它的优势。

框架采用了模板元编程的技术,将计算优化与模型结构彻底分开。这能带来很多好处。比如,我们可能并不需要框架对 mini-batch 的显式支持了。大部分情况下(Batch Normalization 除外),引入 Batch 是为了快速计算,但 Batch 的引入会导致系统变得更加复杂。比如在神经网络机器翻译中,为了使用 Batch ,通常来说我们需要对训练语料按照源语言长度进行排序,引入填充标签等,以保证 Batch 可以更高效地进行。我们甚至要处理高维数据:很多时候我们需要使用矩阵来表示中间结果,而要使用三维或者更高维的结构来表示 Batch 的中间结果——这大大增加了程序的复杂度,影响其理解。但在 MetaNN 中,大部分情况下,我们需要做的只是引入循环,在每个循环中只处理一个条目。循环结束后统一调用 EvalPlan::Eval 即可,EvalPlan::Eval 会帮我们完成之前需要 Batch 才能进行的计算合并。程序的逻辑会更加清楚。

另一个明显的好处是用户在使用上会更加省心,不用考虑一些小技巧。一个非常典型的例子是:很多深度学习框架都包含了 SoftmaxSoftmaxLoss 层或者类似的构造,这是因为后者需要解决 Softmax 可能会引起的数值稳定性问题。使用 MetaNN ,这个问题被掩盖在了求值优化之中。用户不需要再关注这些层行为的细节差异了。

除了深度学习框架之外, MetaNN 对于 C++ 模板元编程也有其贡献。Policy 的概念并非 MetaNN 原创,但它对 Policy 的实现进行了扩展,使得在一个模板中 Policy 的个数不再有限制(事实上,Policy 的个数还会取决于编译器对变长模板的支持程度)。MetaNN 还实现了一个编译期容器,对数据分类、Policy 组织方面都有深入的探讨。它所实现的编译期拓扑排序也将元编程的技术从“小技巧”扩展成切实可用的算法层面。可以说,如果能将这套代码读懂,那么对 C++ 元编程就会有相对深入的理解了。

最后要说的一点是:本套代码没有注释。这个文档是 MetaNN 的一个简介,但并没有深入到其中的细节之中,想通过这篇文档来理解程序的实现还是比较困难的。 MetaNN 包含了很多细节,之所以没有注释,是因为如果读者对 C++ 模板元编程有所了解,那么我认为阅读这个代码中的逻辑对于这些读者来说是比较直白的。反过来,如果读者不了解 C++ 模板元编程,或者了解较少,那么加了注释并不能起到多大作用。要完全描述出这个框架中的实现细节,需要一本书的篇幅——目前该书的中文版(《C++模板元编程实战》)已经出版。但书中的内容是在refactor之前完成的,与当前branch上的内容有一定的出入。书中的内容对应了book分支。目前考虑再出版一个版本,对应当前的系统框架。

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published