Beyond Periodicity: Towards a Unifying Framework for Activations in Coordinate-MLPs
About
Coordinate-MLPs are emerging as an effective tool for modeling multidimensional continuous signals, overcoming many drawbacks associated with discrete grid-based approximations. However, coordinate-MLPs with ReLU activations, in their rudimentary form, demonstrate poor performance in representing signals with high fidelity, promoting the need for positional embedding layers. Recently, Sitzmann et al. proposed a sinusoidal activation function that has the capacity to omit positional embedding from coordinate-MLPs while still preserving high signal fidelity. Despite its potential, ReLUs are still dominating the space of coordinate-MLPs; we speculate that this is due to the hyper-sensitivity of networks -- that employ such sinusoidal activations -- to the initialization schemes. In this paper, we attempt to broaden the current understanding of the effect of activations in coordinate-MLPs, and show that there exists a broader class of activations that are suitable for encoding signals. We affirm that sinusoidal activations are only a single example in this class, and propose several non-periodic functions that empirically demonstrate more robust performance against random initializations than sinusoids. Finally, we advocate for a shift towards coordinate-MLPs that employ these non-traditional activation functions due to their high performance and simplicity.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Audio Modeling | Blues 30s v1 (test) | SNR14.74 | 49 | |
| Audio Modeling | Counting 7s v1 (test) | SNR10.97 | 49 | |
| Audio Modeling | Average Bach Counting Blues v1 (test) | SNR13.76 | 46 | |
| Audio Modeling | Bach 7s v1 (test) | SNR15.57 | 46 | |
| Image Representation | Kodak (test) | PSNR37.9 | 13 | |
| 2D Image Representation | DIV2K LR mild track (first 24 images) (val) | PSNR38.34 | 6 | |
| 3D Occupancy Volume Representation | Thai Statue (test) | IoU0.976 | 5 | |
| Image Representation | Kodak | PSNR27.73 | 5 |