Skip to main navigation Skip to search Skip to main content

p-Meta: Towards On-device Deep Model Adaptation

  • Zhongnan Qu
  • , Zimu Zhou*
  • , Yongxin Tong
  • , Lothar Thiele
  • *Corresponding author for this work
  • ETH Zurich - Institute for Particle Physics and Astrophysics (IPA)
  • Singapore Management University

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Data collected by IoT devices are often private and have a large diversity across users. Therefore, learning requires pre-training a model with available representative data samples, deploying the pre-trained model on IoT devices, and adapting the deployed model on the device with local data. Such an on-device adaption for deep learning empowered applications demands data and memory efficiency. However, existing gradient-based meta learning schemes fail to support memory-efficient adaptation. To this end, we propose p-Meta, a new meta learning method that enforces structure-wise partial parameter updates while ensuring fast generalization to unseen tasks. Evaluations on few-shot image classification and reinforcement learning tasks show that p-Meta not only improves the accuracy but also substantially reduces the peak dynamic memory by a factor of 2.5 on average compared to state-of-the-art few-shot adaptation methods.

Original languageEnglish
Title of host publicationKDD 2022 - Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining
PublisherAssociation for Computing Machinery
Pages1441-1451
Number of pages11
ISBN (Electronic)9781450393850
DOIs
StatePublished - 14 Aug 2022
Event28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, KDD 2022 - Washington, United States
Duration: 14 Aug 202218 Aug 2022

Publication series

NameProceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining

Conference

Conference28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, KDD 2022
Country/TerritoryUnited States
CityWashington
Period14/08/2218/08/22

Keywords

  • deep neural networks
  • memory-efficient training
  • meta learning

Fingerprint

Dive into the research topics of 'p-Meta: Towards On-device Deep Model Adaptation'. Together they form a unique fingerprint.

Cite this