中文
相关论文

相关论文: Efficient Fine-Tuning of BERT Models on the Edge

200 篇论文

In finetuning a large pretrained model to downstream tasks, parameter-efficient fine-tuning (PEFT) methods can effectively finetune pretrained models with few trainable parameters, but suffer from high GPU memory consumption and slow…

计算机视觉与模式识别 · 计算机科学 2024-02-07 Ningyuan Tang , Minghao Fu , Ke Zhu , Jianxin Wu

Strategy training is a multidisciplinary rehabilitation approach that teaches skills to reduce disability among those with cognitive impairments following a stroke. Strategy training has been shown in randomized, controlled clinical trials…

Federated Learning (FL) has emerged as a new paradigm for training machine learning models distributively without sacrificing data security and privacy. Learning models on edge devices such as mobile phones is one of the most common use…

机器学习 · 计算机科学 2023-02-10 Sixing Yu , Phuong Nguyen , Ali Anwar , Ali Jannesari

Deformable Attention Transformers (DAT) have shown remarkable performance in computer vision tasks by adaptively focusing on informative image regions. However, their data-dependent sampling mechanism introduces irregular memory access…

计算机视觉与模式识别 · 计算机科学 2025-07-29 Wendong Mao , Mingfan Zhao , Jianfeng Guan , Qiwei Dong , Zhongfeng Wang

The Metaverse has received much attention recently. Metaverse applications via mobile augmented reality (MAR) require rapid and accurate object detection to mix digital data with the real world. Federated learning (FL) is an intriguing…

机器学习 · 计算机科学 2023-12-08 Xinyu Zhou , Chang Liu , Jun Zhao

The recent trend in industry-setting Natural Language Processing (NLP) research has been to operate large %scale pretrained language models like BERT under strict computational limits. While most model compression work has focused on…

计算与语言 · 计算机科学 2021-04-13 J. S. McCarley , Rishav Chakravarti , Avirup Sil

On-device training enables the model to adapt to new data collected from the sensors by fine-tuning a pre-trained model. Users can benefit from customized AI models without having to transfer the data to the cloud, protecting the privacy.…

计算机视觉与模式识别 · 计算机科学 2024-04-04 Ji Lin , Ligeng Zhu , Wei-Ming Chen , Wei-Chen Wang , Chuang Gan , Song Han

Wireless federated learning (WFL) suffers from heterogeneity prevailing in the data distributions, computing powers, and channel conditions of participating devices. This paper presents a new Federated Learning with Adjusted leaRning ratE…

信号处理 · 电气工程与系统科学 2024-04-24 Bingnan Xiao , Jingjing Zhang , Wei Ni , Xin Wang

We propose a novel parameter-efficient training (PET) method for large language models that adapts models to downstream tasks by optimizing a small subset of the existing model parameters. Unlike prior methods, this subset is not fixed in…

计算与语言 · 计算机科学 2024-11-14 Felix Stahlberg , Jared Lichtarge , Shankar Kumar

Fine-tuning large pre-trained models on downstream tasks has been adopted in a variety of domains recently. However, it is costly to update the entire parameter set of large pre-trained models. Although recently proposed parameter-efficient…

计算与语言 · 计算机科学 2022-11-01 Yi-Lin Sung , Jaemin Cho , Mohit Bansal

Most federated learning (FL) approaches assume a fixed device set. However, real-world scenarios often involve devices dynamically joining or leaving the system, driven by, e.g., user mobility patterns or handovers across cell boundaries.…

机器学习 · 计算机科学 2025-12-30 Zhan-Lun Chang , Dong-Jun Han , Seyyedali Hosseinalipour , Mung Chiang , Christopher G. Brinton

Pre-trained language models, such as BERT, have achieved significant accuracy gain in many natural language processing tasks. Despite its effectiveness, the huge number of parameters makes training a BERT model computationally very…

计算与语言 · 计算机科学 2020-11-30 Cheng Yang , Shengnan Wang , Chao Yang , Yuechuan Li , Ru He , Jingqiao Zhang

Frugal Machine Learning (FML) refers to the practice of designing Machine Learning (ML) models that are efficient, cost-effective, and mindful of resource constraints. This field aims to achieve acceptable performance while minimizing the…

机器学习 · 计算机科学 2025-06-03 John Violos , Konstantina-Christina Diamanti , Ioannis Kompatsiaris , Symeon Papadopoulos

On-device learning enables edge devices to continually adapt the AI models to new data, which requires a small memory footprint to fit the tight memory constraint of edge devices. Existing work solves this problem by reducing the number of…

计算机视觉与模式识别 · 计算机科学 2021-06-08 Han Cai , Chuang Gan , Ligeng Zhu , Song Han

On-device learning allows AI models to adapt to user data, thereby enhancing service quality on edge platforms. However, training AI on resource-limited devices poses significant challenges due to the demanding computing workload and the…

硬件体系结构 · 计算机科学 2023-12-27 Sai Qian Zhang , Thierry Tambe , Nestor Cuevas , Gu-Yeon Wei , David Brooks

Large machine learning models trained on diverse data have recently seen unprecedented success. Federated learning enables training on private data that may otherwise be inaccessible, such as domain-specific datasets decentralized across…

Transfer learning with large pretrained transformer-based language models like BERT has become a dominating approach for most NLP tasks. Simply fine-tuning those large language models on downstream tasks or combining it with task-specific…

计算与语言 · 计算机科学 2021-08-06 Wenjuan Han , Bo Pang , Yingnian Wu

Efficient Federated learning (FL) is crucial for training deep networks over devices with limited compute resources and bounded networks. With the advent of big data, devices either generate or collect multimodal data to train either…

机器学习 · 计算机科学 2025-09-16 Sahil Tyagi

Despite their exceptional performance on various tasks after fine-tuning, pre-trained language models (PLMs) face significant challenges due to growing privacy concerns with data in centralized training methods. We consider federated…

机器学习 · 计算机科学 2024-05-28 Yuxuan Yan , Qianqian Yang , Shunpu Tang , Zhiguo Shi

Pre-trained and fine-tuned transformer models like BERT and T5 have improved the state of the art in ad-hoc retrieval and question-answering, but not as yet in high-recall information retrieval, where the objective is to retrieve…

信息检索 · 计算机科学 2022-08-16 Nima Sadri , Gordon V. Cormack