详解MindSpore自定义模型损失函数
作者：DECHIN 发布时间：2022-07-24 09:41:34
一、技术背景
损失函数是机器学习中直接决定训练结果好坏的一个模块，该函数用于定义计算出来的结果或者是神经网络给出的推测结论与正确结果的偏差程度，偏差的越多，就表明对应的参数越差。而损失函数的另一个重要性在于会影响到优化函数的收敛性，如果损失函数的指数定义的太高，稍有参数波动就导致结果的巨 * 动的话，那么训练和优化就很难收敛。一般我们常用的损失函数是MSE（均方误差）和MAE（平均标准差）等。那么这里我们尝试在MindSpore中去自定义一些损失函数，可用于适应自己的特殊场景。
二、MindSpore内置的损失函数
刚才提到的MSE和MAE等常见损失函数，MindSpore中是有内置的，通过net_loss = nn.loss.MSELoss()即可调用，再传入Model中进行训练，具体使用方法可以参考如下拟合一个非线性函数的案例：
# test_nonlinear.py
from mindspore import context
import numpy as np
from mindspore import dataset as ds
from mindspore import nn, Tensor, Model
import time
from mindspore.train.callback import Callback, LossMonitor
import mindspore as ms
ms.common.set_seed(0)
def get_data(num, a=2.0, b=3.0, c=5.0):
for _ in range(num):
x = np.random.uniform(-1.0, 1.0)
y = np.random.uniform(-1.0, 1.0)
noise = np.random.normal(0, 0.03)
z = a * x ** 2 + b * y ** 3 + c + noise
yield np.array([[x**2], [y**3]],dtype=np.float32).reshape(1,2), np.array([z]).astype(np.float32)
def create_dataset(num_data, batch_size=16, repeat_size=1):
input_data = ds.GeneratorDataset(list(get_data(num_data)), column_names=['xy','z'])
input_data = input_data.batch(batch_size)
input_data = input_data.repeat(repeat_size)
return input_data
data_number = 160
batch_number = 10
repeat_number = 10
ds_train = create_dataset(data_number, batch_size=batch_number, repeat_size=repeat_number)
class LinearNet(nn.Cell):
def __init__(self):
super(LinearNet, self).__init__()
self.fc = nn.Dense(2, 1, 0.02, 0.02)
def construct(self, x):
x = self.fc(x)
return x
start_time = time.time()
net = LinearNet()
model_params = net.trainable_params()
print ('Param Shape is: {}'.format(len(model_params)))
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
net_loss = nn.loss.MSELoss()
optim = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.6)
model = Model(net, net_loss, optim)
epoch = 1
model.train(epoch, ds_train, callbacks=[LossMonitor(10)], dataset_sink_mode=True)
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
print ('The total time cost is: {}s'.format(time.time() - start_time))
训练的结果如下：
epoch: 1 step: 160, loss is 2.5267093
Parameter (name=fc.weight, shape=(1, 2), dtype=Float32, requires_grad=True) [[1.0694231 0.12706374]]
Parameter (name=fc.bias, shape=(1,), dtype=Float32, requires_grad=True) [5.186701]
The total time cost is: 8.412306308746338s
最终优化出来的loss值是2.5，不过在损失函数定义不同的情况下，单纯只看loss值是没有意义的。所以通常是大家统一定一个测试的标准，比如大家都用MAE来衡量最终训练出来的模型的好坏，但是中间训练的过程不一定采用MAE来作为损失函数。
三、自定义损失函数
由于python语言的灵活性，使得我们可以继承基本类和函数，只要使用mindspore允许范围内的算子，就可以实现自定义的损失函数。我们先看一个简单的案例，暂时将我们自定义的损失函数命名为L1Loss：
# test_nonlinear.py
from mindspore import context
import numpy as np
from mindspore import dataset as ds
from mindspore import nn, Tensor, Model
import time
from mindspore.train.callback import Callback, LossMonitor
import mindspore as ms
import mindspore.ops as ops
from mindspore.nn.loss.loss import Loss
ms.common.set_seed(0)
def get_data(num, a=2.0, b=3.0, c=5.0):
for _ in range(num):
x = np.random.uniform(-1.0, 1.0)
y = np.random.uniform(-1.0, 1.0)
noise = np.random.normal(0, 0.03)
z = a * x ** 2 + b * y ** 3 + c + noise
yield np.array([[x**2], [y**3]],dtype=np.float32).reshape(1,2), np.array([z]).astype(np.float32)
def create_dataset(num_data, batch_size=16, repeat_size=1):
input_data = ds.GeneratorDataset(list(get_data(num_data)), column_names=['xy','z'])
input_data = input_data.batch(batch_size)
input_data = input_data.repeat(repeat_size)
return input_data
data_number = 160
batch_number = 10
repeat_number = 10
ds_train = create_dataset(data_number, batch_size=batch_number, repeat_size=repeat_number)
class LinearNet(nn.Cell):
def __init__(self):
super(LinearNet, self).__init__()
self.fc = nn.Dense(2, 1, 0.02, 0.02)
def construct(self, x):
x = self.fc(x)
return x
start_time = time.time()
net = LinearNet()
model_params = net.trainable_params()
print ('Param Shape is: {}'.format(len(model_params)))
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
class L1Loss(Loss):
def __init__(self, reduction="mean"):
super(L1Loss, self).__init__(reduction)
self.abs = ops.Abs()
def construct(self, base, target):
x = self.abs(base - target)
return self.get_loss(x)
user_loss = L1Loss()
optim = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.6)
model = Model(net, user_loss, optim)
epoch = 1
model.train(epoch, ds_train, callbacks=[LossMonitor(10)], dataset_sink_mode=True)
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
print ('The total time cost is: {}s'.format(time.time() - start_time))
这里自己定义的内容实际上有两个部分，一个是construct函数中的计算结果的函数，比如这里使用的是求绝对值。另外一个定义的部分是reduction参数，我们从mindspore的源码中可以看到，这个reduction函数可以决定调用哪一种计算方法，定义好的有平均值、求和、保持不变三种策略。
那么最后看下自定义的这个损失函数的运行结果：
epoch: 1 step: 160, loss is 1.8300734
Parameter (name=fc.weight, shape=(1, 2), dtype=Float32, requires_grad=True) [[ 1.2687287 -0.09565887]]
Parameter (name=fc.bias, shape=(1,), dtype=Float32, requires_grad=True) [3.7297544]
The total time cost is: 7.0749146938323975s
这里不必太在乎loss的值，因为前面也提到了，不同的损失函数框架下，计算出来的值就是不一样的，小一点大一点并没有太大意义，最终还是需要大家统一一个标准才能够进行很好的衡量和对比。
四、自定义其他算子
这里我们仅仅是替换了一个abs的算子为square的算子，从求绝对值变化到求均方误差，这里只是修改了一个算子，内容较为简单：
# test_nonlinear.py
from mindspore import context
import numpy as np
from mindspore import dataset as ds
from mindspore import nn, Tensor, Model
import time
from mindspore.train.callback import Callback, LossMonitor
import mindspore as ms
import mindspore.ops as ops
from mindspore.nn.loss.loss import Loss
ms.common.set_seed(0)
def get_data(num, a=2.0, b=3.0, c=5.0):
for _ in range(num):
x = np.random.uniform(-1.0, 1.0)
y = np.random.uniform(-1.0, 1.0)
noise = np.random.normal(0, 0.03)
z = a * x ** 2 + b * y ** 3 + c + noise
yield np.array([[x**2], [y**3]],dtype=np.float32).reshape(1,2), np.array([z]).astype(np.float32)
def create_dataset(num_data, batch_size=16, repeat_size=1):
input_data = ds.GeneratorDataset(list(get_data(num_data)), column_names=['xy','z'])
input_data = input_data.batch(batch_size)
input_data = input_data.repeat(repeat_size)
return input_data
data_number = 160
batch_number = 10
repeat_number = 10
ds_train = create_dataset(data_number, batch_size=batch_number, repeat_size=repeat_number)
class LinearNet(nn.Cell):
def __init__(self):
super(LinearNet, self).__init__()
self.fc = nn.Dense(2, 1, 0.02, 0.02)
def construct(self, x):
x = self.fc(x)
return x
start_time = time.time()
net = LinearNet()
model_params = net.trainable_params()
print ('Param Shape is: {}'.format(len(model_params)))
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
class L1Loss(Loss):
def __init__(self, reduction="mean"):
super(L1Loss, self).__init__(reduction)
self.square = ops.Square()
def construct(self, base, target):
x = self.square(base - target)
return self.get_loss(x)
user_loss = L1Loss()
optim = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.6)
model = Model(net, user_loss, optim)
epoch = 1
model.train(epoch, ds_train, callbacks=[LossMonitor(10)], dataset_sink_mode=True)
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
print ('The total time cost is: {}s'.format(time.time() - start_time))
关于更多的算子内容，可以参考下这个链接
(https://www.mindspore.cn/doc/api_python/zh-CN/r1.2/mindspore/mindspore.ops.html)中的内容，
上述代码的运行结果如下：
epoch: 1 step: 160, loss is 2.5267093
Parameter (name=fc.weight, shape=(1, 2), dtype=Float32, requires_grad=True) [[1.0694231 0.12706374]]
Parameter (name=fc.bias, shape=(1,), dtype=Float32, requires_grad=True) [5.186701]
The total time cost is: 6.87545919418335s
可以从这个结果中发现的是，计算出来的结果跟最开始使用的内置的MSELoss结果是一样的，这是因为我们自定义的这个求损失函数的形式与内置的MSE是吻合的。
五、多层算子的应用
上面的两个例子都是简单的说明了一下通过单个算子构造的损失函数，其实如果是一个复杂的损失函数，也可以通过多个算子的组合操作来进行实现：
# test_nonlinear.py
from mindspore import context
import numpy as np
from mindspore import dataset as ds
from mindspore import nn, Tensor, Model
import time
from mindspore.train.callback import Callback, LossMonitor
import mindspore as ms
import mindspore.ops as ops
from mindspore.nn.loss.loss import Loss
ms.common.set_seed(0)
def get_data(num, a=2.0, b=3.0, c=5.0):
for _ in range(num):
x = np.random.uniform(-1.0, 1.0)
y = np.random.uniform(-1.0, 1.0)
noise = np.random.normal(0, 0.03)
z = a * x ** 2 + b * y ** 3 + c + noise
yield np.array([[x**2], [y**3]],dtype=np.float32).reshape(1,2), np.array([z]).astype(np.float32)
def create_dataset(num_data, batch_size=16, repeat_size=1):
input_data = ds.GeneratorDataset(list(get_data(num_data)), column_names=['xy','z'])
input_data = input_data.batch(batch_size)
input_data = input_data.repeat(repeat_size)
return input_data
data_number = 160
batch_number = 10
repeat_number = 10
ds_train = create_dataset(data_number, batch_size=batch_number, repeat_size=repeat_number)
class LinearNet(nn.Cell):
def __init__(self):
super(LinearNet, self).__init__()
self.fc = nn.Dense(2, 1, 0.02, 0.02)
def construct(self, x):
x = self.fc(x)
return x
start_time = time.time()
net = LinearNet()
model_params = net.trainable_params()
print ('Param Shape is: {}'.format(len(model_params)))
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
class L1Loss(Loss):
def __init__(self, reduction="mean"):
super(L1Loss, self).__init__(reduction)
self.square = ops.Square()
def construct(self, base, target):
x = self.square(self.square(base - target))
return self.get_loss(x)
user_loss = L1Loss()
optim = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.6)
model = Model(net, user_loss, optim)
epoch = 1
model.train(epoch, ds_train, callbacks=[LossMonitor(10)], dataset_sink_mode=True)
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
print ('The total time cost is: {}s'.format(time.time() - start_time))
这里使用的函数是两个平方算子，也就是四次方的均方误差，运行结果如下：
epoch: 1 step: 160, loss is 16.992222
Parameter (name=fc.weight, shape=(1, 2), dtype=Float32, requires_grad=True) [[0.14460069 0.32045612]]
Parameter (name=fc.bias, shape=(1,), dtype=Float32, requires_grad=True) [5.6676607]
The total time cost is: 7.253541946411133s
在实际的运算过程中，我们肯定不能够说提升损失函数的幂次就一定能够提升结果的优劣，但是通过多种基础算子的组合，理论上说我们在一定的误差允许范围内，是可以实现任意的一个损失函数（通过泰勒展开取截断项）的。
六、重定义reduction
方才提到这里面自定义损失函数的两个重点，一个是上面三个章节中所演示的construct函数的重写，这部分实际上是重新设计损失函数的函数表达式。另一个是reduction的自定义，这部分关系到不同的单点损失函数值之间的关系。举个例子来说，如果我们将reduction设置为求和，那么get_loss()这部分的函数内容就是把所有的单点函数值加起来返回一个最终的值，求平均值也是类似的。那么通过自定义一个新的get_loss()函数，我们就可以实现更加灵活的一些操作，比如我们可以选择将所有的结果乘起来求积而不是求和（只是举个例子，大部分情况下不会这么操作）。在python中要重写这个函数也容易，就是在继承父类的自定义类中定义一个同名函数即可，但是注意我们最好是保留原函数中的一些内容，在原内容的基础上加一些东西，冒然改模块有可能导致不好定位的运行报错。
# test_nonlinear.py
from mindspore import context
import numpy as np
from mindspore import dataset as ds
from mindspore import nn, Tensor, Model
import time
from mindspore.train.callback import Callback, LossMonitor
import mindspore as ms
import mindspore.ops as ops
from mindspore.nn.loss.loss import Loss
ms.common.set_seed(0)
def get_data(num, a=2.0, b=3.0, c=5.0):
for _ in range(num):
x = np.random.uniform(-1.0, 1.0)
y = np.random.uniform(-1.0, 1.0)
noise = np.random.normal(0, 0.03)
z = a * x ** 2 + b * y ** 3 + c + noise
yield np.array([[x**2], [y**3]],dtype=np.float32).reshape(1,2), np.array([z]).astype(np.float32)
def create_dataset(num_data, batch_size=16, repeat_size=1):
input_data = ds.GeneratorDataset(list(get_data(num_data)), column_names=['xy','z'])
input_data = input_data.batch(batch_size)
input_data = input_data.repeat(repeat_size)
return input_data
data_number = 160
batch_number = 10
repeat_number = 10
ds_train = create_dataset(data_number, batch_size=batch_number, repeat_size=repeat_number)
class LinearNet(nn.Cell):
def __init__(self):
super(LinearNet, self).__init__()
self.fc = nn.Dense(2, 1, 0.02, 0.02)
def construct(self, x):
x = self.fc(x)
return x
start_time = time.time()
net = LinearNet()
model_params = net.trainable_params()
print ('Param Shape is: {}'.format(len(model_params)))
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
class L1Loss(Loss):
def __init__(self, reduction="mean", config=True):
super(L1Loss, self).__init__(reduction)
self.square = ops.Square()
self.config = config
def construct(self, base, target):
x = self.square(base - target)
return self.get_loss(x)
def get_loss(self, x, weights=1.0):
print ('The data shape of x is: ', x.shape)
input_dtype = x.dtype
x = self.cast(x, ms.common.dtype.float32)
weights = self.cast(weights, ms.common.dtype.float32)
x = self.mul(weights, x)
if self.reduce and self.average:
x = self.reduce_mean(x, self.get_axis(x))
if self.reduce and not self.average:
x = self.reduce_sum(x, self.get_axis(x))
if self.config:
x = self.reduce_mean(x, self.get_axis(x))
weights = self.cast(-1.0, ms.common.dtype.float32)
x = self.mul(weights, x)
x = self.cast(x, input_dtype)
return x
user_loss = L1Loss()
optim = nn.Momentum(net.trainable_params(), learning_rate=0.01, momentum=0.6)
model = Model(net, user_loss, optim)
epoch = 1
model.train(epoch, ds_train, callbacks=[LossMonitor(10)], dataset_sink_mode=True)
for net_param in net.trainable_params():
print(net_param, net_param.asnumpy())
print ('The total time cost is: {}s'.format(time.time() - start_time))
上述代码就是一个简单的案例，这里我们所做的操作，仅仅是把之前均方误差的求和改成了求和之后取负数。还是需要再强调一遍的是，虽然我们定义的函数是非常简单的内容，但是借用这个方法，我们可以更加灵活的去按照自己的设计定义一些定制化的损失函数。上述代码的执行结果如下：
The data shape of x is:
(10, 10, 1)
...
The data shape of x is:
(10, 10, 1)
epoch: 1 step: 160, loss is -310517200.0
Parameter (name=fc.weight, shape=(1, 2), dtype=Float32, requires_grad=True) [[-6154.176 667.4569]]
Parameter (name=fc.bias, shape=(1,), dtype=Float32, requires_grad=True) [-16418.32]
The total time cost is: 6.681089878082275s
一共打印了160个The data shape of x is...，这是因为我们在划分输入的数据集的时候，选择了将160个数据划分为每个batch含10个元素的模块，那么一共就有16个batch，又对这16个batch重复10次，那么就是一共有160个batch，计算损失函数时是以batch为单位的，但是如果只是计算求和或者求平均值的话，不管划分多少个batch结果都是一致的。
来源：https://www.cnblogs.com/dechinphy/p/msloss.html
0
投稿猜你喜欢
1.奇偶校验我们约定一串编码里1的个数是偶数个，那么这串编码里携带的信息就是对的，否则就是错的。我们可以在开头对这串编码加一位校验码实现奇偶- 很多时候，我们都需要获取windows消息提示框的文本内容，例如系统异常信息，软件错误提示等。。。但是如何获取提示信息呢？通常我们的印象中，
- 一.安装环境这里也顺便记录一下如何在windows7上卸载解压版MySQL5.6数据库,如果无需卸载请忽略下一步，直接看第三步即可二.win
- 该代码主要是基于python实现判断指定文件夹下是否存在指定后缀的文件。代码如下：import osYour_Dir='你的文件夹/
- 一、模型参数的保存和加载 torch.save(module.state_dict(), path)：使用module.state
- 本文做的是基于opencv将视频帧转成图片输出，由于一个视频包含的帧数过多，经常我们并不是需要它的全部帧转成图片，因此我们希望可以设置每隔多
- 系统自带模块（库）```cppimport retarget = 'abc1234xyz're.search('(\
- 1.GridView无代码分页排序：效果图：1.AllowSorting设为True，aspx代码中是AllowSorting="
- 要读懂这些代码主要是要了解ASP中操作二进制数据的对象ADODB.Stream！本程序主要用的就是Adodb.Stream，如果你有这个基础
- 前言本文主要给大家介绍了关于Django快速分页的相关内容，分享出来供大家参考学习，下面话不多说了，来一起看看详细的介绍吧。分页在web开发
- keras 模块里面为我们提供了一个预训练好的模型，也就是开箱即可使用的图像识别模型趁着国庆假期有时间我们就来看看这个预训练模型如何使用吧可
- <SPAN style="FONT-SIZE: 14px"><!DOCTYPE html> &l
- JSON(JavaScript Object Notation) 是一种轻量级的数据交换格式。它基于JavaScript（Standard
- 本文实例讲述了RSA实现JS前端加密与PHP后端解密功能。分享给大家供大家参考，具体如下：web前端，用户注册与登录，不能直接以明文形式提交
- k近邻算法的介绍k近邻算法是一种基本的分类和回归方法，这里只实现分类的k近邻算法。k近邻算法的输入为实例的特征向量，对应特征空间的点；输出为
- 这篇文章主要介绍了python操作openpyxl导出Excel 设置单元格格式及合并处理代码实例,文中通过示例代码介绍的非常详细，对大家的
- 判断函数如下：Public Function CheckBIG(strSource As String) As BooleanDim idx
- 最近切换到了Ubuntu的系统作为工作环境, 在使用Pycharm的时候, 出现了个奇怪的问题中文是无法正常输入的, 然后找遍了网上的解决办
- golang中的string是可以转换为byte数组或者rune数组但是其实byte对应的类型是uint8，而rune对应的数据类型就是in
- 疫情数据程序源码// An highlighted blockimport requestsimport jsonclass epidemi
