WaveMix: A Versatile and Resource-Efficient Neural Network for Image Analysis
WaveMix is a novel neural architecture that achieves comparable or better accuracy than state-of-the-art convolutional neural networks, vision transformers, and token mixers for various vision tasks, while using fewer trainable parameters, GPU RAM, and computations.