Scalable GPU-Accelerated Euler Characteristic Curves: Optimization and Differentiable Learning for PyTorch
By: Udit Saxena
Potential Business Impact:
Helps computers understand shapes in pictures faster.
Topological features capture global geometric structure in imaging data, but practical adoption in deep learning requires both computational efficiency and differentiability. We present optimized GPU kernels for the Euler Characteristic Curve (ECC) computation achieving 16-2000\"O speedups over prior GPU implementations on synthetic grids, and introduce a differentiable PyTorch layer enabling end-to-end learning. Our CUDA kernels, optimized for Ampere GPUs use 128B-coalesced access and hierarchical shared-memory accumulation. Our PyTorch layer learns thresholds in a single direction via a Differentiable Euler Characteristic Transform-style sigmoid relaxation. We discuss downstream relevance, including applications highlighted by prior ECC work, and outline batching/multi-GPU extensions to broaden adoption.
Similar Papers
Vectorized Computation of Euler Characteristic Functions and Transforms
Computational Geometry
Calculates shapes faster for computers.
Topology Optimization in Medical Image Segmentation with Fast Euler Characteristic
Image and Video Processing
Fixes medical scans for better health checks.
Topology Optimization in Medical Image Segmentation with Fast Euler Characteristic
Image and Video Processing
Fixes medical scans to show body parts correctly.