Differentiable learning of sparse lossless Fourier filters
Source abstract
Abstract
Purpose. The Fourier transform-bounded Kolmogorov complexity proposed in the study by Terry-Jack [J. Software Eng. Appl. 15(10), 359 (2022)], as well as in that by Terry-Jack and O’Keefe [Physica D 453, 133824 (2023)], relies on finding the minimal subset of frequency coefficients that can exactly reconstruct the original image. However, we show that finding this minimal subset of coefficients is a combinatorial optimization problem that becomes computationally intractable for large images. As such, naive solutions such as greedy search are impractical at scale, limiting the feasibility and application of Fourier transform bounded Kolmogorov complexity. Methods. We reformulate the hard combinatorial problem as a differentiable optimization task via the introduction of smooth surrogates for non-differentiable components such as the quantization step and use a straight-through estimator (STE) to learn sparse binary masks (frequency filters) that allow for exact reconstruction of the original image via gradient-descent. Results. Experiments on elementary cellular automata, QR codes, a natural binary image, and random noise demonstrate that sparse Fourier masks capable of bit-exact reconstruction can be learned efficiently at scales that are impractical for a combinatorial or greedy search. Although the problem admits a linear LASSO reformulation that enables classical sparse-recovery methods, experiments against a FISTA baseline demonstrate the advantages of directly optimizing the original nonlinear objective via an STE. Conclusion. This work provides the first practical framework for learning lossless Fourier-domain filters at scale by reframing a hard combinatorial problem as a differentiable learning task.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Computability, Logic, AI Algorithms · Machine Learning and Algorithms · Cellular Automata and Applications
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department, University of York , Heslington, York YO10 5DD,
References (21)
Show source-provided references
1. Novel lossless compression method based on the Fourier transform to approximate the Kolmogorov complexity of elementary cellular automata10.4236/jsea.2022.1510021 ↗
2. Fourier transform bounded Kolmogorov complexity10.1016/j.physd.2023.133824 ↗
3. S. Han , H.Mao, and W. J.Dally, “Deep compression: Compressing deep neural networks with pruning, trained quantization and Huffman coding,” arXiv:1510.00149 (2015).
4. The lottery ticket hypothesis: Finding sparse, trainable neural networks
5. Learning sparse neural networks through L0 regularization
6. End-to-end optimized image compression
7. Categorical reparameterization with Gumbel-Softmax
8. The concrete distribution: A continuous relaxation of discrete random variables
9. Soft-to-hard vector quantization for end-to-end learning compressible representations
10. Y. Bengio , N.Léonard, and A.Courville, “Estimating or propagating gradients through stochastic neurons for conditional computation,” arXiv:1308.3432 (2013).
11. Understanding straight-through estimator in training activation quantized neural nets
12. Binarized neural networks
13. Iterative hard thresholding for compressed sensing10.1016/j.acha.2009.04.002 ↗
14. Regression shrinkage and selection via the lasso10.1111/j.2517-6161.1996.tb02080.x ↗
15. Compressed sensing10.1109/tit.2006.871582 ↗
16. Robust uncertainty principles: Exact signal reconstruction from highly incomplete frequency information10.1109/tit.2005.862083 ↗
17. Atomic decomposition by basis pursuit10.1137/s003614450037906x ↗
18. Orthogonal matching pursuit: Recursive function approximation with applications to wavelet decomposition
19. An iterative thresholding algorithm for linear inverse problems with a sparsity constraint10.1002/cpa.20042 ↗
20. A fast iterative shrinkage-thresholding algorithm for linear inverse problems10.1137/080716542 ↗
21. Adam: A method for stochastic optimization