Computer Science ›› 2026, Vol. 53 ›› Issue (7): 139-145.doi: 10.11896/jsjkx.250600038

• Artificial Intelligence • Previous Articles     Next Articles

Gate-controlled Agent Attention Mechanism-based Soft Prompt Transfer Method

ZHANG Yan, ZHOU Jian, HAN Lei, CHENG Chunling   

  1. School of Computer Science,Nanjing University of Posts and Telecommunications,Nanjing 210023,China
  • Received:2025-06-05 Revised:2025-08-12 Online:2026-07-15 Published:2026-07-10
  • About author:ZHANG Yan,born in 2002,postgra-duate.His main research interests include deep learning and prompt lear-ning.
    CHENG Chunling,born in 1972,professor.Her main research interests include data mining and data management.
  • Supported by:
    National Natural Science Foundation of China(62472232).

Abstract: Soft prompt transfer,as a cross-task learning approach,aims to guide the target task model to learn more generalizable prompt representations by leveraging knowledge from source prompts.However,existing methods often ignore the impact of task-specific information embedded in target samples,which may result in biased target prompt training.To address this issue,a gate-controlled agent attention mechanism-based soft prompt transfer method(GPAPT) is proposed.Firstly,to separate and enhance both global and task-specific information in target samples,a dual-path feature extraction strategy is introduced.It employs two lightweight extraction paths to derive agent tokens that encode task-level semantics and local features that capture instance-specific characteristics.Secondly,to assess the transferability of source prompts,a gate-controlled agent attention mechanism is presented.It computes attention distributions between prompt features and both agent tokens and local features to model task similarity.To preserve task-specific information,a gating unit filters local features and integrates the two types of attention.Finally,a perturbation mechanism for attention distribution is introduced.It perturbs the attention weights based on the prediction loss of source prompts on the target task,thereby reduces the influence of source prompts that are similar in the representation space but perform poorly in prediction,and thus improving transfer robustness.Extensive experiments on the GLUE benchmark demonstrate that GPAPT achieves the best average performance compared to nine strong baseline methods,validating the effectiveness of the proposed approach.

Key words: Prompt tuning, Soft prompt transfer, Feature extraction, Gate-controlled agent attention mechanism, Attention distribution perturbation

CLC Number: 

  • TP391.1
[1]RAFFEL C,SHAZEER N,ROBERTS A,et al.Exploring the limits of transfer learning with a unified text-to-text transformer[J].Journal of Machine Learning Research,2020,21(140):1-67.
[2]HOULSBY N,GIURGIU A,JASTRZEBSKI S,et al.Parame-ter-efficient transfer learning for NLP[C]//Proceedings of the 36th International Conference on Machine Learning.2019,97:2790-2799.
[3]LIU P,YUAN W,FU J,et al.Pre-train,prompt,and predict:A systematic survey of prompting methods in natural language processing[J].ACM Computing Surveys,2023,55(9):1-35.
[4]BEN ZAKEN E,GOLDBERG Y,RAVFOGEL S.BitFit:Simpleparameter-efficient fine-tuning for transformer-based masked language models[C]//Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics.2022:1-9.
[5]LESTER B,AL-RFOU R,CONSTANT N.The power of scale for parameter-efficient prompt tuning[C]//Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing(EMNLP).2021:3045-3059.
[6]VU T,LESTER B,CONSTANT N,et al.SPoT:Better frozenmodel adaptation through soft prompt transfer[C]//Procee-dings of the 60th Annual Meeting of the Association for Computational Linguistics.2022:5039-5059.
[7]WANG A,SINGH A,MICHAEL J,et al.GLUE:A multi-task benchmark and analysis platform for natural language understanding[C]//Proceedings of the 2018 EMNLP Workshop BlackboxNLP:Analyzing and Interpreting Neural Networks for NLP.2018:353-355.
[8]HAN D,YE T,HAN Y,et al.Agent attention:On the integration of softmax and linear attention [C]//European Conference on Computer Vision.Cham:Springer,2024:124-140.
[9]WANG Z,PANDA R,KARLINSKY L,et al.Multitask prompt tuning enables parameter-efficient transfer learning[C]//The Eleventh International Conference on Learning Representations(ICLR).2023.
[10]ASAI A,SALEHI M,PETERS M,et al.ATTEMPT:Parameter-efficient multi-task tuning via attentional mixtures of soft prompts[C]//Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing.2022:6655-6672.
[11]WU X,CHEN C L P,LI S,et al.Snapshot prompt ensemble for parameter-efficient soft prompt transfer[C]//ICASSP 2024-2024 IEEE International Conference on Acoustics,Speech and Signal Processing(ICASSP).2024:11946-11950.
[12]WU M,LIU W,XU J,et al.Parameter-efficient multi-task fine-tuning by learning to transfer token-wise prompts[C]//Findings of the Association for Computational Linguistics:EMNLP 2023.2023:8734-8746.
[13]ZHONG Q,DING L,LIU J,et al.Panda:Prompt transfer meets knowledge distillation for efficient model adaptation[J].IEEE Transactions on Knowledge and Data Engineering,2024,36:4835-4848.
[14]WANG A,PRUKSACHATKUN Y,NANGIA N,et al.Super GLUE:A stickier benchmark for general-purpose language understanding systems[C]//Proceedings of the 33rd International Conference on Neural Information Processing Systems.2019:3266-3280.
[15]KHOT T,SABHARWAL A,CLARK P.Sci Tail:A textual entailment dataset from science question answering [C]//Procee-dings of AAAI.2018:5189-5197.
[16]MAHABADI R K,RUDER S,DEHGHANI M,et al.Parameter-efficient multi-task fine-tuning for transformers via shared hypernetworks [C]//Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing.2021:565-576.
[1] YAO Ye, GUO Kangning, ZHU Yian, LIAO Shaochun, ZHANG Ni. Research on Health Evaluation Technology of AHP-FEC Meteorological Equipment Based onLasso Optimization [J]. Computer Science, 2026, 53(6A): 250400123-16.
[2] ZHANG Shouyi, SHEN Qiang, GUO Yiran, WANG Hanyu. Rain and Fog Weather Object Detection Algorithm Based on Improved YOLOv8 Model [J]. Computer Science, 2026, 53(6A): 250300090-7.
[3] MENG Siyu, NIU Chunxiang, TAN Quange, WANG Rong. Deepfake Detection Method Based on Positional Enhancement and Frequency Domain ComponentInteraction [J]. Computer Science, 2026, 53(4): 445-453.
[4] ZHAO Binbei, ZHU Li, ZHAO Hongli, LI Yutong. Computer Vision Applications in Rail Transit Systems [J]. Computer Science, 2026, 53(3): 214-224.
[5] GUO Xingxing, XIAO Yannan, WEN Peizhi, XU Zhi, HUANG Wenming. Attention-based Audio-driven Digital Face Video Generation Method [J]. Computer Science, 2026, 53(2): 245-252.
[6] LI Mengxi, GAO Xindan, LI Xue. Two-way Feature Augmentation Graph Convolution Networks Algorithm [J]. Computer Science, 2025, 52(7): 127-134.
[7] WU Zhihua, CHENG Jianghua, LIU Tong, CAI Yahui, CHENG Bang, PAN Lehao. Human Target Detection Algorithm for Low-quality Laser Through-window Imaging [J]. Computer Science, 2025, 52(6A): 240600069-6.
[8] LIU Yuanhong, WU Yubin. Local Linear Embedding Algorithm Based on Probability Model and Information Entropy [J]. Computer Science, 2025, 52(6A): 240500021-8.
[9] MIAO Zhuang, CUI Haoran, ZHANG Qiyang, WANG Jiabao, LI Yang. Restoration of Atmospheric Turbulence-degraded Images Based on Contrastive Learning [J]. Computer Science, 2025, 52(5): 171-178.
[10] KONG Yu, XIONG Fengguang, ZHANG Zhiqiang, SHEN Chaofan, HU Mingyue. Low Overlap Point Cloud Registration Method Based on Deep Position-aware Transformer [J]. Computer Science, 2025, 52(5): 199-211.
[11] LI Xiaolan, MA Yong. Study on Lightweight Flame Detection Algorithm with Progressive Adaptive Feature Fusion [J]. Computer Science, 2025, 52(4): 64-73.
[12] WANG Mengwei, YANG Zhe. Speaker Verification Method Based on Sub-band Front-end Model and Inverse Feature Fusion [J]. Computer Science, 2025, 52(3): 214-221.
[13] ZUO Xuhong, WANG Yongquan, QIU Geping. Study on Integrated Model of Securities Illegal Margin Trading Accounts Identification Based on Trading Behavior Characteristics [J]. Computer Science, 2025, 52(2): 125-133.
[14] WANG Kangyue, CHENG Ming, XIE Yixiang, ZOU Xiaobing, LI Ming. Role-aware Speaker Diarization in Autism Interview Scenarios [J]. Computer Science, 2025, 52(2): 231-241.
[15] ZHU Sifan, ZHU Guosheng. Retinal Vessel Segmentation Based on Multi-scale Attention [J]. Computer Science, 2025, 52(11A): 241200112-10.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!