Chainer - 核心组件

Chainer 是一个多功能的深度学习框架,旨在轻松促进神经网络的开发和训练。Chainer 的 核心组件 为构建复杂模型和执行高效计算提供了坚实的基础。

在 chainer 中,核心组件 Chain 类用于管理网络层和参数,例如用于定义和应用模型操作的 LinksFunctions 以及用于处理数据和梯度的 Variable 类。

此外,Chainer 还集成了强大的 Optimizers 来更新模型参数,Utilities 来管理 xDatasetDataLoader,以及支持灵活模型架构的动态 Computational Graph。所有这些组件共同使 Chainer 成为深度学习任务的综合工具,从而简化了模型创建、训练和优化。

以下是 Chainer 框架的不同核心组件 −

变量

在 Chainer 中,变量 类是一个基本构建块,它表示神经网络训练期间的数据及其相关梯度。变量不仅封装了输入、输出或中间计算等数据,还封装了自动微分所需的信息,这对于反向传播至关重要。

变量的主要特征

以下是 Chainer 框架中变量的主要特征 −

  • 数据存储:变量以多维数组的形式保存数据,通常是 NumPy 或 CuPy 数组,具体取决于计算是在 CPU 还是 GPU 上执行。变量中存储的数据可以是输入数据、输出预测或在网络前向传递期间计算的任何中间值。
  • 梯度存储:在反向传播期间,Chainer 计算每个变量的损失函数梯度。这些梯度存储在变量本身中。变量的 grad 属性包含梯度数据,用于在训练期间更新模型参数。
  • 自动微分:在将操作应用于变量对象时,Chainer 会自动构建计算图。此图通过在反向传递期间高效计算梯度来跟踪操作顺序和变量之间的依赖关系。可以在变量上调用反向方法来触发整个网络中的梯度计算。
  • 设备灵活性:变量通过使用 NumPy 支持 CPU,通过使用 CuPy 数组支持 GPU,从而可以轻松地在设备之间移动计算。对变量的操作会自动适应数据所在的设备。

示例

以下示例展示了如何使用 Chainer 的 Variable 类执行基本操作并通过反向传播计算梯度 −

import chainer
import numpy as np

# 创建包含数据的变量
x = chainer.Variable(np.array([1.0, 2.0, 3.0], dtype=np.float32))

# 对变量执行操作
y = x ** 2 + 2 * x + 1

# 打印结果
print("Result:", y.data) # 输出:[4. 9. 16.]

# 假设 y 为损失并执行反向传播
y.grad = np.ones_like(y.data) # 将 y 的梯度设置为 1 以进行反向传播
y.backward() # 计算梯度

# 打印 x 的梯度
print("x 的梯度:", x.grad) # 输出:[4. 6. 8.]

以下是 chainer 变量类的输出 −

结果:[ 4. 9. 16.]
x 的梯度:[4. 6. 8.]

函数

在 Chainer 中,函数是应用于神经网络内数据的操作。这些函数是基本构建块,在数据流经网络时执行数学运算、激活函数、损失计算和其他转换。

Chainer 在 chainer.functions 模块中提供了广泛的预定义函数,使用户能够轻松构建和自定义神经网络。

Chainer 中的关键函数

激活函数:神经网络中的这些函数通过使模型能够学习数据中的复杂模式,为模型引入非线性。它们应用于每一层的输出,以确定网络的最终输出。以下是 chainer − 中的激活函数

  • ReLU(整流线性单元):如果 ReLU 输出为正,则直接作为输入,否则输出为零。它在神经网络中被广泛使用,因为它有助于缓解消失梯度问题,并且通过使其有效地训练深度模型而具有计算效率。ReLU 的公式为 −

    $$ReLU(x) = max( heta, x)$$

    chainer.functions 模块中的 ReLU 函数为 F.relu(x)

  • sigmoid:此函数将输入映射到 0 到 1 之间的值,使其成为二元分类任务的理想选择。它提供了平滑的梯度,有助于基于梯度的优化,但可能会受到深度网络中消失梯度问题的影响。 sigmoid 的公式为 −

    $$Sigmoid(x)=\frac{1}{1+e^{-x}}$$

    chainer.functions模块中的 Sigmoid 函数为 F.sigmoid(x)

  • Tanh(双曲正切):Chainer 中的此函数用作神经网络中的激活函数。它通过产生零中心输出将输入转换为 -1 和 1 之间的值。此特性在训练期间非常有用,因为它有助于解决与非中心数据相关的问题,从而有可能改善模型的收敛性。 Tanh 的公式为 −

    $$Tanh(x) = \frac{e^x - e^{-x}}{e^x + e^{-x}}$$

    chainer.functions 模块中,我们有函数 F.tanh(x),用于计算 chainer 中的 Tanh。

  • Leaky ReLU:这在神经网络中也称为 Leaky Rectified Linear Unit 函数,是标准 ReLU 激活函数的一种变体。与 ReLU 不同,ReLU 对负输入输出零,而 Leaky ReLU 允许对负输入使用较小的非零梯度。

    这种调整有助于防止"死亡 ReLU"问题,即神经元变得不活跃并停止学习,方法是确保所有神经元继续为模型的学习过程做出贡献。Leaky ReLU 的公式为 −

    $$Leaky Relu(x) = max(\alpha x, x)$$

    其中,$\alpha$ 是一个小常数。chainer.functions 模块具有函数 F.leaky_relu(x),用于计算 chainer 中的 Leaky ReLu。

  • Softmax:这是一种激活函数,通常用于神经网络的输出层,尤其是用于多类分类任务。它将原始预测分数 (logits) 向量转换为概率分布,其中每个概率与相应输入值的指数成比例。

    输出向量中的概率总和为 1,这使得 Softmax 成为表示分类问题中每个类的可能性的理想选择。 Softmax 的公式为 −

    $$Softmax(x_{i})=\frac{e^{x_{i}}}{\sum_{j} e^{x_{j}}}$$

    chainer.functions 模块具有函数 F.softmax(x) 来计算 chainer 中的 Softmax。

示例

以下示例展示了如何在简单的神经网络中使用 Chainer 中的各种激活函数 −

import chainer
import chainer.links as L
import chainer.functions as F
import numpy as np

# 使用 Chainer 的 Chain 类定义一个简单的神经网络
class SimpleNN(chainer.Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
            # 定义层:两个线性层
            self.l1 = L.Linear(4, 3) # 输入层有 4 个特征,隐藏层有 3 个单元
            self.l2 = L.Linear(3, 2) # 隐藏层有 3 个单元,输出层有 2 个单元
    
    def __call__(self, x):
        # 使用不同的激活函数进行前向传递
        
        # 在第一层之后应用 ReLU 激活
        h = F.relu(self.l1(x))
        
        # 在第二层之后应用 Sigmoid 激活
        y = F.sigmoid(self.l2(h))
        
        return y

# 创建具有 4 个特征的样本输入数据
x = np.array([[0.5, -1.2, 3.3, 0.7]], dtype=np.float32)

# 将输入转换为Chainer 的变量
x_var = chainer.Variable(x)

# 实例化神经网络
model = SimpleNN()

# 执行前向传递
output = model(x_var)

# 打印输出
print("应用 ReLU 和 Sigmoid 激活后的网络输出:", output.data)

以下是简单神经网络中使用的激活函数的输出 −

应用 ReLU 和 Sigmoid 激活后的网络输出:[[0.20396319 0.7766712 ]]

Chain 和 ChainList

在 Chainer 中,Chain 和 ChainList 是便于组织和管理神经网络中的层和参数的基本类。Chain 和 ChainList 均派生自 chainer.Link,后者是负责定义模型参数的基类。但它们的用途不同,并且用于不同的场景。让我们详细了解链和链列表,如下所示 −

Chain

Chain 类旨在将神经网络或网络中的模块表示为链接(层)的集合。使用 Chain 时,我们可以通过明确将每个层指定为实例变量来定义网络结构。这种方法对具有固定架构的网络很有帮助。

当我们有一个定义明确、固定的网络架构时,我们可以使用 Chain,我们想要直接访问和组织模型的每个层或组件。

以下是 Chain Class 的主要功能 −

  • 命名组件:通过直接引用网络的特定部分,可以通过名称访问添加到 Chain 的层或链接。
  • 静态架构: Chain 的结构通常在初始化时定义,在训练或推理期间不会动态更改。

示例

以下是展示 Chainer 框架中 Chain 类用法的示例 −

import chainer
import chainer.links as L
import chainer.functions as F

# 使用 Chain 定义一个简单的神经网络
class SimpleChain(chainer.Chain):
    def __init__(self):
        super(SimpleChain, self).__init__()
        with self.init_scope():
        self.l1 = L.Linear(4, 3) # 具有 4 个输入和 3 个输出的线性层
        self.l2 = L.Linear(3, 2) # 具有 3 个输入和 2 个输出的线性层

    def forward(self, x):
        h = F.relu(self.l1(x)) # 在第一层之后应用 ReLU
        y = self.l2(h) # 第二层之后没有激活
        return y

# 实例化模型
model = SimpleChain()
print(model)

以下是上述示例的输出 −

SimpleChain(
  (l1): Linear(in_size=4, out_size=3, nobias=False),
  (l2): Linear(in_size=3, out_size=2, nobias=False),
)

ChainList

ChainList 类与 Chain 类似,但我们不必将每一层定义为实例变量,而是将它们存储在类似列表的结构中。当层数或组件数可能变化或架构动态时,ChainList 非常有用。

当我们的模型层数可变或网络结构可以动态变化时,我们可以使用 ChainList。它也适用于像循环网络这样的架构,其中同一类型的层被多次使用。

以下是 ChainList 的主要功能 −

  • 无序组件:添加到 ChainList 的层或链接通过其索引而不是名称访问。
  • 灵活的架构:它更适合网络结构可能发生变化或在循环或列表中处理层的情况。

示例

以下示例展示了如何在 Chainer Framework 中使用 ChainList 类 −

import chainer
import chainer.links as L
import chainer.functions as F

# 使用 ChainList 定义神经网络
class SimpleChainList(chainer.ChainList):
    def __init__(self):
        super(SimpleChainList, self).__init__(
        L.Linear(4, 3), # 具有 4 个输入和 3 个输出的线性层
        L.Linear(3, 2) # 具有 3 个输入和 2 个输出的线性层
        )
    
    def forward(self, x):
        h = F.relu(self[0](x)) # 在第一层之后应用 ReLU
        y = self[1](h) # 第二层之后没有激活
        return y

# 实例化模型
model = SimpleChainList()
print(model)

以下是在 Chainer Framework 中使用 ChainList 类的输出 −

SimpleChainList(
  (0): Linear(in_size=4, out_size=3, nobias=False),
  (1): Linear(in_size=3, out_size=2, nobias=False),
)

优化器

在 Chainer 中,优化器在训练神经网络中起着至关重要的作用,它通过调整模型的参数(例如权重和偏差)来最小化损失函数。

在训练期间,通过反向传播计算损失函数相对于参数的梯度后,优化器使用这些梯度以逐渐减少损失的方式更新参数。

Chainer 提供了各种内置优化器,每个优化器都采用不同的参数更新策略来适应不同类型的模型和任务。以下是 Chainer − 中的关键优化器

SGD(随机梯度下降)

最基本的优化器是 SGD 更新,其中每个参数都沿其负梯度的方向进行更新,并按学习率缩放。它很简单,但收敛速度可能很慢。

通常,它们可用于更简单或更小的模型,或作为与更复杂的优化器进行比较的基线。chainer 中用于计算 SGD 的函数为 chainer.optimizers.SGD

示例

这是一个在 Chainer 中使用 随机梯度下降 (SGD) 训练基本神经网络的简单示例。我们将使用定义神经网络模型的小数据集,然后在训练期间应用 SGD 优化器来更新模型的参数 −

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain
import numpy as np
from chainer import Variable
from chainer import optimizers

class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
        self.fc1 = L.Linear(None, 100) # 具有 100 个单元的完全连接层
        self.fc2 = L.Linear(100, 10) # 具有 10 个单元的输出层(例如,针对 10 个类)
    
    def forward(self, x):
        h = F.relu(self.fc1(x)) # 应用 ReLU 激活函数
        return self.fc2(h) # 输出层

# 虚拟数据:5 个样本,每个样本有 50 个特征
x_data = np.random.rand(5, 50).astype(np.float32)
# 虚拟标签:5 个样本,每个样本有 10 个类(独热编码)
y_data = np.random.randint(0, 10, 5).astype(np.int32)

# 转换为 Chainer 变量
x = Variable(x_data)
y = Variable(y_data)

# 初始化模型
model = SimpleNN()

# 设置 SGD 优化器,学习率为 0.01
optimizer = optimizers.SGD(lr=0.01)
optimizer.setup(model)
def loss_func(predictions, target):
return F.softmax_cross_entropy(predictions, target)
# 训练循环
for epoch in range(10): # epoch 数
    # 将梯度归零
    model.cleargrads()
    
    # 前向传递
    predictions = model(x)
    
    # 计算损失
    loss = loss_func(predictions, y)
    
    # 后向传递
    loss.backward()
    
    # 更新参数
    optimizer.update()

# 打印损失
print(f'Epoch {epoch + 1}, Loss: {loss.data}')

以下是 SGD 优化器的输出 −

Epoch 1, Loss: 2.3100974559783936
Epoch 2, Loss: 2.233552932739258
Epoch 3, Loss: 2.1598660945892334
Epoch 4, Loss: 2.0888497829437256
Epoch 5, Loss: 2.020642042160034
Epoch 6, Loss: 1.9552147388458252
Epoch 7, Loss: 1.8926388025283813
Epoch 8, Loss: 1.8325523138046265
Epoch 9, Loss: 1.7749309539794922
Epoch 10, Loss: 1.7194255590438843

动量 SGD

动量 SGD 是 SGD 的扩展,它包含动量,有助于加速梯度向量在正确方向上的收敛速度。它在梯度方向上累积速度向量。

这适用于 vanilla SGD 难以收敛的模型。我们有一个名为 chainer.optimizers.MomentumSGD 的函数来执行动量 SGD 优化。

动量项:将前一个梯度更新的一小部分添加到当前更新中。这有助于在正确的方向上加速梯度向量并抑制振荡。

公式:具有动量的参数 θ 的更新规则为 −

$$v_{t} = \beta v_{t-1} + (1 - \beta) abla L( heta)$$ $$ heta = heta-\alpha v_{t}$$

其中 −

  • $v_{t}$ 是速度(或累积梯度)
  • $\beta$ 是动量系数(通常在 0.9 左右)
  • $\alpha$ 是学习率
  • $ abla L( heta)$ 是损失函数相对于参数。

示例

这是一个基本示例,说明如何在 Chainer 中将 Momentum SGD 优化器与简单的神经网络结合使用 −

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain
from chainer import optimizers
import numpy as np
from chainer import Variable

class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
            self.fc1 = L.Linear(None, 100) # 具有 100 个单元的完全连接层
            self.fc2 = L.Linear(100, 10) # 具有 10 个单元的输出层(例如,针对 10 个类)
    
    def forward(self, x):
        h = F.relu(self.fc1(x)) # 应用 ReLU 激活函数
        return self.fc2(h) # 输出层

# 虚拟数据:5 个样本,每个样本有 50 个特征
x_data = np.random.rand(5, 50).astype(np.float32)
# 虚拟标签:5 个样本,每个样本有 10 个类(独热编码)
y_data = np.random.randint(0, 10, 5).astype(np.int32)

# 转换为 Chainer 变量
x = Variable(x_data)
y = Variable(y_data)

# 初始化模型
model = SimpleNN()

# 设置 Momentum SGD 优化器,学习率为 0.01,动量为 0.9
optimizer = optimizers.MomentumSGD(lr=0.01, momentum=0.9)
optimizer.setup(model)
def loss_func(predictions, targets):
   return F.softmax_cross_entropy(predictions, targets)
# 训练循环
for epoch in range(10): # epoch 数
    # 将梯度归零
    model.cleargrads()
    
    # 前向传递
    predictions = model(x)
    
    # 计算损失
    loss = loss_func(predictions, y)
    
    # 后向传递
    loss.backward()
    
    # 更新参数
    optimizer.update()
    
    # 打印损失
    print(f'Epoch {epoch + 1}, Loss: {loss.data}')

以下是 Momentum SGD 优化器的输出 −

Epoch 1, Loss: 2.4459869861602783
Epoch 2, Loss: 2.4109833240509033
Epoch 3, Loss: 2.346194267272949
Epoch 4, Loss: 2.25825572013855
Epoch 5, Loss: 2.153470754623413
Epoch 6, Loss: 2.0379838943481445
Epoch 7, Loss: 1.9174035787582397
Epoch 8, Loss: 1.7961997985839844
Epoch 9, Loss: 1.677260398864746
Epoch 10, Loss: 1.5634090900421143

Adam

Adam 优化器结合了 SGD 的另外两个扩展的优点,即 AdaGrad(适用于稀疏梯度)和 RMSProp(适用于非平稳设置)。Adam 保持梯度及其平方的移动平均值,并根据这些平均值更新参数。

由于其在各种任务和模型中都具有稳健性和效率,因此通常将其用作默认优化器。在 chainer 中,我们有函数 chainer.optimizers.Adam 来执行 Adam 优化。

Adam 优化器的主要功能如下 −

  • 自适应学习率:Adam 动态调整每个参数的学习率,使其在各种任务中都有效。
  • 梯度矩:它计算梯度的一阶矩(均值)和二阶矩(非中心方差)以改进优化。
  • 偏差校正:Adam 使用偏差校正来解决初始化期间引入的偏差,特别是在训练早期。
  • 公式:Adam 优化的公式为 −

    $$m_t = \beta_1 m_{t-1} + (1 - \beta_1) abla L( heta)$$ $$v_t = \beta_2 v_{t-1} + (1 - \beta_2) ( abla L( heta))^2$$ $$\hat{m}_t = \frac{m_t}{1 - \beta_1^t}$$ $$\hat{v}_t = \frac{v_t}{1 - \beta_2^t}$$ $$ heta = heta - \frac{\alpha\hat{m}_t}{\sqrt{\hat{v}_t} + \epsilon}$$

    其中,$\alpha$ 是学习率,$\beta 1$$\beta 2$ 是梯度及其平方的移动平均值的衰减率,通常分别为 0.9 和 0.999, ${m_t}$ 和 ${v_t}$ 是一阶矩和二阶矩估计值,$\epsilon$ 是为了数值稳定性而添加的小常数。

示例

以下示例展示了如何在带有神经网络的 chainer 中使用 Adam Optimizer −

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain
from chainer import optimizers
import numpy as np
from chainer import Variable

class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
        self.fc1 = L.Linear(None, 100) # 具有 100 个单元的完全连接层
        self.fc2 = L.Linear(100, 10) # 具有 10 个单元的输出层(例如,针对 10 个类)
    
    def forward(self, x):
        h = F.relu(self.fc1(x)) # 应用 ReLU 激活函数
        return self.fc2(h) # 输出层

# 虚拟数据:5 个样本,每个样本有 50 个特征
x_data = np.random.rand(5, 50).astype(np.float32)
# 虚拟标签:5 个样本,每个样本有 10 个类(独热编码)
y_data = np.random.randint(0, 10, 5).astype(np.int32)

# 转换为 Chainer 变量
x = Variable(x_data)
y = Variable(y_data)

# 初始化模型
model = SimpleNN()

# 使用默认参数设置 Adam 优化器
optimizer = optimizers.Adam()
optimizer.setup(model)
def loss_func(predictions, target):
return F.softmax_cross_entropy(predictions, target)
# 训练循环
for epoch in range(10): # epoch 数
    # 将梯度归零
    model.cleargrads()
    
    # 前向传递
    predictions = model(x)
    
    # 计算损失
    loss = loss_func(predictions, y)
    
    # 后向传递
    loss.backward()
    
    # 更新参数
    optimizer.update()

    # 打印损失
    print(f'Epoch {epoch + 1}, Loss: {loss.data}')

以下是将 Adam 优化器 应用于神经网络的输出 −

Epoch 1, Loss: 2.4677982330322266
Epoch 2, Loss: 2.365001678466797
Epoch 3, Loss: 2.2655398845672607
Epoch 4, Loss: 2.1715924739837646
Epoch 5, Loss: 2.082294464111328
Epoch 6, Loss: 1.9973262548446655
Epoch 7, Loss: 1.9164447784423828
Epoch 8, Loss: 1.8396313190460205
Epoch 9, Loss: 1.7676666975021362
Epoch 10, Loss: 1.7006778717041016

AdaGrad

AdaGrad 也称为 自适应梯度算法,它是一种优化算法,可根据训练期间累积的梯度历史调整每个参数的学习率。它对于稀疏数据和特征频率或重要性变化的场景特别有效。

这适用于稀疏数据问题以及处理某些参数需要比其他参数更多调整的模型。函数 chainer.optimizers.AdaGrad 用于在 Chainer 中执行 AdaGrad 优化。

以下是 AdaGrad Optimizer 的主要功能 −

  • 自适应学习率:AdaGrad 根据梯度平方的累积和单独调整每个参数的学习率。这会导致不频繁参数的更新较大,而频繁参数的更新较小。
  • 无需调整学习率: AdaGrad 会自动调整学习率,通常无需手动调整。

公式: AdaGrad 的公式如下 −

$$g_t = abla L( heta)$$ $$G_t = G_{t-1} +{g_t}^2$$ $$ heta = heta - \frac{\alpha}{\sqrt{G_t} + \epsilon} g_t$$

其中 −

  • $g_t$ 是时间步长 $t$ 的梯度。
  • $G_t$ 是平方的累积和梯度直到时间 $t$。
  • $\alpha$ 是全局学习率。
  • $\epsilon$ 是一个小常数,用于防止除以零。

示例

以下是如何在 Chainer 中使用 AdaGrad 优化器和简单神经网络的示例 −

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain
from chainer import optimizers
import numpy as np
from chainer import Variable

class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
        self.fc1 = L.Linear(None, 100) # 具有 100 个单元的完全连接层
        self.fc2 = L.Linear(100, 10) # 具有 10 个单元的输出层(例如,针对 10 个类)
    
    def forward(self, x):
        h = F.relu(self.fc1(x)) # 应用 ReLU 激活函数
        return self.fc2(h) # 输出层

# 虚拟数据:5 个样本,每个样本有 50 个特征
x_data = np.random.rand(5, 50).astype(np.float32)
# 虚拟标签:5 个样本,每个样本有 10 个类(独热编码)
y_data = np.random.randint(0, 10, 5).astype(np.int32)

# 转换为 Chainer 变量
x = Variable(x_data)
y = Variable(y_data)

# 初始化模型
model = SimpleNN()

# 设置 AdaGrad 优化器,学习率为 0.01
optimizer = optimizers.AdaGrad(lr=0.01)
optimizer.setup(model)
def loss_func(predictions, target):
return F.softmax_cross_entropy(predictions, target)
# 训练循环
for epoch in range(10): # epoch 数
    # 将梯度归零
    model.cleargrads()
    
    # 前向传递
    predictions = model(x)
    
    # 计算损失
    loss = loss_func(predictions, y)
    
    # 后向传递
    loss.backward()
    
    # 更新参数
    optimizer.update()
    
    # 打印损失
    print(f'Epoch {epoch + 1}, Loss: {loss.data}')

以下是将 AdaGrad 优化器 应用于神经网络的输出 −

Epoch 1, Loss: 2.2596702575683594
Epoch 2, Loss: 1.7732301950454712
Epoch 3, Loss: 1.4647505283355713
Epoch 4, Loss: 1.2398217916488647
Epoch 5, Loss: 1.0716438293457031
Epoch 6, Loss: 0.9412426352500916
Epoch 7, Loss: 0.8350374102592468
Epoch 8, Loss: 0.7446572780609131
Epoch 9, Loss: 0.6654194593429565
Epoch 10, Loss: 0.59764164686203

RMSProp

RMSProp 优化器在 AdaGrad 的基础上进行了改进,通过引入梯度平方和的衰减因子来防止学习率过度缩小。它在循环神经网络或需要快速适应不同梯度尺度的模型中特别有效。

在 Chainer 中,为了执行 RMSProp 优化器,我们有函数 chainer.optimizers.RMSprop

以下是 RMSProp 优化器的主要功能 −

  • 衰减因子:RMSProp 通过防止学习率变得太小并允许更稳定的收敛,为累积的平方梯度和引入了衰减因子。
  • 自适应学习率:与 AdaGrad 一样,RMSProp 优化器根据梯度历史单独调整每个参数的学习率,但它通过限制过去平方梯度的累积来避免学习率递减问题。

公式:RMSProp 优化器的公式如下−

$$g_t = abla L( heta)$$ $$E[g^2]_t = \gamma E[g^2]_{t-1} + (1 - \gamma){g_t}^2$$ $$ heta = heta - \frac{\alpha}{\sqrt{E[g^2]_t} + \epsilon} g_t$$

其中 −

  • $g_t$ 是时间步长 $t$ 的梯度。
  • $E[g_2]$ 是梯度平方的移动平均值。
  • $\gamma$ 是衰减因子,通常在 0.9 左右。
  • $\alpha$ 是全局学习率。
  • $\epsilon$ 是一个小常数,用于防止除以零。

示例

以下示例展示了如何在 Chainer 中使用 RMSProp 优化器和简单的神经网络 −

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain
import numpy as np
from chainer import Variable
from chainer import optimizers

class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
        self.fc1 = L.Linear(None, 100) # 具有 100 个单元的完全连接层
        self.fc2 = L.Linear(100, 10) # 具有 10 个单元的输出层(例如,针对 10 个类)
    
    def forward(self, x):
        h = F.relu(self.fc1(x)) # 应用 ReLU 激活函数
        return self.fc2(h) # 输出层

# 虚拟数据:5 个样本,每个样本有 50 个特征
x_data = np.random.rand(5, 50).astype(np.float32)
# 虚拟标签:5 个样本,每个样本有 10 个类(独热编码)
y_data = np.random.randint(0, 10, 5).astype(np.int32)

# 转换为 Chainer 变量
x = Variable(x_data)
y = Variable(y_data)

# 初始化模型
model = SimpleNN()

# 设置 RMSProp 优化器,学习率为 0.01,衰减因子为 0.9
optimizer = optimizers.RMSprop(lr=0.01, alpha=0.9)
optimizer.setup(model)
def loss_func(predictions, target):
return F.softmax_cross_entropy(predictions, target)
# 训练循环
for epoch in range(10): # epoch 数
    # 将梯度归零
    model.cleargrads()
    
    # 前向传递
    predictions = model(x)
    
    # 计算损失
    loss = loss_func(predictions, y)
    
    # 后向传递
    loss.backward()
    
    # 更新参数
    optimizer.update()
    
    # 打印损失
    print(f'Epoch {epoch + 1}, Loss: {loss.data}')

以下是使用 RMSProp 优化的上述示例的输出 −

Epoch 1, Loss: 2.3203792572021484
Epoch 2, Loss: 1.1593462228775024
Epoch 3, Loss: 1.2626817226409912
Epoch 4, Loss: 0.6015896201133728
Epoch 5, Loss: 0.3906801640987396
Epoch 6, Loss: 0.28964582085609436
Epoch 7, Loss: 0.21569299697875977
Epoch 8, Loss: 0.15832018852233887
Epoch 9, Loss: 0.12146510928869247
Epoch 10, Loss: 0.09462013095617294

Chainer 中的数据集和迭代器

在 Chainer 中,高效处理数据对于训练神经网络至关重要。为了实现这一点,Chainer 框架提供了两个基本组件,即数据集迭代器。这些组件有助于管理数据,确保以结构化和高效的方式将数据输入模型。

数据集

Chainer 中的数据集是一组数据样本,可以将其输入神经网络进行训练、验证或测试。Chainer 提供了一个 Dataset 类,可以扩展该类以创建自定义数据集,以及用于常见任务的几个内置数据集类。

Chainer 中的数据集类型

Chainer 提供了几种类型的数据集来处理各种数据格式和结构。这些数据集大致可分为内置数据集、自定义数据集和数据集转换。

内置数据集

Chainer 附带一些常用于基准测试和实验的流行数据集。这些数据集随时可用,可以使用内置函数轻松加载。

以下是获取 Chainer 中所有可用数据集列表的代码 −

import chainer.datasets as datasets

# 获取数据集模块中的所有属性
all_datasets = [attr for attr in dir(datasets) if attr.startswith('get_')]

# 打印可用数据集
print("Chainer 中可用的内置数据集:")
for dataset in all_datasets:
print(f"- {dataset}")

以下是显示 Chainer Framework 中所有内置数据集的输出 −

Built-in datasets available in Chainer:
- get_cifar10
- get_cifar100
- get_cross_validation_datasets
- get_cross_validation_datasets_random
- get_fashion_mnist
- get_fashion_mnist_labels
- get_kuzushiji_mnist
- get_kuzushiji_mnist_labels
- get_mnist
- get_ptb_words
- get_ptb_words_vocabulary
- get_svhn

自定义数据集

使用自定义数据时,我们可以通过子类化 chainer.dataset.DatasetMixin 来创建自己的数据集。这使我们能够定义如何加载和返回数据。

以下是使用 chainer.dataset.DatasetMixin 创建自定义数据集并打印其中第一行的示例 −

import chainer
import numpy as np

class MyDataset(chainer.dataset.DatasetMixin):
   def __init__(self, data, labels):
      self.data = data
      self.labels = labels

   def __len__(self):
      return len(self.data)

   def get_example(self, i):
      return self.data[i], self.labels[i]

# 创建自定义数据集
data = np.random.rand(100, 3)
labels = np.random.randint(0, 2, 100)
dataset = MyDataset(data, labels)
print(dataset[0])

以下是自定义数据集第一行的输出 −

(array([0.82744124, 0.33828446, 0.06409377]), 0)

预处理数据集

Chainer 提供工具来对数据集应用转换,例如缩放、规范化或数据增强。可以使用 TransformDataset 即时应用这些转换。

以下是在 chainer 中使用预处理数据集的示例 −

from chainer.datasets import TransformDataset

def transform(data):
   x, t = data
   x = x / 255.0  # Normalize input data
   return x, t

# 将转换应用于数据集
transformed_dataset = TransformDataset(dataset, transform)
print(transformed_dataset[0])

下面是借助 TransformDataset() 函数预处理的数据集的第一行 −

(array([0.00324487, 0.00132661, 0.00025135]), 0)

连接数据集

ConcatDataset 用于将多个数据集连接成一个数据集。当我们的数据分布在不同来源时,这很有用。以下是在 chainer 框架中使用连接数据集的示例,其中从连接数据集中打印出每个样本的数据和标签。组合数据集包括来自数据集 1 和数据集 2 的所有样本 −

import numpy as np
from chainer.datasets import ConcatenatedDataset
from chainer.dataset import DatasetMixin

# 定义自定义数据集类
class MyDataset(DatasetMixin):
   def __init__(self, data, labels):
      self.data = data
      self.labels = labels

   def __len__(self):
      return len(self.data)

   def get_example(self, i):
      return self.data[i], self.labels[i]

# 样本数据数组
data1 = np.random.rand(5, 3) # 5 个样本,每个样本有 3 个特征
labels1 = np.random.randint(0, 2, 5) # data1 的二进制标签

data2 = np.random.rand(5, 3) # 另外 5 个样本,每个样本有 3 个特征
labels2 = np.random.randint(0, 2, 5) # data2 的二进制标签

# 创建 MyDataset 实例
dataset1 = MyDataset(data1, labels1)
dataset2 = MyDataset(data2, labels2)

# 连接数据集
combined_dataset = ConcatenatedDataset(dataset1, dataset2)

# 遍历组合数据集并打印每个示例
for i in range(len(combined_dataset)):
    data, label = combined_dataset[i]
    print(f"Sample {i+1}: Data = {data}, Label = {label}")

以下是 Chainer 中连接数据集的输出 −

Sample 1: Data = [0.6153635  0.19185915 0.26029754], Label = 1
Sample 2: Data = [0.69201927 0.70393578 0.85382294], Label = 1
Sample 3: Data = [0.46647242 0.37787839 0.37249345], Label = 0
Sample 4: Data = [0.2975833  0.90399536 0.15978975], Label = 1
Sample 5: Data = [0.29939455 0.21290926 0.97327959], Label = 1
Sample 6: Data = [0.68297438 0.64874375 0.09129224], Label = 1
Sample 7: Data = [0.52026288 0.24197601 0.5239313 ], Label = 0
Sample 8: Data = [0.63250008 0.85023346 0.94985447], Label = 1
Sample 9: Data = [0.75183151 0.01774763 0.66343944], Label = 0
Sample 10: Data = [0.60212864 0.48215319 0.02736618], Label = 0

Tuple 和 Dict 数据集

Chainer 提供了称为 TupleDatasetDictDataset 的特殊数据集类,使我们能够方便地管理多个数据源。当我们拥有多种类型的数据(例如特征和标签)或想要一起处理的多个特征集时,这些类非常有用。

  • 元组数据集:这用于将多个数据集或数据数组组合成单个数据集,其中每个元素都是来自原始数据集的相应元素的元组。
  • 以下示例展示了如何在神经网络中使用元组数据集 −

    import numpy as np
    from chainer.datasets import TupleDataset
    
    # 创建两个数据集(或数据数组)
    data1 = np.random.rand(100, 3) # 100 个样本,每个样本有 3 个特征
    data2 = np.random.rand(100, 5) # 100 个样本,每个样本有 5 个特征
    
    # 创建一个结合两个数据的 TupleDataset数组
    tuple_dataset = TupleDataset(data1, data2)
    
    # 从 TupleDataset 访问数据
    for i in range(5):
        print(f"Sample {i+1}: Data1 = {tuple_dataset[i][0]}, Data2 = {tuple_dataset[i][1]}")
    

    以下是 Tuple Dataset 的输出 −

     
    Sample 1: Data1 = [0.32992823 0.57362303 0.95586597], Data2 = [0.41455   0.52850591 0.55602243 0.36316931 0.93588697]
    Sample 2: Data1 = [0.37731994 0.00452533 0.67853069], Data2 = [0.71637691 0.04191565 0.54027323 0.68738626 0.01887967]
    Sample 3: Data1 = [0.85808665 0.15863516 0.51649116], Data2 = [0.9596284  0.12417238 0.22897152 0.63822924 0.99434029]
    Sample 4: Data1 = [0.2477932  0.27937585 0.59660463], Data2 = [0.92666318 0.93611279 0.96622103 0.41834484 0.72602107]
    Sample 5: Data1 = [0.71989544 0.46155552 0.31835487], Data2 = [0.27475741 0.33759694 0.22539997 0.40985004 0.00469414]
    
  • DictDataset: 这也类似于 TupleDataset,但允许我们用一个键标记每个元素,从而更轻松地访问和理解数据。
  • 以下示例展示了如何在 chainer − 中使用 Dict 数据集

    import numpy as np
    from chainer.datasets import DictDataset
    
    # 创建两个数据集(或数据数组)
    data1 = np.random.rand(100, 3) # 100 个样本,每个样本有 3 个特征
    labels = np.random.randint(0, 2, 100) # 每个样本的二进制标签
    
    # 创建 DictDataset
    dict_dataset = DictDataset(data=data1, label=labels)
    
    # 从 DictDataset 访问数据
    for i in range(5):
        print(f"Sample {i+1}: Data = {dict_dataset[i]['data']}, Label = {dict_dataset[i]['label']}")
    

    以下是元组数据集的输出 −

     
    Sample 1: Data = [0.09362018 0.33198328 0.11421714], Label = 1
    Sample 2: Data = [0.53655817 0.9115115  0.0192754 ], Label = 0
    Sample 3: Data = [0.48746879 0.18567869 0.88030764], Label = 0
    Sample 4: Data = [0.10720832 0.79523399 0.56056922], Label = 0
    Sample 5: Data = [0.76360577 0.69915416 0.64604595], Label = 1
    

迭代器

在 Chainer 中,迭代器对于在机器学习模型训练期间管理数据至关重要。它们将大型数据集分解为较小的块,称为小批量,可以逐步处理。这种方法允许模型更频繁地更新其参数,从而提高内存效率并加快训练过程。

Chainer 中的迭代器类型

Chainer 提供各种类型的迭代器来在机器学习模型的训练和评估期间处理数据集。这些迭代器旨在用于不同的场景和要求,例如处理大型数据集、并行数据加载或确保数据混洗以实现更好的泛化。

SerialIterator

这是 Chainer 中最常见的迭代器。它以串行(顺序)方式迭代数据集,提供小批量数据。当到达数据集的末尾时,迭代器可以停止或从头开始,具体取决于重复选项。这非常适合用于保留数据顺序的标准训练。

以下示例展示了如何在 chainer − 中使用 SerialIterator

import chainer
import numpy as np
from chainer import datasets, iterators

# 创建一个简单的数据集(例如,虚拟数据)
x_data = np.random.rand(100, 2).astype(np.float32) # 100 个样本,每个样本有 2 个特征
y_data = np.random.randint(0, 2, size=(100,)).astype(np.int32) # 100 个二进制标签

# 将特征和标签组合成 Chainer 数据集
dataset = datasets.TupleDataset(x_data, y_data)

# 初始化SerialIterator
iterator = iterators.SerialIterator(dataset, batch_size=10, repeat=True, shuffle=True)

# 迭代数据集的示例
for epoch in range(2): # 运行两个 epoch
   	while True:
        batch = iterator.next() # 获取下一个批次
        
        # 手动解压批次
        x_batch = np.array([example[0] for example in batch]) # 提取 x 数据
        y_batch = np.array([example[1] for example in batch]) # 提取 y 数据
        
        print("X batch:", x_batch)
        print("Y batch:", y_batch)
        
        if iterator.is_new_epoch: # 检查新纪元是否已开始
        print("End of epoch")
        break

# 将迭代器重置为数据集的开头(可选)
iterator.reset()

以下是 Chainer 中使用的 SerialIterator 的输出 −

 
X batch: [[0.00603645 0.13716008]
 [0.97394305 0.9035589 ]
 [0.93046355 0.63140464]
 [0.44332692 0.5307854 ]
 [0.48565307 0.845648  ]
 [0.98147005 0.47466147]
 [0.3036461  0.62494874]
 [0.31664708 0.7176309 ]
 [0.14955625 0.65800977]
 [0.72328717 0.33383074]]
Y batch: [1 0 0 1 0 0 1 1 1 0]
----------------------------
----------------------------
----------------------------
X batch: [[0.10038178 0.32700586]
 [0.4653218  0.11713986]
 [0.10589143 0.5662842 ]
 [0.9196327  0.08948212]
 [0.13177629 0.59920484]
 [0.46034923 0.8698121 ]
 [0.24727622 0.8066094 ]
 [0.01744546 0.88371164]
 [0.18966147 0.9189765 ]
 [0.06658458 0.02469426]]
Y batch: [0 1 0 0 0 0 0 0 0 1]
End of epoch

MultiprocessIterator

此迭代器旨在通过使用多个进程来加速数据加载。它在处理大型数据集或数据预处理耗时的情况下特别有用。

以下是在 chainer 框架中使用多处理器迭代器的示例 −

import chainer
import numpy as np
from chainer import datasets, iterators

# 创建一个简单的数据集(例如,虚拟数据)
x_data = np.random.rand(1000, 2).astype(np.float32) # 1000 个样本,每个样本有 2 个特征
y_data = np.random.randint(0, 2, size=(1000,)).astype(np.int32) # 1000 个二进制标签

# 将特征和标签组合成 Chainer 数据集
dataset = datasets.TupleDataset(x_data, y_data)

# 初始化MultiprocessIterator
# n_processes:要使用的工作进程数
iterator = iterators.MultiprocessIterator(dataset, batch_size=32, n_processes=4, repeat=True, shuffle=True)

# 迭代数据集的示例
for epoch in range(2):# 运行两个 epoch
   	while True:
        batch = iterator.next() # 获取下一个批次
        
        # 手动解压批次
        x_batch = np.array([example[0] for example in batch]) # 提取 x 数据
        y_batch = np.array([example[1] for example in batch]) # 提取 y 数据
        
        print("X batch shape:", x_batch.shape)
        print("Y batch shape:", y_batch.shape)
        
        if iterator.is_new_epoch: # 检查新纪元是否已开始
        print("End of epoch")
        break

# 将迭代器重置为数据集的开头(可选)
iterator.reset()

以下是多处理器迭代器的输出 −

X batch shape: (32, 2)
Y batch shape: (32,)
X batch shape: (32, 2)
Y batch shape: (32,)
X batch shape: (32, 2)
Y batch shape: (32,)
---------------------
---------------------
X batch shape: (32, 2)
Y batch shape: (32,)
X batch shape: (32, 2)
Y batch shape: (32,)
End of epoch

MultithreadIterator

MultithreadIterator 是 Chainer 中的一个迭代器,设计用于使用多个线程进行并行数据加载。此迭代器在处理可从并发数据处理中受益的数据集时特别有用,例如当数据加载或预处理是训练中的瓶颈时。

与使用多个进程的 MultiprocessIterator 不同,MultithreadIterator 使用线程,使其更适合需要共享内存访问或轻量级并行性的场景。

以下是在 chainer Framework − 中使用 MultithreadIterator 的示例

import numpy as np
from chainer.datasets import TupleDataset
from chainer.iterators import MultithreadIterator

# 创建样本数据集
data1 = np.random.rand(100, 3) # 100 个样本,每个样本有 3 个特征
data2 = np.random.rand(100, 5) # 100 个样本,每个样本有 5 个特征

# 创建一个TupleDataset
dataset = TupleDataset(data1, data2)

# 创建一个具有 4 个线程且批处理大小为 10 的 MultithreadIterator
iterator = MultithreadIterator(dataset, batch_size=10, n_threads=4, repeat=False, shuffle=True)

# 迭代数据集
for batch in iterator:
# 解压批次中的每个元组
    data_batch_1 = np.array([item[0] for item in batch]) # 从每个元组中提取第一个元素
    data_batch_2 = np.array([item[1] for item in batch]) # 从每个元组中提取第二个元素
    
   	print("Data batch 1:", data_batch_1)
   	print("Data batch 2:", data_batch_2)

以下是多线程迭代器的输出 −

Data batch 1: [[0.38723876 0.66585393 0.74603754]
 [0.136392   0.23425485 0.6053701 ]
 [0.99668734 0.13096871 0.13114792]
 [0.32277508 0.3718192  0.42083016]
 [0.93408236 0.59433832 0.23590596]
 [0.16351005 0.82340571 0.08372471]
 [0.78469682 0.81117013 0.41653794]
 [0.32369538 0.77524528 0.10378537]
 [0.21678887 0.8905319  0.88525376]
 [0.41348068 0.43437296 0.90430938]]
---------------------
---------------------
Data batch 2: [[0.20541319 0.69626397 0.81508325 0.49767042 0.92252953]
 [0.12794664 0.33955336 0.81339754 0.54042266 0.44137714]
 [0.52487615 0.59930116 0.96334436 0.61622956 0.34192033]
 [0.93474439 0.37455884 0.94954379 0.73027705 0.24333167]
 [0.24805745 0.80921792 0.91316062 0.59701139 0.25295744]
 [0.27026875 0.67836862 0.16911597 0.50452568 0.86257208]
 [0.81722752 0.41361153 0.43188091 0.98313524 0.28605503]
 [0.50885091 0.80546812 0.89346966 0.63828489 0.8231125 ]
 [0.78996715 0.05338346 0.16573956 0.89421364 0.54267903]
 [0.05804313 0.5613496  0.09146587 0.79961318 0.02466306]]

ShuffleOrderSampler

ShuffleOrderSampler 是 Chainer 中的一个组件,用于随机化数据集中索引的顺序。它确保每个训练阶段都以不同的顺序查看数据,这有助于减少过度拟合并提高模型的泛化能力。

import numpy as np
from chainer.datasets import TupleDataset
from chainer.iterators import SerialIterator, ShuffleOrderSampler

# 创建样本数据集
data = np.random.rand(100, 3) # 100 个样本,每个样本有 3 个特征
labels = np.random.randint(0, 2, size=100) # 100 个二进制标签

# 创建 TupleDataset
dataset = TupleDataset(data, labels)

# 初始化 ShuffleOrderSampler
sampler = ShuffleOrderSampler()

# 使用 ShuffleOrderSampler 创建 SerialIterator
iterator = SerialIterator(dataset, batch_size=10, repeat=False, order_sampler=sampler)

# 迭代数据集
for batch in iterator:
    # 由于批次包含元组,我们分别提取数据和标签
    data_batch, label_batch = zip(*batch)
 	print("Data batch:", np.array(data_batch))
   	print("Label batch:", np.array(label_batch))

以下是在 Chainer 中应用 ShuffleOrderSampler 迭代器的输出 −

Data batch: [[0.93062607 0.68334939 0.73764239]
 [0.87416648 0.50679946 0.17060853]
 [0.19647824 0.2195698  0.5010152 ]
 [0.28589369 0.08394862 0.28748563]
 [0.55498598 0.73032299 0.01946458]
 [0.68907645 0.8920713  0.7224627 ]
 [0.36771187 0.91855943 0.87878009]
 [0.14039665 0.88076789 0.76606626]
 [0.84889666 0.57975573 0.70021538]
 [0.45484641 0.17291856 0.42353947]]
Label batch: [0 1 1 0 1 0 1 1 0 0]
-------------------------------------
-------------------------------------
Data batch: [[0.0692231  0.24701816 0.24603659]
 [0.72014948 0.67211487 0.45648504]
 [0.8625562  0.45570299 0.58156546]
 [0.60350332 0.81757841 0.30411054]
 [0.93224841 0.3055118  0.07809648]
 [0.16425884 0.69060297 0.36452719]
 [0.79252781 0.35895253 0.26741555]
 [0.27568602 0.38510119 0.36718876]
 [0.58806512 0.35221788 0.08439596]
 [0.13015496 0.81817428 0.86631724]]
Label batch: [0 0 1 0 1 0 1 0 0 1]

训练循环

训练循环是机器学习的核心机制,模型通过该机制从数据中学习。它们涉及一个重复的过程,即将数据输入模型,计算误差(损失),调整模型的参数以减少该误差,然后重复该过程,直到模型在任务上表现足够好。训练循环是训练神经网络和其他机器学习模型的基础。

训练循环中的关键组件

  • 模型:您想要训练的神经网络或机器学习模型。
  • 损失函数:这是一个衡量模型预测与实际数据匹配程度的函数,例如均方误差、交叉熵。
  • 优化器:一种用于根据计算的梯度更新模型参数的算法,例如 SGD、Adam。
  • 数据:用于训练的数据集通常分为小批量以便高效处理。

为什么训练循环很重要?

训练循环是深度学习和机器学习的基础,原因如下 −

  • 效率:通过以小块(即小批量)处理数据,它们允许在大型数据集上训练模型。
  • 迭代改进:通过反复调整模型的参数,训练循环使模型能够随着时间的推移学习并提高其准确性。
  • 灵活性:可以自定义训练循环以包含其他功能,例如学习率计划、提前停止或监控指标。

训练循环中的关键步骤

以下是训练循环中要遵循的步骤 −

  • 前向传递:首先将输入数据输入模型,然后模型通过其层处理数据以产生输出(预测)。
  • 损失计算:将输出与实际目标值进行比较使用损失函数。损失函数计算预测输出与实际目标之间的误差或差异。
  • 反向传播:计算相对于模型每个参数(权重)的损失梯度。这些梯度表明每个参数对误差的贡献程度。
  • 参数更新:在这里,使用优化算法(如 SGD、Adam 等)更新模型的参数。调整参数以最小化损失。
  • 重复:该过程重复多次迭代(时期),其中模型多次查看数据。目标是让模型通过逐渐减少损失来学习和改进其预测。

示例

在 Chainer 中,训练循环用于遍历数据集、计算损失并更新模型参数。下面是一个使用 Chainer 演示基本训练循环的示例。此示例使用在 MNIST 数据集上训练的简单前馈神经网络。

import chainer
import chainer.functions as F
import chainer.links as L
from chainer import Chain, optimizers, training, serializers
from chainer.datasets import TupleDataset
from chainer.iterators import SerialIterator
from chainer.training import extensions
import numpy as np

# 定义神经网络模型
class SimpleNN(Chain):
    def __init__(self):
        super(SimpleNN, self).__init__()
        with self.init_scope():
        self.l1 = L.Linear(3, 5) # 输入层到隐藏层
        self.l2 = L.Linear(5, 2) # 隐藏层到输出层
    
    def forward(self, x):
        h = F.relu(self.l1(x)) # 应用 ReLU 激活
        y = self.l2(h) # 输出层
        return y
    
    def __call__(self, x, t):
        y = self.forward(x)
        return F.softmax_cross_entropy(y, t)

# 生成合成数据
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=np.float32)
labels = np.array([0, 1, 0], dtype=np.int32)

# 创建数据集和迭代器
dataset = TupleDataset(data, labels)
iterator = SerialIterator(dataset, batch_size=1, shuffle=True)

# 初始化模型、优化器和更新器
model = SimpleNN()
optimizer = optimizers.Adam()
optimizer.setup(model)

# 设置训练器
updater = training.StandardUpdater(iterator, optimizer, device=-1)
trainer = training.Trainer(updater, (10, 'epoch'), out='result')

# 添加扩展以监控训练
trainer.extend(extensions.Evaluator(iterator, model, device=-1))
trainer.extend(extensions.LogReport())
trainer.extend(extensions.PrintReport(['epoch', 'main/loss', 'validation/main/loss']))
trainer.extend(extensions.ProgressBar())

# 开始训练
trainer.run()

这是训练循环的输出 −

epoch      main/loss   validation/main/loss