Neural Space-filling Curves

University of Maryland, College Park

Abstract

We present Neural Space-filling Curves (SFCs), a data-driven approach to infer a context-based scan order for a single image or a set of images. Linear ordering of pixels forms the basis for many applications such as video scrambling, compression, and auto-regressive models that are used in generative modeling for images. Existing algorithms resort to a fixed scanning algorithm such as Raster scan or Hilbert scan. Instead, our work learns a spatially coherent linear ordering of pixels from the dataset of images using a graph-based neural network. The resulting Neural SFC is optimized for an objective suitable for the downstream task when the image is traversed along with the scan line order. We show the advantage of using Neural SFCs in downstream applications such as image compression.

Comparison between Hilbert curves and Neural SFCs



Flattened 1D pixel sequences comparison on FFHQ

Up: Neural SFCs

Down: Hilbert curves

Flattened 1D pixel sequences comparison on Fashion-MNIST

Up: Neural SFCs

Down: Hilbert curves

BibTeX

@inproceedings{wang2022neural,
        title={Neural Space-Filling Curves},
        author={Wang, Hanyu and Gupta, Kamal and Davis, Larry and Shrivastava, Abhinav},
        booktitle={European Conference on Computer Vision},
        pages={418--434},
        year={2022},
        organization={Springer}
}