arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.03315cs.CLcs.AI
收藏

使用双教师自训练的少样本理由生成

Few Shot Rationale Generation using Self-Training with Dual Teachers

Carnegie Mellon University(卡内基梅隆大学);Amazon(亚马逊)

Aditya Srikanth Veerubhotla, Lahari Poddar, Jun Yin, György Szarvas, Sharanya Eswaran

AI总结:

针对少样本且缺乏大规模人工理由与标签的场景,提出双教师自训练框架及MLR损失,将预测与理由教师蒸馏给多任务学生,提升标签建模和忠实理由生成。

摘要:

自我合理化模型还会为其预测标签生成自由文本解释,是构建可信赖AI应用的重要工具。由于为带标注标签生成解释是一项费力且成本高昂的过程,近期模型依赖大型预训练语言模型(PLM)作为骨干并采用少样本学习。在本工作中,我们探索一种利用有标签和无标签数据的自训练方法,以进一步改进少样本模型,其假设是人工编写的理由和带标注的任务标签都无法大规模获得。我们引入一种新颖的双教师学习框架,该框架通过自训练学习两个专门用于任务预测和合理化的教师模型,并将其知识蒸馏到一个多任务学生模型中,该学生模型可联合生成任务标签和理由。此外,我们提出了一种新的损失函数——掩码标签正则化(MLR),它促使解释强烈地以预测标签为条件。在三个公开数据集上的评估表明,所提出的方法在建模任务标签和生成忠实理由方面是有效的。

英文摘要:

Self-rationalizing models that also generate a free-text explanation for their predicted labels are an important tool to build trustworthy AI applications. Since generating explanations for annotated labels is a laborious and costly pro cess, recent models rely on large pretrained language models (PLMs) as their backbone and few-shot learning. In this work we explore a self-training approach leveraging both labeled and unlabeled data to further improve few-shot models, under the assumption that neither human written rationales nor annotated task labels are available at scale. We introduce a novel dual-teacher learning framework, which learns two specialized teacher models for task prediction and rationalization using self-training and distills their knowledge into a multi-tasking student model that can jointly generate the task label and rationale. Furthermore, we formulate a new loss function, Masked Label Regularization (MLR) which promotes explanations to be strongly conditioned on predicted labels. Evaluation on three public datasets demonstrate that the proposed methods are effective in modeling task labels and generating faithful rationales.

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

↑