← 研踪 PaperTrailarXiv:2605.07270 · v2
bispectrum: Selective G-Bispectra Made Practical
首次提交 2026-05-08 13:35
版本更新 2026-11-10 00:49
雷达观测 2026-11-10 15:22
摘要
Many machine learning tasks are invariant under transformations such as translations and rotations. The $G$-bispectrum is a complete invariant that determines a signal up to a group action, but its computation is costly and existing implementations each cover a single group. We present bispectrum, an open-source, fully unit-tested PyTorch library providing efficient, differentiable selective $G$-bispectra for seven group actions. Selectivity reduces the number of coefficients from $O(|G|^2)$ to $O(|G|)$ for finite groups, from $O(m^{5/2})$ to $O(m)$ for planar rotations on the disk with $m$ harmonics, and from $Θ(L^3)$ to $Θ(L^2)$ for spherical 3D rotations using our augmented selective spherical bispectrum at band limit $L$. The library is invariant up to floating-point error and computes selective $G$-bispectra in sub-millisecond time on GPU. We evaluate them as pooling layers on PatchCamelyon and OrganMNIST3D against five baselines. Across 46 comparisons, bispectral pooling remains competitive, matching all baselines on PCam and improving on max pooling by 5.0 percentage points on OrganMNIST3D. On Spherical MNIST, our method with an MLP head reaches 95.8% on rotated test digits, matching a spherical CNN at $4\times$ lower latency and coming within 0.3 percentage points of a tensor-product network at $364\times$ lower latency. Selective $G$-bispectra provide fast, competitive invariant layers for finite and continuous groups.
作者备注：NeurIPS 2026 camera-ready
观测到的版本
相邻记录可能跳过中间版本；首次发现 v2 及以上也会标为版本更新。
v2 · 2026-11-10 00:49 · 最新观测版本
bispectrum: Selective G-Bispectra Made Practical
Many machine learning tasks are invariant under transformations such as translations and rotations. The $G$-bispectrum is a complete invariant that determines a signal up to a group action, but its computation is costly and existing implementations each cover a single group. We present bispectrum, an open-source, fully unit-tested PyTorch library providing efficient, differentiable selective $G$-bispectra for seven group actions. Selectivity reduces the number of coefficients from $O(|G|^2)$ to $O(|G|)$ for finite groups, from $O(m^{5/2})$ to $O(m)$ for planar rotations on the disk with $m$ harmonics, and from $Θ(L^3)$ to $Θ(L^2)$ for spherical 3D rotations using our augmented selective spherical bispectrum at band limit $L$. The library is invariant up to floating-point error and computes selective $G$-bispectra in sub-millisecond time on GPU. We evaluate them as pooling layers on PatchCamelyon and OrganMNIST3D against five baselines. Across 46 comparisons, bispectral pooling remains competitive, matching all baselines on PCam and improving on max pooling by 5.0 percentage points on OrganMNIST3D. On Spherical MNIST, our method with an MLP head reaches 95.8% on rotated test digits, matching a spherical CNN at $4\times$ lower latency and coming within 0.3 percentage points of a tensor-product network at $364\times$ lower latency. Selective $G$-bispectra provide fast, competitive invariant layers for finite and continuous groups.
查看此版本解读 →查看此版本原文 ↗