能否在Amazon SageMaker模型注册表存储自定义可视化图形（如ROC曲线）

问题描述 #

• 我尝试将部分可视化图形与Pipeline/Model一同存储在SageMaker中，目前仅能通过JSON/PropertyFile存储指标，希望也能存储这些可视化内容。
• 请问是否可通过SageMaker（Studio）模型注册表实现该需求？

相关疑问与观察 #

• 从SageMaker MLOps的通用描述来看似乎可行，但不确定具体实现方式。
• 了解到自定义元数据功能，但它似乎不支持直接存储图片。
• 注意到AutoML已经支持存储图形类内容。

可行实现方案 #

是的，可以通过SageMaker模型注册表实现可视化图形与模型的关联存储，具体可参考以下几种方式：

1. 模型包附加文件存储：将可视化图形（如PNG、JPG等格式）保存为文件后上传至S3，在创建模型包时，通过指定关联的S3路径，将这些文件作为模型包的附加资源进行关联，后续可在SageMaker Studio中通过模型注册表访问这些文件。
2. 实验与模型注册表集成：借助SageMaker实验的Artifacts功能记录可视化图形，再将该实验与模型注册表中的目标模型包关联，这样在模型注册表页面就能查看关联实验中的可视化内容。
3. 自定义报告路径关联：将可视化图形嵌入HTML报告并上传至S3，在模型包的自定义元数据中存储该报告的S3路径（自定义元数据虽不直接存图片，但可存储指向资源的路径），后续在SageMaker Studio中可通过该路径打开查看可视化内容。

AutoML原生支持存储模型洞察图形，是因为其流程内置了可视化的记录与关联逻辑，自定义模型可参考上述方法，实现可视化内容与模型的绑定存储。

内容的提问来源于stack exchange，提问作者stats-hb