Learning to Generate Parameters of ConvNets for Unseen Image Data

被引:0
作者
Wang, Shiye [1 ]
Feng, Kaituo [1 ]
Li, Changsheng [1 ]
Yuan, Ye [1 ]
Wang, Guoren [1 ]
机构
[1] Beijing Inst Technol, Sch Comp Sci & Technol, Beijing 100081, Peoples R China
关键词
Training; Task analysis; Correlation; Metalearning; Graphics processing units; Vectors; Adaptive systems; Parameter generation; hypernetwork; adaptive hyper-recurrent units;
D O I
10.1109/TIP.2024.3445731
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Typical Convolutional Neural Networks (ConvNets) depend heavily on large amounts of image data and resort to an iterative optimization algorithm (e.g., SGD or Adam) to learn network parameters, making training very time- and resource-intensive. In this paper, we propose a new training paradigm and formulate the parameter learning of ConvNets into a prediction task: considering that there exist correlations between image datasets and their corresponding optimal network parameters of a given ConvNet, we explore if we can learn a hyper-mapping between them to capture the relations, such that we can directly predict the parameters of the network for an image dataset never seen during the training phase. To do this, we put forward a new hypernetwork-based model, called PudNet, which intends to learn a mapping between datasets and their corresponding network parameters, then predicts parameters for unseen data with only a single forward propagation. Moreover, our model benefits from a series of adaptive hyper-recurrent units sharing weights to capture the dependencies of parameters among different network layers. Extensive experiments demonstrate that our proposed method achieves good efficacy for unseen image datasets in two kinds of settings: Intra-dataset prediction and Inter-dataset prediction. Our PudNet can also well scale up to large-scale datasets, e.g., ImageNet-1K. It takes 8,967 GPU seconds to train ResNet-18 on the ImageNet-1K using GC from scratch and obtain a top-5 accuracy of 44.65%. However, our PudNet costs only 3.89 GPU seconds to predict the network parameters of ResNet-18 achieving comparable performance (44.92%), more than 2,300 times faster than the traditional training paradigm.
引用
收藏
页码:5577 / 5592
页数:16
相关论文
共 69 条
  • [51] Snell J, 2017, ADV NEUR IN, V30
  • [52] Tieleman T., 2017, Tech. Rep.
  • [53] Touvron H, 2023, Arxiv, DOI arXiv:2302.13971
  • [54] Vinyals O., 2016, Adv. Neural Inf. Process. Syst., V29, P3630
  • [55] von Oswald J., 2020, 8 INT C LEARNING REP
  • [56] Wah C., 2011, CALTECH UCSD BIRDS 2
  • [57] Image quality assessment: From error visibility to structural similarity
    Wang, Z
    Bovik, AC
    Sheikh, HR
    Simoncelli, EP
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2004, 13 (04) : 600 - 612
  • [58] Weenink D., 2003, P I PHONETIC SCI U A, V25, P81
  • [59] Large Scale Incremental Learning
    Wu, Yue
    Chen, Yinpeng
    Wang, Lijuan
    Ye, Yuancheng
    Liu, Zicheng
    Guo, Yandong
    Fu, Yun
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 374 - 382
  • [60] Xiao H, 2017, Arxiv, DOI arXiv:1708.07747