MLflow Model Alias 怎么替代固定版本号部署

MLflow Model Alias 适合解决一个很具体的部署问题：推理服务需要稳定的引用名，但模型版本会不断增加。生产代码可以固定使用 models:/recommendation@champion，由 Model Registry 把 champion 绑定到经过检查的版本；发布新模型时改绑定，回滚时再指回上一版本。这样不必修改、打包和重启每个调用方。

官方地址：https://mlflow.org/

固定版本号为什么不适合直接写进生产代码

models:/recommendation/17 能准确指向第 17 个版本，适合离线复现和回归测试；但当第 18 个版本通过验证后，所有服务仍会继续加载 17。若把 URI 写在环境变量、镜像或代码里，升级就变成一次全量配置变更。

Alias 把“模型是什么版本”和“当前哪个版本承担生产职责”拆开。注册模型仍保留每个版本，调用方只依赖别名。注意别名不是版本快照，它会移动，所以审计记录不能只保存 champion 这个词。

用 MlflowClient 把别名绑定到指定版本

模型已经注册后，可以用客户端设置别名。别名名称应表达职责，例如 champion 表示当前生产模型，challenger 表示待比较候选。下面代码只展示注册表操作，版本号应替换为实际通过检查的版本。

from mlflow import MlflowClient

# 连接到当前环境的 Model Registry。
client = MlflowClient()
model_name = "recommendation"
candidate_version = "18"

# 只有完成离线评估、签名和依赖检查后，才把候选版本标为生产候选。
client.set_model_version_tag(
    model_name, candidate_version, "validation_status", "approved"
)

# Alias 是可移动指针；这里让 champion 指向候选版本。
client.set_registered_model_alias(model_name, "champion", candidate_version)

# 发布记录应保存旧版本与新版本，便于出现异常时回滚。
current = client.get_model_version_by_alias(model_name, "champion")
print(current.version)

这段操作的关键不是“把数字改成别名”，而是把别名移动纳入发布门禁。若验证失败，不应绑定 champion；若切换后发现输入分布、延迟或输出格式异常，应使用发布记录中的上一稳定版本重新绑定。

推理端只加载 Alias URI，切换时不改业务代码

推理服务可以在启动时或受控刷新时读取 Alias URI。加载模型的代码保持不变，变化只发生在注册表中的绑定关系。生产服务不应把 get_model_version_by_alias 的结果当成永久版本；要把解析到的版本号写进启动日志和请求链路，方便定位某次预测实际使用了什么。

import mlflow.pyfunc

# 生产配置只保存稳定的 Alias URI，不把模型版本号写死。
model_uri = "models:/recommendation@champion"
model = mlflow.pyfunc.load_model(model_uri)

# 这里的输入应与注册模型的 signature 兼容；异常要交给发布监控处理。
prediction = model.predict(input_frame)
print(prediction)

如果需要严格复现某次结果，仍应保存解析后的固定 URI，例如 models:/recommendation/18，而不是只保存 champion。Alias 解决的是部署引用的稳定性，不替代实验追踪、模型签名或制品留存。

把切换和回滚放进发布检查

实际切换可以是小流量验证后的别名重绑定，也可以由发布脚本调用同一组 API。无论采用哪种方式，都要先读取当前 champion 指向的版本，再写入变更记录。回滚动作不是删除新版本，而是让 Alias 恢复到上一稳定版本。

Alias 的边界与常见误区

• 把 Alias 当不可变版本：Alias 可以被重新绑定，复现任务必须保存具体版本号。
• 只改注册表不留日志：切换后要能回答“哪个时间点由哪个版本服务”。
• 忽略模型签名：别名切换不会自动修复输入列、类型或输出协议不兼容。

相关问题

Alias 和模型版本号可以同时使用吗？

可以。Alias 用于部署指针，固定版本 URI 用于回归、复现和事故取证，两者职责不同。

一个注册模型能有多个 Alias 吗？

可以使用 champion、challenger 等不同别名，但名称应对应清晰的团队职责，避免把环境名和质量状态混在一起。

切换 Alias 后所有服务会立刻加载新模型吗？

不一定。已经把模型加载到内存的进程可能继续使用旧对象，是否刷新取决于服务的加载策略；应在发布方案中明确重载或滚动更新边界。