Memristive Cosine-Similarity-Based Few-Shot Learning with Lifelong Memory Adaptation

被引:3
|
作者
Zhou, Houji [1 ,2 ]
Yang, Ling [2 ]
Bao, Han [2 ]
Li, Jiancong [2 ]
Li, Yi [1 ,2 ]
Miao, Xiangshui [1 ,2 ]
机构
[1] Hubei Yangtze Memory Labs, Wuhan 430205, Peoples R China
[2] Huazhong Univ Sci & Technol, Sch Integrated Circuits, Wuhan 430074, Peoples R China
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
cosine similarity; few-shot learning; lifelong learning; memristive in-memory computing; prototype augmentation; NEURAL-NETWORK;
D O I
10.1002/aisy.202200173
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
External-memory-based few-shot learning has emerged as a promising solution to deal with the shortage of labeled samples in traditional deep neural networks (DNNs). The external memories here are used to perform explicit similarity calculations. Despite several encouraging schemes were proposed to merge data storage and similarity calculation using non-volatile memories, two major challenges, including the long encoding bits and the degraded accuracy of hardware implementations, remain to be tackled. Herein, a cosine-similarity-based few-shot learning method with analog memristors is designed. A regulation block in the embedding net is used as a hardware-software co-optimization unit to reduce the encoding lengths of the embedding vectors. Then, a lifelong memory adaptation method, termed the online prototype augmentation, is proposed for the first time to promote the accuracy of extremely low-shot learning problems. Results show that the encoding bits are 4-32 times smaller than popular schemes. Using the Omniglot dataset, record accuracy (98.3% and 97.3% for the 5-way 5-shot and 5-way 1-shot problems, respectively) can be obtained for the few-shot learning with memristive cosine similarities. In addition, the online prototype augmentation method leads to at least approximate to 2% improvement in accuracy for 1-shot learning, which is outstanding compared to prior studies.
引用
收藏
页数:9
相关论文
共 50 条
  • [1] Few-Shot Lifelong Learning
    Mazumder, Pratik
    Singh, Pravendra
    Rai, Piyush
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 2337 - 2345
  • [2] Cosine Similarity based Few-Shot Video Classifier with Attention-based Aggregation
    Debnath, Biplob
    Po, Oliver
    Chowdhury, Farhan Asif
    Chakradhar, Srimat
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 1273 - 1279
  • [3] NegCosIC: Negative Cosine Similarity-Invariance- Covariance Regularization for Few-Shot Learning
    Liu, Wei Han
    Lim, Kian Ming
    Ong, Thian Song
    Lee, Chin Poo
    IEEE ACCESS, 2024, 12 : 52867 - 52877
  • [4] Understanding Cross-Domain Few-Shot Learning Based on Domain Similarity and Few-Shot Difficulty
    Oh, Jaehoon
    Kim, Sungnyun
    Ho, Namgyu
    Kim, Jin-Hwa
    Song, Hwanjun
    Yun, Se-Young
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [5] Only Image Cosine Embedding for Few-Shot Learning
    Gao, Songyi
    Shen, Weijie
    Liu, Zelin
    Zhu, An
    Yu, Yang
    NEURAL INFORMATION PROCESSING (ICONIP 2019), PT II, 2019, 11954 : 83 - 94
  • [6] Smoothness Similarity Regularization for Few-Shot GAN Adaptation
    Sushko, Vadim
    Wang, Ruyu
    Gall, Juergen
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION, ICCV, 2023, : 7050 - 7059
  • [7] Tomato Fungal Disease Diagnosis Using Few-Shot Learning Based on Deep Feature Extraction and Cosine Similarity
    Javidan, Seyed Mohamad
    Ampatzidis, Yiannis
    Banakar, Ahmad
    Vakilian, Keyvan Asefpour
    Rahnama, Kamran
    AGRIENGINEERING, 2024, 6 (04): : 4233 - 4247
  • [8] Few-shot Learning for New Environment Adaptation
    Wang, Ouya
    Zhou, Shenglong
    Li, Geoffrey Ye
    IEEE CONFERENCE ON GLOBAL COMMUNICATIONS, GLOBECOM, 2023, : 351 - 356
  • [9] Subspace Adaptation Prior for Few-Shot Learning
    Mike Huisman
    Aske Plaat
    Jan N. van Rijn
    Machine Learning, 2024, 113 : 725 - 752
  • [10] Subspace Adaptation Prior for Few-Shot Learning
    Huisman, Mike
    Plaat, Aske
    van Rijn, Jan N.
    MACHINE LEARNING, 2024, 113 (02) : 725 - 752