TomoSphero: Fast Differentiable Projector for Planetary and Solar Tomography on Spherical Grids

Abstract

Computational tomography is a tool for determining the internal structure of objects from a set of projections, typically taken along some regular path. In recent years, methods and GPU-accelerated libraries have emerged that allow for fast reconstruction from projections along more complicated paths. Most of these libraries rely on a Cartesian discretization of the object, which is not appropriate for all scenarios. We present TomoSphero, a differentiable tomographic projector over spherical grids which are often used in planetary and solar tomography. TomoSphero is designed to be used as a building block in reconstruction algorithms and includes common projection types such as cone-beam and parallel-beam, but is flexible enough to accommodate arbitrary projections. TomoSphero is implemented in PyTorch which allows for fast projection computation on GPUs, easy access to modern machine learning optimizers, and automatic differentiation for rapid prototyping of parametric models.