Transverse-layer partitioning of artificial neural networks for image classification

• Published in: Kompʹûternaâ optika Vol. 48; no. 2; pp. 312 - 320
• Main Authors: Vershkov, N.A., Babenko, M.G., Kuchukova, N.N., Kuchukov, V.A., Kucherov, N.N.
• Format: Journal Article
• Language: English
• Published: Samara National Research University 01.04.2024
• Subjects: artificial neural networks; convolutional layer; modular learning; orthogonal transforms; wavelet transform
• ISSN: 0134-2452; 2412-6179
• DOI: 10.18287/2412-6179-CO-1278

We discuss issues of modular learning in artificial neural networks and explore possibilities of the partial use of modules when the computational resources are limited. The proposed method is based on the ability of a wavelet transform to separate information into high- and low-frequency parts. Using the expertise gained in developing convolutional wavelet neural networks, the authors perform a transverse-layer partitioning of the network into modules for the further partial use on devices with low computational capability. The theoretical justification of this approach in the paper is supported by experimentally dividing the MNIST database into 2 and 4 modules before using them sequentially and measuring the respective accuracy and performance. When using the individual modules, a two-fold (or higher) performance gain is achieved. The theoretical statements are verified using an AlexNet-like network on the GTSRB dataset, with a performance gain of 33% per module with no loss of accuracy.