首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

机器学习理论 | 大型神经语言模型的对抗训练

泛化性和鲁棒性是设计机器学习方法的关键。对抗性训练可以增强鲁棒性,但过去的研究经常发现它会损害泛化能力。在自然语言处理(NLP)中,预训练的大型神经语言模型(如BERT)在各种任务的泛化方面表现出了令人印象深刻的增益,而且通过对抗性微调还可以得到进一步的改进。然而,这些模型仍然容易受到对抗性攻击。在本文中,我们证明了对抗性预训练可以提高泛化性和鲁棒性。我们提出了一种通用算法ALUM(Adversarial training for large neural LangUage Models,大型神经语言模型的对抗性训练),它通过在嵌入空间中施加扰动使对抗性损失最大化来调整训练目标。我们首次全面研究了对抗性训练的各个阶段,包括从头开始的预训练、在训练有素的模式下持续的预训练以及特定任务中的微调。ALUM在各种NLP任务上都比BERT获得了可观的收益,无论是在常规场景还是在对抗场景中。即使是在非常大的文本语料库上受过良好训练的模型,如RoBERTa,ALUM仍然可以从连续的预训练中获得显著的收益,而传统的非对抗性方法则不能。ALUM可以进一步与特定任务的微调相结合,以获得额外的收益。代码和预训练模型可在以下网址获得:https://github.com/namisan/mt-dnn。

03

每日论文速递 | GaLore: 使用梯度低秩映射进行大模型 Memory-Efficient 全参训练

摘要:训练大型语言模型(LLMs)面临着显著的内存挑战,主要是由于权重和优化器状态的不断增大。常见的内存降低方法,如低秩适应(LoRA),在每一层中向冻结的预训练权重添加一个可训练的低秩矩阵,从而减少可训练参数和优化器状态。然而,这些方法通常在预训练和微调阶段的性能上都不如使用全秩权重训练,因为它们将参数搜索限制在低秩子空间中,改变了训练动态,并且可能需要全秩热启动。在这项工作中,我们提出了Gradient Low-Rank Projection(GaLore),一种允许全参数学习但比LoRA等常见低秩适应方法更节省内存的训练策略。我们的方法在优化器状态的内存使用上最多减少了65.5%,同时在使用C4数据集进行LLaMA 1B和7B架构的预训练以及在GLUE任务上对RoBERTa进行微调时,保持了效率和性能。我们的8位GaLore相较于BF16基准,将优化器内存进一步降低了82.5%,总训练内存降低了63.3%。值得注意的是,我们首次证明了在具有24GB内存的消费级GPU上(例如NVIDIA RTX 4090)进行7B模型的预训练是可行的,而无需模型并行、检查点策略或卸载策略。

01

每日论文速递 | 【ICLR24】用语言模型预测表格Tabular

摘要:深度神经网络(DNNs)的可迁移性在图像和语言处理领域取得了显著进展。然而,由于表格之间的异构性,这种DNN的优势在表格数据预测(例如回归或分类任务)方面仍未充分利用。语言模型(LMs)通过从不同领域提炼知识,具有理解来自各种表格的特征名称的能力,有望成为在不同表格和多样化预测任务之间转移知识的多才多艺的学习者,但它们的离散文本表示空间与表格中的数值特征值不兼容。在本文中,我们介绍了TP-BERTa,这是一个专门针对表格数据预测进行预训练的LM模型。具体而言,一种新颖的相对大小标记化将标量数值特征值转换为精细离散的高维标记,而一种内部特征注意方法则将特征值与相应的特征名称集成在一起。全面的实验证明,我们的预训练TP-BERTa在表格DNNs中表现出色,并且在典型的表格数据领域与梯度提升决策树模型相竞争。

01

每日论文速递 | 使用LoRA微调也会过拟合?探索LoRA中的Dropout

摘要:以显著的能力,大语言模型(LLMs)已成为许多自然语言处理应用中不可或缺的元素,而参数高效的微调,特别是 LoRA,已经因其轻量级的模型定制方法而备受青睐。与此同时,各种dropout方法,最初设计用于所有参数更新的完全微调(full weight),缓解了与过度参数冗余相关的过拟合问题。因此,由于 LoRA 的可忽略的可训练参数和先前dropout方法的有效性之间存在可能的矛盾,这一点在很大程度上被忽视了。为了填补这一空白,我们首先确认参数高效的 LoRA 也容易出现过拟合问题。然后,我们重新审视了特定于 transformer 的dropout方法,并从数学和经验上建立了它们的等价性和差异。在这种比较分析的基础上,我们引入了一个统一的框架进行全面的研究,该框架基于dropout位置、结构模式和补偿措施实例化这些方法。通过这个框架,我们揭示了当涉及到有限的可训练参数时,它们的新偏好和性能比较。这个框架还允许我们将最有利的方面融合成一种名为 HiddenKey 的新dropout方法。大量实验证实了 HiddenKey 在多个模型和任务中的显著优越性和充分性,这凸显了它作为大型语言模型的高性能和参数高效微调的首选方法。

01
领券