Efficient Approximate Online Convolutional Dictionary Learning

被引:1
作者
Veshki, Farshad G. [1 ,2 ]
Vorobyov, Sergiy A. [1 ,2 ]
机构
[1] Aalto Univ, Dept Informat & Commun Engn, Espoo 02150, Finland
[2] Nokia, Espoo 11351, Finland
基金
芬兰科学院;
关键词
Training; Dictionaries; Convolutional codes; Convolution; Memory management; Optimization; Machine learning; Convolutional sparse coding; online convolutional dictionary learning; SPARSE; ALGORITHMS;
D O I
10.1109/TCI.2023.3340612
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Most existing convolutional dictionary learning (CDL) algorithms are based on batch learning, where the dictionary filters and the convolutional sparse representations are optimized in an alternating manner using a training dataset. When large training datasets are used, batch CDL algorithms become prohibitively memory-intensive. An online-learning technique is used to reduce the memory requirements of CDL by optimizing the dictionary incrementally after finding the sparse representations of each training sample. Nevertheless, learning large dictionaries using the existing online CDL (OCDL) algorithms remains highly computationally expensive. In this paper, we present a novel approximate OCDL method that incorporates sparse decomposition of the training samples. The resulting optimization problems are addressed using the alternating direction method of multipliers. Extensive experimental evaluations using several image datasets and based on an image fusion task show that the proposed method substantially reduces computational costs while preserving the effectiveness of the state-of-the-art OCDL algorithms.
引用
收藏
页码:1165 / 1175
页数:11
相关论文
共 33 条
[1]   K-SVD: An algorithm for designing overcomplete dictionaries for sparse representation [J].
Aharon, Michal ;
Elad, Michael ;
Bruckstein, Alfred .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2006, 54 (11) :4311-4322
[2]  
[Anonymous], 2009, International Conference on Machine Learning, DOI [DOI 10.1145/1553374.1553463, 10.1145/1553374.1553463]
[3]   Distributed optimization and statistical learning via the alternating direction method of multipliers [J].
Boyd S. ;
Parikh N. ;
Chu E. ;
Peleato B. ;
Eckstein J. .
Foundations and Trends in Machine Learning, 2010, 3 (01) :1-122
[4]   Fast Convolutional Sparse Coding [J].
Bristow, Hilton ;
Eriksson, Anders ;
Lucey, Simon .
2013 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2013, :391-398
[5]  
Chalasani R, 2013, IEEE IJCNN
[6]   Dictionary Fields: Learning a Neural Basis Decomposition [J].
Chen, Anpei ;
Xu, Zexiang ;
Wei, Xinyue ;
Tang, Siyu ;
Su, Hao ;
Geiger, Andreas .
ACM TRANSACTIONS ON GRAPHICS, 2023, 42 (04)
[7]   Hyperspectral Image Classification Using Dictionary-Based Sparse Representation [J].
Chen, Yi ;
Nasrabadi, Nasser M. ;
Tran, Trac D. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2011, 49 (10) :3973-3985
[8]   Consensus Convolutional Sparse Coding [J].
Choudhury, Biswarup ;
Swanson, Robin ;
Heide, Felix ;
Wetzstein, Gordon ;
Heidrich, Wolfgang .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :4290-4298
[9]  
Engan K, 1999, INT CONF ACOUST SPEE, P2443, DOI 10.1109/ICASSP.1999.760624
[10]   Convolutional Dictionary Learning: A Comparative Review and New Algorithms [J].
Garcia-Cardona, Cristina ;
Wohlberg, Brendt .
IEEE TRANSACTIONS ON COMPUTATIONAL IMAGING, 2018, 4 (03) :366-381