Burn

—Stars

以 Rust 构建的跨平台深度学习框架,兼顾训练、推理、自动微分和多后端部署,强调类型安全、性能以及从桌面到嵌入式设备的可移植性。

RustMulti-backendTrain · Infer · Deploy
RUST TENSOR KERNEL · IEEE 754

保住快速路径,也保住浮点语义

常见有限权重继续走向量化热路径,只有非有限值和 padding 同时出现时才精确修正。

COMMON PATHfinite weights原向量化快速卷积FAST
padding?Inf / NaN?
RARE PATHnon-finite weights精确后处理 padding 区域IEEE 754
路径差异

快速卷积改变了 Inf / NaN 行为

Burn Flex 的 depthwise 与 small-channel 卷积快速路径会跳过 padding 区域,而通用 im2col 路径会把这些位置显式物化为 0。有限权重下两者等价,但 IEEE 754 规定 `0 × Inf` 与 `0 × NaN` 结果为 NaN;仅改变通道数就可能让同一卷积的边界输出在有限值、Inf 与 NaN 之间变化。

性能边界

异常值不能拖慢普通输入

不能简单让所有 padded 卷积回退到通用实现,否则罕见异常值会拖慢全部正常模型。真正需要修正的仅是“非有限权重参与了被省略的零 padding 项”的输出位置;无 padding、全有限权重以及其他通道都应继续复用现有向量化累加。

条件分发

只让罕见组合走修正路径

在进入带 padding 的快速路径时只扫描一次权重是否含非有限值,并精确标记哪些输出位置省略了对应的零乘项;保留原有向量化计算,再仅对受影响位置恢复 NaN 语义。回归覆盖 small-channel、generic、depthwise 三条分发及跨通道隔离。

语义一致

速度与 IEEE 754 同时保留

快速路径与通用 im2col 路径现在对 Inf / NaN 产生一致的 IEEE 754 结果,非有限权重不会污染无关通道;有限权重仍走原有热路径,无 padding 时连额外扫描也不会发生。Burn Flex 相关 741 项测试及 19 项后端卷积测试通过。

PR #5394 已进入 Burn v0.22.0-pre.3 预发布版
  • 卷积非有限 padding 语义修复
  • 预发布说明收录 PR #5394
  • 官方署名 @tandede
查看预发布版本