深度概率编程库

查看源文件

MindSpore深度概率编程的目标是将深度学习和贝叶斯学习结合,包括概率分布、概率分布映射、深度概率网络、概率推断算法、贝叶斯层、贝叶斯转换和贝叶斯工具箱,面向不同的开发者。对于专业的贝叶斯学习用户,提供概率采样、推理算法和模型构建库;另一方面,为不熟悉贝叶斯深度学习的用户提供了高级的API,从而不用更改深度学习编程逻辑,即可利用贝叶斯模型。

概率分布

概率分布(mindspore.nn.probability.distribution)是概率编程的基础。Distribution 类提供多样的概率统计接口,例如概率密度函数 pdf 、累积密度函数 cdf 、散度计算 kl_loss 、抽样 sample 等。现有的概率分布实例包括高斯分布,伯努利分布,指数型分布,几何分布和均匀分布。

概率分布类

  • Distribution:所有概率分布的基类。

  • Bernoulli:伯努利分布。参数为试验成功的概率。

  • Exponential: 指数型分布。参数为率参数。

  • Geometric:几何分布。参数为一次伯努利试验成功的概率。

  • Normal:正态(高斯)分布。参数为均值和标准差。

  • Uniform:均匀分布。参数为数轴上的最小值和最大值。

Distribution基类

Distribution 是所有概率分布的基类。

接口介绍:Distribution 类支持的函数包括 problog_probcdflog_cdfsurvival_functionlog_survivalmeansdvarentropykl_losscross_entropysample 。分布不同,所需传入的参数也不同。只有在派生类中才能使用,由派生类的函数实现决定参数。

  • prob :概率密度函数(PDF)/ 概率质量函数(PMF)。

  • log_prob :对数似然函数。

  • cdf :累积分布函数(CDF)。

  • log_cdf :对数累积分布函数。

  • survival_function :生存函数。

  • log_survival :对数生存函数。

  • mean :均值。

  • sd :标准差。

  • var :方差。

  • entropy :熵。

  • kl_loss :Kullback-Leibler 散度。

  • cross_entropy :两个概率分布的交叉熵。

  • sample :概率分布的随机抽样。

伯努利分布(Bernoulli)

伯努利分布,继承自 Distribution 类。

属性:

  • Bernoulli.probs:伯努利试验成功的概率。

Distribution 基类调用 Bernoulli 中私有接口以实现基类中的公有接口。Bernoulli 支持的公有接口为:

  • meanmodevar:可选择传入 试验成功的概率 probs1

  • entropy:可选择传入 试验成功的概率 probs1

  • cross_entropykl_loss:必须传入 distprobs1_bdist 为另一分布的类型,目前只支持此处为 ‘Bernoulli’probs1_b 为分布 b 的试验成功概率。可选择传入分布 a 的参数 probs1_a

  • problog_probcdflog_cdfsurvival_functionlog_survival:必须传入 value 。可选择传入试验成功的概率 probs

  • sample:可选择传入样本形状 shape 和试验成功的概率 probs1

指数分布(Exponential)

指数分布,继承自 Distribution 类。

属性:

  • Exponential.rate:率参数。

Distribution 基类调用 Exponential 私有接口以实现基类中的公有接口。Exponential 支持的公有接口为:

  • meanmodevar:可选择传入率参数 rate

  • entropy:可选择传入率参数 rate

  • cross_entropykl_loss:必须传入 distrate_bdist 为另一分布的类型的名称, 目前只支持此处为 ‘Exponential’rate_b 为分布 b 的率参数。可选择传入分布 a 的参数 rate_a

  • problog_probcdflog_cdfsurvival_functionlog_survival:必须传入 value 。可选择传入率参数 rate

  • sample:可选择传入样本形状 shape 和率参数 rate

几何分布(Geometric)

几何分布,继承自 Distribution 类。

属性:

  • Geometric.probs:伯努利试验成功的概率。

Distribution 基类调用 Geometric 中私有接口以实现基类中的公有接口。Geometric 支持的公有接口为:

  • meanmodevar:可选择传入 试验成功的概率 probs1

  • entropy:可选择传入 试验成功的概率 probs1

  • cross_entropykl_loss:必须传入 distprobs1_bdist 为另一分布的类型的名称,目前只支持此处为 ‘Geometric’probs1_b 为分布 b 的试验成功概率。可选择传入分布 a 的参数 probs1_a

  • problog_probcdflog_cdfsurvival_functionlog_survival:必须传入 value 。可选择传入试验成功的概率 probs1

  • sample:可选择传入样本形状 shape 和试验成功的概率 probs1

正态分布(Normal)

正态(高斯)分布,继承自 Distribution 类。

Distribution 基类调用 Normal 中私有接口以实现基类中的公有接口。Normal 支持的公有接口为:

  • meanmodevar:可选择传入分布的参数均值 mean 和标准差 sd

  • entropy:可选择传入分布的参数均值 mean 和标准差 sd

  • cross_entropykl_loss:必须传入 distmean_bsd_bdist 为另一分布的类型的名称,目前只支持此处为 ‘Normal’mean_bsd_b 为分布 b 的均值和标准差。可选择传入分布的参数 a 均值 mean_a 和标准差 sd_a

  • problog_probcdflog_cdfsurvival_functionlog_survival:必须传入 value 。可选择分布的参数包括均值 mean_a 和标准差 sd_a

  • sample:可选择传入样本形状 shape 和分布的参数包括均值 mean_a 和标准差 sd_a

均匀分布(Uniform)

均匀分布,继承自 Distribution 类。

属性:

  • Uniform.low:最小值。

  • Uniform.high:最大值。

Distribution 基类调用 Uniform 以实现基类中的公有接口。Uniform 支持的公有接口为:

  • meanmodevar:可选择传入分布的参数最大值 high 和最小值 low

  • entropy:可选择传入分布的参数最大值 high 和最小值 low

  • cross_entropykl_loss:必须传入 disthigh_blow_bdist 为另一分布的类型的名称,目前只支持此处为 ‘Uniform’high_blow_b 为分布 b 的参数。可选择传入分布 a 的参数即最大值 high_a 和最小值 low_a

  • problog_probcdflog_cdfsurvival_functionlog_survival:必须传入 value 。可选择传入分布的参数最大值 high 和最小值 low

  • sample:可选择传入 shape 和分布的参数即最大值 high 和最小值 low

概率分布类在PyNative模式下的应用

Distribution 子类可在 PyNative 模式下使用。

导入相关模块:

from mindspore import Tensor
from mindspore import dtype as mstype
import mindspore.context as context
import mindspore.nn.probability.distribution as msd
context.set_context(mode=context.PYNATIVE_MODE)

Normal 为例, 创建一个均值为0.0、标准差为1.0的正态分布:

my_normal = msd.Normal(0.0, 1.0, dtype=mstype.float32)

计算均值:

mean = my_normal.mean()
print(mean)

输出为:

0.0

计算方差:

var = my_normal.var()
print(var)

输出为:

1.0

计算熵:

entropy = my_normal.entropy()
print(entropy)

输出为:

1.4189385

计算概率密度函数:

value = Tensor([-0.5, 0.0, 0.5], dtype=mstype.float32)
prob = my_normal.prob(value)
print(prob)

输出为:

[0.35206532, 0.3989423, 0.35206532]

计算累积分布函数:

cdf = my_normal.cdf(value)
print(cdf)

输出为:

[0.30852754, 0.5, 0.69146246]

计算 Kullback-Leibler 散度:

mean_b = Tensor(1.0, dtype=mstype.float32)
sd_b = Tensor(2.0, dtype=mstype.float32)
kl = my_normal.kl_loss('Normal', mean_b, sd_b)
print(kl)

输出为:

0.44314718

概率分布类在图模式下的应用

在图模式下,Distribution 子类可用在网络中。

导入相关模块:

import mindspore.nn as nn
from mindspore import Tensor
from mindspore import dtype as mstype
import mindspore.context as context
import mindspore.nn.probability.distribution as msd
context.set_context(mode=context.GRAPH_MODE)

创建网络:

# 网络继承nn.Cell
class Net(nn.Cell):
    def __init__(self):
        super(Net, self).__init__()
        self.normal = msd.Normal(0.0, 1.0, dtype=mstype.float32)

    def construct(self, value, mean, sd):
        pdf = self.normal.prob(value)
        kl = self.normal.kl_loss("Normal", mean, sd)
        return pdf, kl

调用网络:

net = Net()
value = Tensor([-0.5, 0.0, 0.5], dtype=mstype.float32)
mean = Tensor(1.0, dtype=mstype.float32)
sd = Tensor(1.0, dtype=mstype.float32)
pdf, kl = net(value, mean, sd)
print("pdf: ", pdf)
print("kl: ", kl)

输出为:

pdf: [0.3520653, 0.39894226, 0.3520653]
kl: 0.5

TransformedDistribution类接口设计

TransformedDistribution 继承自 Distribution ,是可通过映射f(x)变化得到的数学分布的基类。其接口包括:

  1. 类特征函数

    • bijector:无参函数,返回分布的变换方法。

    • distribution:无参函数,返回原始分布。

    • is_linear_transformation:无参函数,返回线性变换标志。

  2. 接口函数(以下接口函数的参数与构造函数中 distribution 的对应接口的参数相同)。

    • cdf:累积分布函数(CDF)。

    • log_cdf:对数累积分布函数。

    • survival_function:生存函数。

    • log_survival:对数生存函数。

    • prob:概率密度函数(PDF)/ 概率质量函数(PMF)。

    • log_prob:对数似然函数。

    • sample:随机取样。

    • mean:无参数。只有当 Bijector.is_constant_jacobian=true 时可调用。

PyNative模式下调用TransformedDistribution实例

TransformedDistribution 子类可在 PyNative 模式下使用。 在执行之前,我们需要导入需要的库文件包。

导入相关模块:

import numpy as np
import mindspore.nn as nn
import mindspore.nn.probability.bijector as msb
import mindspore.nn.probability.distribution as msd
import mindspore.context as context
from mindspore import Tensor
from mindspore import dtype
context.set_context(mode=context.PYNATIVE_MODE)

构造一个 TransformedDistribution 实例,使用 Normal 分布作为需要变换的分布类,使用 Exp 作为映射变换,可以生成 LogNormal 分布。

normal = msd.Normal(0.0, 1.0, dtype=dtype.float32)
exp = msb.Exp()
LogNormal = msd.TransformedDistribution(exp, normal, dtype=dtype.float32, seed=0, name="LogNormal")
print(LogNormal)

输出为:

TransformedDistribution<
  (_bijector): Exp<power = 0>
  (_distribution): Normal<mean = 0.0, standard deviation = 1.0>
  >

可以对 LogNormal 进行概率分布计算。例如:

计算累积分布函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
cdf = LogNormal.cdf(tx)
print(cdf)

输出为:

[7.55891383e-01, 9.46239710e-01, 9.89348888e-01]

计算对数累积分布函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
log_cdf = LogNormal.log_cdf(tx)
print(log_cdf)

输出为:

[-2.79857576e-01, -5.52593507e-02, -1.07082408e-02]

计算生存函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
survival_function = LogNormal.survival_function(tx)
print(survival_function)

输出为:

[2.44108617e-01, 5.37602901e-02, 1.06511116e-02]

计算对数生存函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
log_survival = LogNormal.log_survival(tx)
print(log_survival)

输出为:

[-1.41014194e+00, -2.92322016e+00, -4.54209089e+00]

计算概率密度函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
prob = LogNormal.prob(tx)
print(prob)

输出为:

[1.56874031e-01, 2.18507163e-02, 2.81590177e-03]

计算对数概率密度函数:

x = np.array([2.0, 5.0, 10.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
log_prob = LogNormal.log_prob(tx)
print(log_prob)

输出为:

[-1.85231221e+00, -3.82352161e+00, -5.87247276e+00]

调用取样函数 sample 抽样:

shape = ((3, 2))
sample = LogNormal.sample(shape)
print(sample)

输出为:

[[7.64315844e-01, 3.01435232e-01],
 [1.17166102e+00, 2.60277224e+00],
 [7.02699006e-01, 3.91564220e-01]])

当构造 TransformedDistribution 映射变换的 is_constant_jacobian = true 时(如 ScalarAffine),构造的 TransformedDistribution 实例可以使用直接使用 mean 接口计算均值,例如:

normal = msd.Normal(0.0, 1.0, dtype=dtype.float32)
scalaraffine = msb.ScalarAffine(1.0, 2.0)
trans_dist = msd.TransformedDistribution(scalaraffine, normal, dtype=dtype.float32, seed=0)
mean = trans_dist.mean()
print(mean)

输出为:

2.0

图模式下调用TransformedDistribution实例

在图模式下,TransformedDistribution 类可用在网络中。

导入相关模块:

import mindspore.nn as nn
from mindspore import Tensor
from mindspore import dtype
import mindspore.context as context
import mindspore.nn.probability.Bijector as msb
import mindspore.nn.probability.Distribution as msd
context.set_context(mode=self.GRAPH_MODE)

创建网络:

class Net(nn.Cell):
    def __init__(self, shape, dtype=dtype.float32, seed=0, name='transformed_distribution'):
        super(Net, self).__init__()
        # 创建TransformedDistribution实例
        self.exp = msb.Exp()
        self.normal = msd.Normal(0.0, 1.0, dtype=dtype)
        self.lognormal = msd.TransformedDistribution(self.exp, self.normal, dtype=dtype, seed=seed, name=name)
        self.shape = shape

    def construct(self, value):
        cdf = self.lognormal.cdf(value)
        sample = self.lognormal.sample(self.shape)
        return cdf, sample

调用网络:

shape = (2, 3)
net = Net(shape=shape, name="LogNormal")
x = np.array([2.0, 3.0, 4.0, 5.0]).astype(np.float32)
tx = Tensor(x, dtype=dtype.float32)
cdf, sample = net(tx)
print("cdf: ", cdf)
print("sample: ", sample)

输出为:

cdf:  [0.7558914 0.8640314 0.9171715 0.9462397]
sample:  [[0.21036398 0.44932044 0.5669641 ]
 [1.4103683  6.724116   0.97894996]]

概率分布映射

Bijector(mindspore.nn.probability.bijector)是概率编程的基本组成部分。Bijector描述了一种随机变量的变换方法,可以通过一个已有的随机变量X和一个映射函数f生成一个新的随机变量\(Y = f(x)\)Bijector 提供了映射相关的四种变换方法。它可以当做算子直接使用,也可以作用在某个随机变量 Distribution 类实例上生成新的随机变量的 Distribution 类实例。

Bijector类接口设计

Bijector基类

Bijector 类是所有概率分布映射的基类。其接口包括:

  1. 类特征函数

    • name:无参函数,返回 name 的值。

    • is_dtype:无参函数,返回 dtype 的值。

    • parameter:无参函数,返回 parameter 的值。

    • is_constant_jacobian:无参函数,返回 is_constant_jacobian 的值。

    • is_injective:无参函数,返回 is_injective 的值。

  2. 映射函数

    • forward:正向映射,创建派生类后由派生类的 _forward 决定参数。

    • inverse:反向映射,创建派生类后由派生类的 _inverse 决定参数。

    • forward_log_jacobian:正向映射的导数的对数,创建派生类后由派生类的 _forward_log_jacobian 决定参数。

    • inverse_log_jacobian:反向映射的导数的对数,创建派生类后由派生类的 _inverse_log_jacobian 决定参数。

Bijector 作为函数调用: 输入是一个 Distribution 类:生成一个 TransformedDistribution (不可在图内调用)

幂函数变换映射(PowerTransform)

PowerTransform 做如下变量替换:\(Y = g(X) = {(1 + X * c)}^{1 / c}\)。其接口包括:

  1. 类特征函数

    • power:无参函数,返回 power 的值。

  2. 映射函数

    • forward:正向映射,输入为 Tensor

    • inverse:反向映射,输入为 Tensor

    • forward_log_jacobian:正向映射的导数的对数,输入为 Tensor

    • inverse_log_jacobian:反向映射的导数的对数,输入为 Tensor

指数变换映射(Exp)

Exp 做如下变量替换:\(Y = g(X)= exp(X)\)。其接口包括:

映射函数

  • forward:正向映射,输入为 Tensor

  • inverse:反向映射,输入为 Tensor

  • forward_log_jacobian:正向映射的导数的对数,输入为 Tensor

  • inverse_log_jacobian:反向映射的导数的对数,输入为 Tensor

标量仿射变换映射(ScalarAffine)

ScalarAffine 做如下变量替换:Y = g(X) = a * X + b。其接口包括:

  1. 类特征函数

    • scale:无参函数,返回scale的值。

    • shift:无参函数,返回shift的值。

  2. 映射函数

    • forward:正向映射,输入为 Tensor

    • inverse:反向映射,输入为 Tensor

    • forward_log_jacobian:正向映射的导数的对数,输入为 Tensor

    • inverse_log_jacobian:反向映射的导数的对数,输入为 Tensor

Softplus变换映射(Softplus)

Softplus 做如下变量替换:\(Y = g(X) = log(1 + e ^ {kX}) / k \)。其接口包括:

  1. 类特征函数

    • sharpness:无参函数,返回 sharpness 的值。

  2. 映射函数

    • forward:正向映射,输入为 Tensor

    • inverse:反向映射,输入为 Tensor

    • forward_log_jacobian:正向映射的导数的对数,输入为 Tensor

    • inverse_log_jacobian:反向映射的导数的对数,输入为 Tensor

PyNative模式下调用Bijector实例

在执行之前,我们需要导入需要的库文件包。双射类最主要的库是 mindspore.nn.probability.bijector,导入后我们使用 msb 作为库的缩写并进行调用。

导入相关模块:

import numpy as np
import mindspore.nn as nn
import mindspore.nn.probability.bijector as msb
import mindspore.context as context
from mindspore import Tensor
from mindspore import dtype
context.set_context(mode=context.PYNATIVE_MODE)

下面我们以 PowerTransform 为例。创建一个指数为2的 PowerTransform 对象。

构造 PowerTransform

powertransform = msb.PowerTransform(power=2)
print(powertransform)

输出:

PowerTransform<power = 2>

接下来可以使用映射函数进行运算。

调用 forward 方法,计算正向映射:

x = np.array([2.0, 3.0, 4.0, 5.0], dtype=np.float32)
tx = Tensor(x, dtype=dtype.float32)
forward = powertransform.forward(tx)
print(forward)

输出为:

[2.23606801e+00, 2.64575124e+00, 3.00000000e+00, 3.31662488e+00]

输入 inverse 方法,计算反向映射:

inverse = powertransform.inverse(tx)
print(inverse)

输出为:

[1.50000000e+00, 4.00000048e+00, 7.50000000e+00, 1.20000010e+01]

输入 forward_log_jacobian 方法,计算正向映射导数的对数:

forward_log_jaco = powertransform.forward_log_jacobian(tx)
print(forward_log_jaco)

输出:

[-8.04718971e-01, -9.72955048e-01, -1.09861231e+00, -1.19894767e+00]

输入 inverse_log_jacobian 方法,计算反向映射导数的对数:

inverse_log_jaco = powertransform.inverse_log_jacobian(tx)
print(inverse_log_jaco)

输出为:

[6.93147182e-01  1.09861231e+00  1.38629436e+00  1.60943794e+00]

图模式下调用Bijector实例

在图模式下,Bijector 子类可用在网络中。

导入相关模块:

import mindspore.nn as nn
from mindspore import Tensor
from mindspore import dtype as mstype
import mindspore.context as context
import mindspore.nn.probability.Bijector as msb
context.set_context(mode=context.GRAPH_MODE)

创建网络:

class Net(nn.Cell):
    def __init__(self):
        super(Net, self).__init__()
        # 创建PowerTransform实例
        self.powertransform = msb.PowerTransform(power=2)

    def construct(self, value):
        forward = self.s1.forward(value)
        inverse = self.s1.inverse(value)
        forward_log_jaco = self.s1.forward_log_jacobian(value)
        inverse_log_jaco = self.s1.inverse_log_jacobian(value)
        return forward, inverse, forward_log_jaco, inverse_log_jaco

调用网络:

net = Net()
x = np.array([2.0, 3.0, 4.0, 5.0]).astype(np.float32)
tx = Tensor(x, dtype=dtype.float32)
forward, inverse, forward_log_jaco, inverse_log_jaco = net(tx)
print("forward: ", forward)
print("inverse: ", inverse)
print("forward_log_jaco: ", forward_log_jaco)
print("inverse_log_jaco: ", inverse_log_jaco)

输出为:

forward:  [2.236068  2.6457512 3.        3.3166249]
inverse:  [ 1.5        4.0000005  7.5       12.000001 ]
forward_log_jaco:  [-0.804719   -0.97295505 -1.0986123  -1.1989477 ]
inverse_log_jaco:  [0.6931472 1.0986123 1.3862944 1.609438 ]

深度概率网络

使用MindSpore深度概率编程库(mindspore.nn.probability.dpn)来构造变分自编码器(VAE)进行推理尤为简单。我们只需要自定义编码器和解码器(DNN模型),调用VAE或CVAE接口形成其派生网络,然后调用ELBO接口进行优化,最后使用SVI接口进行变分推理。这样做的好处是,不熟悉变分推理的用户可以像构建DNN模型一样来构建概率模型,而熟悉的用户可以调用这些接口来构建更为复杂的概率模型。VAE的接口在mindspore.nn.probability.dpn下面,dpn代表的是Deep probabilistic network,这里提供了一些基本的深度概率网络的接口,例如VAE。

VAE

首先,我们需要先自定义encoder和decoder,调用mindspore.nn.probability.dpn.VAE接口来构建VAE网络,我们除了传入encoder和decoder之外,还需要传入encoder输出变量的维度hidden size,以及VAE网络存储潜在变量的维度latent size,一般latent size会小于hidden size。

import mindspore.nn as nn
import mindspore.ops as ops
from mindspore.nn.probability.dpn import VAE

IMAGE_SHAPE = (-1, 1, 32, 32)


class Encoder(nn.Cell):
    def __init__(self):
        super(Encoder, self).__init__()
        self.fc1 = nn.Dense(1024, 800)
        self.fc2 = nn.Dense(800, 400)
        self.relu = nn.ReLU()
        self.flatten = nn.Flatten()

    def construct(self, x):
        x = self.flatten(x)
        x = self.fc1(x)
        x = self.relu(x)
        x = self.fc2(x)
        x = self.relu(x)
        return x


class Decoder(nn.Cell):
    def __init__(self):
        super(Decoder, self).__init__()
        self.fc1 = nn.Dense(400, 1024)
        self.sigmoid = nn.Sigmoid()
        self.reshape = ops.Reshape()

    def construct(self, z):
        z = self.fc1(z)
        z = self.reshape(z, IMAGE_SHAPE)
        z = self.sigmoid(z)
        return z


encoder = Encoder()
decoder = Decoder()
vae = VAE(encoder, decoder, hidden_size=400, latent_size=20)

ConditionalVAE

类似地,ConditionalVAE与VAE的使用方法比较相近,不同的是,ConditionalVAE利用了数据集的标签信息,属于有监督学习算法,其生成效果一般会比VAE好。

首先,先自定义encoder和decoder,并调用mindspore.nn.probability.dpn.ConditionalVAE接口来构建ConditionalVAE网络,这里的encoder和VAE的不同,因为需要传入数据集的标签信息;decoder和上述的一样。ConditionalVAE接口的传入则还需要传入数据集的标签类别个数,其余和VAE接口一样。

import mindspore.nn as nn
import mindspore.ops as ops
from mindspore.nn.probability.dpn import ConditionalVAE

IMAGE_SHAPE = (-1, 1, 32, 32)


class Encoder(nn.Cell):
    def __init__(self, num_classes):
        super(Encoder, self).__init__()
        self.fc1 = nn.Dense(1024 + num_classes, 400)
        self.relu = nn.ReLU()
        self.flatten = nn.Flatten()
        self.concat = ops.Concat(axis=1)
        self.one_hot = nn.OneHot(depth=num_classes)

    def construct(self, x, y):
        x = self.flatten(x)
        y = self.one_hot(y)
        input_x = self.concat((x, y))
        input_x = self.fc1(input_x)
        input_x = self.relu(input_x)
        return input_x


class Decoder(nn.Cell):
    def __init__(self):
        super(Decoder, self).__init__()
        self.fc1 = nn.Dense(400, 1024)
        self.sigmoid = nn.Sigmoid()
        self.reshape = ops.Reshape()

    def construct(self, z):
        z = self.fc1(z)
        z = self.reshape(z, IMAGE_SHAPE)
        z = self.sigmoid(z)
        return z


encoder = Encoder(num_classes=10)
decoder = Decoder()
cvae = ConditionalVAE(encoder, decoder, hidden_size=400, latent_size=20, num_classes=10)

加载数据集,我们可以使用Mnist数据集,具体的数据加载和预处理过程可以参考这里实现一个图片分类应用,这里会用到create_dataset函数创建数据迭代器。

ds_train = create_dataset(image_path, 128, 1)

接下来,需要用到infer接口进行VAE网络的变分推断。

概率推断算法

调用ELBO接口(mindspore.nn.probability.infer.ELBO)来定义VAE网络的损失函数,调用WithLossCell封装VAE网络和损失函数,并定义优化器,之后传入SVI接口(mindspore.nn.probability.infer.SVI)。SVI的run函数可理解为VAE网络的训练,可以指定训练的epochs,返回结果为训练好的网络;get_train_loss函数可以返回训练好后模型的loss。

from mindspore.nn.probability.infer import ELBO, SVI

net_loss = ELBO(latent_prior='Normal', output_prior='Normal')
net_with_loss = nn.WithLossCell(vae, net_loss)
optimizer = nn.Adam(params=vae.trainable_params(), learning_rate=0.001)

vi = SVI(net_with_loss=net_with_loss, optimizer=optimizer)
vae = vi.run(train_dataset=ds_train, epochs=10)
trained_loss = vi.get_train_loss()

最后,得到训练好的VAE网络后,我们可以使用vae.generate_sample生成新样本,需要传入待生成样本的个数,及生成样本的shape,shape需要保持和原数据集中的样本shape一样;当然,我们也可以使用vae.reconstruct_sample重构原来数据集中的样本,来测试VAE网络的重建能力。

generated_sample = vae.generate_sample(64, IMAGE_SHAPE)
for sample in ds_train.create_dict_iterator():
    sample_x = Tensor(sample['image'], dtype=mstype.float32)
    reconstructed_sample = vae.reconstruct_sample(sample_x)
print('The shape of the generated sample is ', generated_sample.shape)

我们可以看一下新生成样本的shape:

The shape of the generated sample is  (64, 1, 32, 32)

ConditionalVAE训练过程和VAE的过程类似,但需要注意的是使用训练好的ConditionalVAE网络生成新样本和重建新样本时,需要输入标签信息,例如下面生成的新样本就是64个0-7的数字。

sample_label = Tensor([i for i in range(0, 8)] * 8, dtype=mstype.int32)
generated_sample = cvae.generate_sample(sample_label, 64, IMAGE_SHAPE)
for sample in ds_train.create_dict_iterator():
    sample_x = Tensor(sample['image'], dtype=mstype.float32)
    sample_y = Tensor(sample['label'], dtype=mstype.int32)
    reconstructed_sample = cvae.reconstruct_sample(sample_x, sample_y)
print('The shape of the generated sample is ', generated_sample.shape)

查看一下新生成的样本的shape:

The shape of the generated sample is  (64, 1, 32, 32)

如果希望新生成的样本更好,更清晰,用户可以自己定义更复杂的encoder和decoder,这里的示例只用了两层全连接层,仅供示例的指导。

贝叶斯层

下面的范例使用MindSpore的nn.probability.bnn_layers中的API实现BNN图片分类模型。MindSpore的nn.probability.bnn_layers中的API包括NormalPriorNormalPosteriorConvReparamDenseReparamWithBNNLossCell。BNN与DNN的最大区别在于,BNN层的weight和bias不再是确定的值,而是服从一个分布。其中,NormalPriorNormalPosterior分别用来生成服从正态分布的先验分布和后验分布;ConvReparamDenseReparam分别是使用reparameteration方法实现的贝叶斯卷积层和全连接层;WithBNNLossCell是用来封装BNN和损失函数的。

如何使用nn.probability.bnn_layers中的API构建贝叶斯神经网络并实现图片分类,可以参考教程使用贝叶斯网络

贝叶斯转换

对于不熟悉贝叶斯模型的研究人员,MDP提供了贝叶斯转换接口(mindspore.nn.probability.transform),支持DNN (Deep Neural Network)模型一键转换成BNN (Bayesian Neural Network)模型。

其中的模型转换APITransformToBNN__init__函数定义如下:

class TransformToBNN:
    def __init__(self, trainable_dnn, dnn_factor=1, bnn_factor=1):
        net_with_loss = trainable_dnn.network
        self.optimizer = trainable_dnn.optimizer
        self.backbone = net_with_loss.backbone_network
        self.loss_fn = getattr(net_with_loss, "_loss_fn")
        self.dnn_factor = dnn_factor
        self.bnn_factor = bnn_factor
        self.bnn_loss_file = None

参数trainable_bnn是经过TrainOneStepCell包装的可训练DNN模型,dnn_factorbnn_factor分别为由损失函数计算得到的网络整体损失的系数和每个贝叶斯层的KL散度的系数。 APITransformToBNN主要实现了两个功能:

  • 功能一:转换整个模型

    transform_to_bnn_model方法可以将整个DNN模型转换为BNN模型。其定义如下:

      def transform_to_bnn_model(self,
                                 get_dense_args=lambda dp: {"in_channels": dp.in_channels, "has_bias": dp.has_bias,
                                                            "out_channels": dp.out_channels, "activation": dp.activation},
                                 get_conv_args=lambda dp: {"in_channels": dp.in_channels, "out_channels": dp.out_channels,
                                                           "pad_mode": dp.pad_mode, "kernel_size": dp.kernel_size,
                                                           "stride": dp.stride, "has_bias": dp.has_bias,
                                                           "padding": dp.padding, "dilation": dp.dilation,
                                                           "group": dp.group},
                                 add_dense_args=None,
                                 add_conv_args=None):
          r"""
          Transform the whole DNN model to BNN model, and wrap BNN model by TrainOneStepCell.
    
          Args:
              get_dense_args (function): The arguments gotten from the DNN full connection layer. Default: lambda dp:
                  {"in_channels": dp.in_channels, "out_channels": dp.out_channels, "has_bias": dp.has_bias}.
              get_conv_args (function): The arguments gotten from the DNN convolutional layer. Default: lambda dp:
                  {"in_channels": dp.in_channels, "out_channels": dp.out_channels, "pad_mode": dp.pad_mode,
                  "kernel_size": dp.kernel_size, "stride": dp.stride, "has_bias": dp.has_bias}.
              add_dense_args (dict): The new arguments added to BNN full connection layer. Default: {}.
              add_conv_args (dict): The new arguments added to BNN convolutional layer. Default: {}.
    
          Returns:
              Cell, a trainable BNN model wrapped by TrainOneStepCell.
         """
    
    

    参数get_dense_args指定从DNN模型的全连接层中获取哪些参数,默认值是DNN模型的全连接层和BNN的全连接层所共有的参数,参数具体的含义可以参考API说明文档get_conv_args指定从DNN模型的卷积层中获取哪些参数,默认值是DNN模型的卷积层和BNN的卷积层所共有的参数,参数具体的含义可以参考API说明文档;参数add_dense_argsadd_conv_args分别指定了要为BNN层指定哪些新的参数值。需要注意的是,add_dense_args中的参数不能与get_dense_args重复,add_conv_argsget_conv_args也是如此。

  • 功能二:转换指定类型的层

    transform_to_bnn_layer方法可以将DNN模型中指定类型的层(nn.Dense或者nn.Conv2d)转换为对应的贝叶斯层。其定义如下:

     def transform_to_bnn_layer(self, dnn_layer, bnn_layer, get_args=None, add_args=None):
          r"""
          Transform a specific type of layers in DNN model to corresponding BNN layer.
    
          Args:
              dnn_layer_type (Cell): The type of DNN layer to be transformed to BNN layer. The optional values are
              nn.Dense, nn.Conv2d.
              bnn_layer_type (Cell): The type of BNN layer to be transformed to. The optional values are
                  DenseReparameterization, ConvReparameterization.
              get_args (dict): The arguments gotten from the DNN layer. Default: None.
              add_args (dict): The new arguments added to BNN layer. Default: None.
    
          Returns:
              Cell, a trainable model wrapped by TrainOneStepCell, whose sprcific type of layer is transformed to the corresponding bayesian layer.
          """
    

    参数dnn_layer指定将哪个类型的DNN层转换成BNN层,bnn_layer指定DNN层将转换成哪个类型的BNN层,get_argsadd_args分别指定从DNN层中获取哪些参数和要为BNN层的哪些参数重新赋值。

如何在MindSpore中使用APITransformToBNN可以参考教程DNN一键转换成BNN

贝叶斯工具箱

贝叶斯神经网络的优势之一就是可以获取不确定性,MDP在上层提供了不确定性估计的工具箱(mindspore.nn.probability.toolbox),用户可以很方便地使用该工具箱计算不确定性。不确定性意味着深度学习模型对预测结果的不确定程度。目前,大多数深度学习算法只能给出高置信度的预测结果,而不能判断预测结果的确定性,不确定性主要有两种类型:偶然不确定性和认知不确定性。

  • 偶然不确定性(Aleatoric Uncertainty):描述数据中的内在噪声,即无法避免的误差,这个现象不能通过增加采样数据来削弱。

  • 认知不确定性(Epistemic Uncertainty):模型自身对输入数据的估计可能因为训练不佳、训练数据不够等原因而不准确,可以通过增加训练数据等方式来缓解。

不确定性评估工具箱的接口如下:

  • model:待评估不确定性的已训练好的模型。

  • train_dataset:用于训练的数据集,迭代器类型。

  • task_type:模型的类型,字符串,输入“regression”或者“classification”。

  • num_classes:如果是分类模型,需要指定类别的标签数量。

  • epochs:用于训练不确定模型的迭代数。

  • epi_uncer_model_path:用于存储或加载计算认知不确定性的模型的路径。

  • ale_uncer_model_path:用于存储或加载计算偶然不确定性的模型的路径。

  • save_model:布尔类型,是否需要存储模型。

在使用前,需要先训练好模型,以LeNet5为例,使用方式如下:

from mindspore.nn.probability.toolbox.uncertainty_evaluation import UncertaintyEvaluation
from mindspore.train.serialization import load_checkpoint, load_param_into_net

if __name__ == '__main__':
    # get trained model
    network = LeNet5()
    param_dict = load_checkpoint('checkpoint_lenet.ckpt')
    load_param_into_net(network, param_dict)
    # get train and eval dataset
    ds_train = create_dataset('workspace/mnist/train')
    ds_eval = create_dataset('workspace/mnist/test')
    evaluation = UncertaintyEvaluation(model=network,
                                       train_dataset=ds_train,
                                       task_type='classification',
                                       num_classes=10,
                                       epochs=1,
                                       epi_uncer_model_path=None,
                                       ale_uncer_model_path=None,
                                       save_model=False)
    for eval_data in ds_eval.create_dict_iterator():
        eval_data = Tensor(eval_data['image'], mstype.float32)
        epistemic_uncertainty = evaluation.eval_epistemic_uncertainty(eval_data)
        aleatoric_uncertainty = evaluation.eval_aleatoric_uncertainty(eval_data)
    print('The shape of epistemic uncertainty is ', epistemic_uncertainty.shape)
    print('The shape of epistemic uncertainty is ', aleatoric_uncertainty.shape)

eval_epistemic_uncertainty计算的是认知不确定性,也叫模型不确定性,对于每一个样本的每个预测标签都会有一个不确定值;eval_aleatoric_uncertainty计算的是偶然不确定性,也叫数据不确定性,对于每一个样本都会有一个不确定值。 所以输出为:

The shape of epistemic uncertainty is (32, 10)
The shape of epistemic uncertainty is (32,)

uncertainty的值位于[0,1]之间,越大表示不确定性越高。