Saved in:
Bibliographic Details
Main Authors: Korviakov, Vladimir, Koposov, Denis
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.11251
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909139519668224
author Korviakov, Vladimir
Koposov, Denis
author_facet Korviakov, Vladimir
Koposov, Denis
contents Most of the computer vision architectures nowadays are built upon the well-known foundation operations: fully-connected layers, convolutions and multi-head self-attention blocks. In this paper we propose a novel foundation operation - NeoCell - which learns matrix patterns and performs patchwise matrix multiplications with the input data. The main advantages of the proposed operator are (1) simple implementation without need in operations like im2col, (2) low computational complexity (especially for large matrices) and (3) simple and flexible implementation of up-/down-sampling. We validate NeoNeXt family of models based on this operation on ImageNet-1K classification task and show that they achieve competitive quality.
format Preprint
id arxiv_https___arxiv_org_abs_2403_11251
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle NeoNeXt: Novel neural network operator and architecture based on the patch-wise matrix multiplications
Korviakov, Vladimir
Koposov, Denis
Computer Vision and Pattern Recognition
Most of the computer vision architectures nowadays are built upon the well-known foundation operations: fully-connected layers, convolutions and multi-head self-attention blocks. In this paper we propose a novel foundation operation - NeoCell - which learns matrix patterns and performs patchwise matrix multiplications with the input data. The main advantages of the proposed operator are (1) simple implementation without need in operations like im2col, (2) low computational complexity (especially for large matrices) and (3) simple and flexible implementation of up-/down-sampling. We validate NeoNeXt family of models based on this operation on ImageNet-1K classification task and show that they achieve competitive quality.
title NeoNeXt: Novel neural network operator and architecture based on the patch-wise matrix multiplications
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.11251