LieCConv: An Image Classification Algorithm Based on Lie Group Convolutional Neural Network

被引:0
作者
Zhang, Yunjie [1 ]
Luo, Xizhao [1 ]
Tao, Chongben [2 ]
Qin, Bo [3 ]
Yang, Anjia [4 ]
Cao, Feng [5 ]
机构
[1] Soochow Univ, Sch Comp Sci & Technol, 333 Ganjiang East Rd, Suzhou 215008, Jiangsu, Peoples R China
[2] Suzhou Univ Sci & Technol, Sch Elect & Informat Engn, 99 Xuefu Rd, Suzhou 215009, Jiangsu, Peoples R China
[3] Renmin Univ China, Sch Informat, 59 Zhongguancun St, Beijing 100872, Peoples R China
[4] Jinan Univ, Coll Cyber Secur, 855 East Xingyedadao Ave, Guangzhou 511443, Guangdong, Peoples R China
[5] Shanxi Univ, Sch Comp & Informat Technol, Sch Big Data, 92 Wucheng Rd, Taiyuan 030006, Shanxi, Peoples R China
基金
中国国家自然科学基金;
关键词
Lie group; Convolutional neural network; Sampling algorithm; Image classification;
D O I
10.1007/s11063-024-11691-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In Lie group convolutional neural networks (LG-CNNs), the calculation and storage of Lie group distances have quadratic space complexity. In order to improve the memory utilization efficiency of LG-CNNs, a novel Lie group convolutional neural network called LieCConv is proposed. LieCConv utilizes an innovative sampling algorithm and a linear space complexity calculation and storage approach for Lie group distances, substantially enhancing network memory efficiency. Firstly, LieCConv employs a novel sampling algorithm called array-neighborhood sampling (ANS) in the downsampling stage. ANS only requires neighborhood information to obtain an excellent sample set with a low threshold of use. The sample set generated by ANS reflects the distribution of the original set. Then, LieCConv adopts a batch calculation and storage scheme for Lie group distances, which effectively declines the space complexity of calculating and storing Lie group distances from quadratic complexity to linear complexity, reducing the memory consumption during training. Finally, the contrast between ANS and farthest point sampling was presented, demonstrating that ANS better captures the distribution characteristics of the original dataset. The memory usage of LieCConv and LieConv was compared, revealing that LieCConv reduces the memory usage for calculating and storing Lie group distances to less than 500 MB. And the performance of LieCConv was evaluated on RotMNIST, RotFashionMNIST and TT100K, validating that LieCConv is universal and effective.
引用
收藏
页数:21
相关论文
共 36 条
[1]   Fractional derivative based weighted skip connections for satellite image road segmentation [J].
Arora, Sugandha ;
Suman, Harsh Kumar ;
Mathur, Trilok ;
Pandey, Hari Mohan ;
Tiwari, Kamlesh .
NEURAL NETWORKS, 2023, 161 :142-153
[2]  
Bekkers EJ, 2020, 8 INT C LEARN REPR I
[3]   Improved Unsupervised Neural Machine Translation with Semantically Weighted Back Translation for Morphologically Rich and Low Resource Languages [J].
Chauhan, Shweta ;
Saxena, Shefali ;
Daniel, Philemon .
NEURAL PROCESSING LETTERS, 2022, 54 (03) :1707-1726
[4]   FireClassNet: a deep convolutional neural network approach for PJF fire images classification [J].
Daoud, Zeineb ;
Ben Hamida, Amal ;
Ben Amar, Chokri .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (26) :19069-19085
[5]   A Novel Lightweight Deep Learning-Based Histopathological Image Classification Model for IoMT [J].
Datta Gupta, Koyel ;
Sharma, Deepak Kumar ;
Ahmed, Shakib ;
Gupta, Harsh ;
Gupta, Deepak ;
Hsu, Ching-Hsien .
NEURAL PROCESSING LETTERS, 2023, 55 (01) :205-228
[6]  
Dehmamy N, 2021, ADV NEUR IN, V34
[7]   Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs [J].
Ding, Xiaohan ;
Zhang, Xiangyu ;
Han, Jungong ;
Ding, Guiguang .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :11953-11965
[8]   Lie group symmetry analysis and invariant difference schemes of the two-dimensional shallow water equations in Lagrangian coordinates [J].
Dorodnitsyn, V. A. ;
Kaptsov, E. I. ;
Meleshko, S. V. .
COMMUNICATIONS IN NONLINEAR SCIENCE AND NUMERICAL SIMULATION, 2023, 119
[9]   Learning SO(3) Equivariant Representations with Spherical CNNs [J].
Esteves, Carlos ;
Allen-Blanchette, Christine ;
Makadia, Ameesh ;
Daniilidis, Kostas .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2020, 128 (03) :588-600
[10]  
Finzi M, 2020, PR MACH LEARN RES, V119