arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2110.08190cs.CL

Sparse Progressive Distillation: Resolving Overfitting under Pretrain-and-Finetune Paradigm

University of Connecticut(康涅狄格大学);Penn State University(宾夕法尼亚州立大学);Moffett AI(莫菲特人工智能公司);Northeastern University(东北大学);Stevens Institute of Technology(史蒂文斯理工学院);University of Texas at San Antonio(德克萨斯大学圣安东尼奥分校)

Shaoyi Huang, Dongkuan Xu, Ian E. H. Yen, Yijue Wang, Sung-en Chang, Bingbing Li, Shiyang Chen, Mimi Xie, Sanguthevar Rajasekaran, Hang Liu, Caiwen Ding

英文摘要:

Conventional wisdom in pruning Transformer-based language models is that pruning reduces the model expressiveness and thus is more likely to underfit rather than overfit. However, under the trending pretrain-and-finetune paradigm, we postulate a counter-traditional hypothesis, that is: pruning increases the risk of overfitting when performed at the fine-tuning phase. In this paper, we aim to address the overfitting problem and improve pruning performance via progressive knowledge distillation with error-bound properties. We show for the first time that reducing the risk of overfitting can help the effectiveness of pruning under the pretrain-and-finetune paradigm. Ablation studies and experiments on the GLUE benchmark show that our method outperforms the leading competitors across different tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑