12 月 20 日消息,索尼 PlayStation 5 游戏主机标准和 Pro 版本首席架构师 Mark Cerny 近日对 PS5 Pro 主机支持的 PSSR 超分辨率技术进行了详细介绍。

整个 PSSR 超分辨率工作流程由前处理 + 轻量级 CNN 卷积神经网络 + 后处理三部分组成,而 NPU 在前后处理中的表现不如 AI GPU 方案。最终索尼选择了对 RDNA 架构的着色器核心和缓存系统针对性增强的方式(注:即“定制 RDNA”) 来实现超分辨率。

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第1张

索尼为 PS5 Pro 设下了 300TOPs(8bit 整型)的 AI 性能目标,而这意味着 PS5 Pro 的 GDDR6 带宽远不能满足每步 CNN 处理中间结果均导出至片外缓存的需求。

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第2张

索尼因此追求尽可能在片上缓存中存储中间步骤数据,将访问片外 GDDR 的需求降到最低。而 300TOPs 算力目标对应的数百 TB/s 量级带宽无法由数 TB/s 的 L2 缓存满足。

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第3张

▲ “全融合网络”(完全片上处理)的理想
该目标在 PS5 Pro 上未能实现但已接近

PS5 Pro 开发团队最终将目光盯上了 WGP 矢量寄存器,并通过图像分块降低了对片上缓存容量的需求。整个 PS5 Pro 芯片的 WGP 矢量寄存器合计容量达 15MB,可提供 200TB/s 带宽,承担了 CNN 中间数据存储的重任。

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第4张

▲ 图像分块以减少片上缓存需求

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第5张

索尼在 PS5 上也对 RDNA GPU 的架构和指令集进行了修改,新增了 44 条与 CNN 计算和 WGP 矢量寄存器操作有关的着色器指令。

此外 PSSR 和其它超分的主要区别之一是在 PC 游戏中超分辨率一般而言是固定原生分辨率和超分比例,动态调整最终帧率;而 PSSR 技术的最终帧率一般处于固定状态,超分比例随可达到的原生分辨率而变动。

索尼介绍 PS5 Pro 主机 PSSR 超分技术:利用 WGP 矢量寄存器缓解带宽瓶颈  第6张