TensorFlow中如何进行滑动平均_使用ExponentialMovingAverage实现

老萱姑娘_3123

老萱姑娘_3123

2026-04-26

432人浏览

原创

ExponentialMovingAverage 的 apply() 仅注册更新操作而不自动执行,需手动加入 train_op;decay 过高响应慢、过低平滑差,典型值为0.999;保存恢复必须用 ema.variables_to_restore() 构造 Saver。

tensorflow中如何进行滑动平均_使用exponentialmovingaverage实现

ExponentialMovingAverage 为什么不能直接用 apply() 就完事?

因为 ExponentialMovingAverage 的 apply() 只注册更新操作,不自动执行——它只是把变量更新逻辑塞进一个 ops 列表里,你得手动把它加进训练流程,否则模型参数根本不会被平滑。

  • 常见错误现象:EMA 变量值始终和原始变量一样,sess.run() 查看发现 ema_var 和 var 完全相等
  • 正确做法:把 ema.apply([var1, var2]) 返回的 op 加进 train_op 链,比如用 tf.group(train_op, ema_op)
  • 注意 apply() 必须在所有要跟踪的变量创建后调用,否则会漏掉变量;建议放在 build_graph() 最后、train_op 构造前

滑动衰减率 decay 设成 0.999 还是 0.9999?

不是越大越好。过高的 decay(比如 0.9999)会让 EMA 响应梯度变化极慢,在学习率下降或 fine-tuning 阶段拖累收敛;太低(如 0.9)又起不到平滑噪声的效果。

  • 典型场景:ImageNet 训练常用 0.999;小数据集或 warmup 阶段可设为 0.99,避免初期 EMA 被噪声带偏
  • decay 实际影响的是“有效窗口大小”:近似等价于过去 1/(1-decay) 步的加权平均,0.999 → ~1000 步,0.9999 → ~10000 步
  • TensorFlow 支持动态 decay:传入一个 tf.Tensor(如随 step 线性增长),但要注意它必须是标量且 shape 兼容

保存/加载 EMA 变量时,saver 怎么配才不丢权重?

默认 tf.train.Saver() 只管 trainable_variables,而 EMA 创建的影子变量(如 kernel/ExponentialMovingAverage:0)不在其中——不显式指定,保存和恢复都会失效。

  • 保存时:用 tf.train.Saver(ema.variables_to_restore()) 构造 saver,它返回的是 {ema_var: orig_var} 的映射字典
  • 加载时:如果想用 EMA 权重做 inference,必须 restore 到原始变量名上(即让 kernel:0 被 kernel/ExponentialMovingAverage:0 覆盖),而不是反过来
  • 容易踩的坑:用普通 saver 保存后,再用 ema.variables_to_restore() 加载,会报 Assign requires shapes of both tensors to match——因为变量名对不上,得统一用后者构造 saver

TensorFlow 2.x 用户别硬套 EMA —— tf.keras.optimizers.schedules.ExponentialDecay 不相关

ExponentialMovingAverage 是 TF 1.x Graph 模式下的工具,TF 2.x 默认 Eager,原生没这玩意儿;别被名字误导去查学习率调度器。

  • TF 2.x 替代方案:用 tf.train.Checkpoint + 手动维护影子变量,或直接用第三方库如 torch_ema 的思路移植(需自己写 @tf.function 更新逻辑)
  • Keras 用户更简单:训练完用 model.get_weights() 拿出所有权重,按 decay 加权平均历史 checkpoint,不用碰底层变量名
  • 关键提醒:TF 1.x 代码迁移到 2.x 时,ExponentialMovingAverage 相关部分必须重写,没有兼容层

事情说清了就结束。真正难的不是调通那几行代码,而是想明白——你到底是要用 EMA 做最终 inference 权重,还是只当个训练辅助?前者必须严格控制 restore 流程,后者甚至可能不需要保存 EMA 变量。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1871

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

5084

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1809

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

30637

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3647

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3647

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1243

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2823

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 134.1万人学习