Multi-Attribute Multi-Grained Adaptation of Pre-Trained Language Models for Text Understanding from Bayesian Perspective
You Zhang, Jin Wang, Liang-Chih Yu, Dan Xu, Xuejie Zhang
Abstract
Current neural networks often employ multi-domain-learning or attribute-injecting mechanisms to incorporate non-independent and identically distributed (non-IID) information for text understanding tasks by capturing individual characteristics and the relationships among samples. However, the extent of the impact of non-IID information and how these methods affect pre-trained language models (PLMs) remains unclear. This study revisits the assumption that non-IID information enhances PLMs to achieve performance improvements from a Bayesian perspective, which unearths and integrates non-IID and IID features. Furthermore, we proposed a multi-attribute multi-grained framework for PLM adaptations (M2A), which combines multi-attribute and multi-grained views to mitigate uncertainty in a lightweight manner. We evaluate M2A through prevalent text-understanding datasets and demonstrate its superior performance, mainly when data are implicitly non-IID, and PLMs scale larger.
Ask Lune
Ask anything. The answer quotes the paper.
Lune reads all of it and the papers around it, then shows you the passages behind every answer.
Builds on8
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- Knowledge Distillation from Internal RepresentationsGustavo Aguilar, Yuan Ling, Yu Zhang, Benjamin Z. Yao et al.AAAI 2020 · 199 citations
- The Power of Scale for Parameter-Efficient Prompt TuningBrian Lester, Rami Al-Rfou, Noah ConstantEMNLP 2021 · 94 citations
- FedBE: Making Bayesian Model Ensemble Applicable to Federated LearningHong-You Chen, Wei-Lun ChaoICLR 2021 · 28 citations
- MA-BERT: Learning Representation by Incorporating Multi-Attribute Knowledge in TransformersYou Zhang, Jin Wang, Liang-Chih Yu, Xuejie ZhangACL Findings 2021 · 23 citations
Related papers
- The Inductive Bias of In-Context Learning: Rethinking Pretraining Example DesignYoav Levine, Noam Wies, Daniel Jannai, Dan Navon et al.ICLR 2022 · 43 citations
- Enhancing LLM Language Adaption through Cross-lingual In-Context Pre-trainingLinjuan Wu, Haoran Wei, Huan Lin, Tianhao Li et al.EMNLP 2025 · 1 citation
- MergeDistill: Merging Language Models using Pre-trained DistillationSimran Khanuja, Melvin Johnson, Partha P. TalukdarACL Findings 2021 · 8 citations
- VarMAE: Pre-training of Variational Masked Autoencoder for Domain-adaptive Language UnderstandingDou Hu, Xiaolong Hou, Xiyang Du, Mengyuan Zhou et al.EMNLP Findings 2022 · 7 citations
- A Hierarchical Encoding-Decoding Scheme for Abstractive Multi-document SummarizationChenhui Shen, Liying Cheng, Xuan-Phi Nguyen, Yang You et al.EMNLP Findings 2023 · 18 citations
