English

Scalable GPU-Accelerated Euler Characteristic Curves: Optimization and Differentiable Learning for PyTorch

Machine Learning 2025-10-24 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2510.20271,
  title  = {Scalable GPU-Accelerated Euler Characteristic Curves: Optimization and Differentiable Learning for PyTorch},
  author = {Udit Saxena},
  journal= {arXiv preprint arXiv:2510.20271},
  year   = {2025}
}

Comments

Extended Abstract: Accepted to the NeurReps 2025 workshop at NeurIPS 2025. 4 pages, 3 figures

R2 v1 2026-07-01T07:01:29.630Z