技术
StopGrad(梯度截断)
StopGrad是一种在反向传播中阻止梯度流经特定张量的技术,常用于多任务学习或对抗训练中分离梯度。
详细解释
StopGrad(Stop Gradient)是深度学习中的一种操作,在前向传播时正常计算,但在反向传播时截断梯度,使该张量不参与梯度更新。它常用于多任务学习中避免任务间干扰,或在生成对抗网络中分离生成器和判别器的梯度。在PyTorch中通过detach()实现,在TensorFlow中通过tf.stop_gradient()实现。该技术有助于稳定训练和实现复杂的优化目标。
别名与相关写法
stopgrad
相关资讯
StopGrads的平稳性、收敛性与流图学习2026/9/16