- AI新智界
本周值得关注的大模型 / AIGC 前沿研究：
只需半天，训练一个 15 亿参数小模型 Megalodon：具有无限文本长度的高效 LLM 预训练和推理 Melodist：实现包含人声和伴奏的可控文生歌曲模型 替代 Transformer？一文读懂状态空间模型（SSM） 谷歌推出新型 Transformer 架构：反馈注意力就是工作记忆 COMBO：帮助多智能体合作的组合世界模型 MoA：新型“文生图”扩散模型个性化架构 Google DeepMind 新研究：多样本上下文学习 MIT、谷歌新研究：基于零样本跨语言对齐的奖励模型转移 TriForce：利用分层投机解码无损加速长序列生成 1.只需半天，训练一个 15 亿参数小模型德州大学奥斯汀分校团队研究了一种基于现有大型基础语言模型开发小型基础语言模型的简单方法的有效性：首先从大型语言模型中继承几个 transformer 块，然后在大型模型的原始预训练数据的极小子集（0.1%）上训练这个较小的模型。
他们将这一简单方法称为 Inheritune，并首次演示了如何使用 1B token（以及具有 3B 参数的大型语言模型的起始几层）构建具有 1.5B 参数的小型基础语言模型；他们使用单张 A6000 GPU 完成了这一工作，耗时不到半天。
在 9 个不同的评估数据集以及 MMLU 基准测试中，所得到的模型与公开的 1B-2B 大小的基础模型相比性能相当，其中一些模型使用了 50-1000 倍的 token 进行训练。
他们在一个略有不同的环境中研究了 Inheritune，即利用较大的语言模型及其完整的预训练数据集来训练较小的语言模型。他们展示了利用 GPT2-medium（3.55 亿）和 GPT2-large（7.7 亿）的部分层训练的较小语言模型，在具有 9B token 的 OpenWebText 数据集上从头开始训练时，在相同的训练步骤数下，可以有效地与较大语言模型的估值损失相媲美。
论文链接：https://arxiv.org/abs/2404.08634
GitHub 地址：https://github.com/sanyalsunny111/LLM-Inheritune
2.Megalodon：具有无限文本长度的高效 LLM 预训练和推理Transformers 的二次方复杂性和较弱的长度外推能力限制了其扩展至长序列的能力，虽然存在线性注意和状态空间模型等二次方以下的解决方案，但根据经验，它们在预训练效率和下游任务准确性方面都不如 Transformers。
来自 Meta、南加州大学、卡内基梅隆大学和加州大学圣地亚哥分校的研究团队提出了一种用于高效序列建模、上下文长度不受限制的神经架构 Megalodon。
Megalodon 继承了 Mega 架构，并进一步引入了多种技术组件来提高其能力和稳定性，包括复杂指数移动平均线（CEMA）、时间步归一化层、归一化注意力机制和带双跳残差配置的 pre-norm。
在与 Llama2 的可控正面对照比较中，Megalodon 在 70 亿参数和 2 万亿训练 Tokens 的规模上取得了比 transformer 更好的效率。Megalodon 的训练损失为 1.70，在 Llama2-7B (1.75) 和 13B (1.67) 之间。
论文链接：https://arxiv.org/abs/2404.08801
GitHub 地址：https://github.com/XuezheMax/megalodon
3.Melodist：实现包含人声和伴奏的可控文生歌曲模型歌曲是歌声和伴奏的结合，然而，现有的工作主要集中在歌唱声音合成和音乐生成上，很少有人关注歌曲合成。
浙江大学研究团队提出了一项名为“文本到歌曲合成”（text-to-song synthesis）的新任务，其中包含人声和伴奏的生成，他们开发的 Melodist 是一种两阶段文本到歌曲方法，包括歌唱语音合成 （SVS）和人声到伴奏合成 （V2A）。Melodist 利用三塔对比预训练来学习更有效的文本表示，用于可控的 V2A 合成。
为了缓解数据的稀缺性问题，他们构建了一个从音乐网站中挖掘出的中文歌曲数据集。在他们的数据集上的评估结果表明，Melodist 可以合成具有相当质量和风格一致性的歌曲。
论文链接：https://arxiv.org/abs/2404.09313
项目地址：https://text2songmelodist.github.io/Sample/
4.替代 Transformer？一文读懂状态空间模型（SSM）近来，状态空间模型（State Space Model，SSM）作为一种可能替代基于自注意力的 Transformer 的方法，受到了越来越多的关注。在这项工作中，来自安徽大学、哈尔滨工业大学和北京大学的研究团队，首先对这些工作进行了全面的综述，并进行了实验比较和分析，从而更好地展示 SSM 的特点和优势。
具体而言，他们首先对 SSM 的原理进行了详细描述，从而帮助读者快速捕捉 SSM 的主要思想；然后，对现有的 SSM 及其各种应用进行综述，包括自然语言处理、计算机视觉、图、多模态多媒体、点云/事件流、时间序列数据等领域。
此外，他们还对这些模型进行了统计上的比较和分析，希望能帮助读者了解不同结构在各种任务上的有效性。
最后，他们提出了该方向可能的研究方向，从而更好地推动 SSM 理论模型和应用的发展。
论文链接：https://arxiv.org/abs/2404.09516
GitHub 地址：https://github.com/Event-AHU/Mamba_State_Space_Model_Paper_List
5.谷歌推出新型 Transformer 架构：反馈注意力就是工作记忆虽然 Transformer 给深度学习带来了革命性的变化，但二次注意力复杂性阻碍了其处理无限长输入的能力。
谷歌研究团队提出了一种新型 Transformer 架构“反馈注意力记忆”（Feedback Attention Memory，FAM），其利用反馈环路使网络能够关注自身的潜在表征。这种设计促进了 Transformer 工作记忆的出现，使其能够处理无限长的序列。TransformerFAM 不需要额外的权重，因此可以与预训练模型无缝集成。
实验表明，TransformerFAM 显著提高了 Transformer 在各种模型大小（1B、8B 和 24B）的长上下文任务中的性能。
论文链接：https://arxiv.org/abs/2404.09173
6.COMBO：帮助多智能体合作的组合世界模型来自马萨诸塞大学阿默斯特分校、清华大学和北京大学的研究团队及其合作者，探讨了具身多智能体合作问题，在这种情况下，去中心化智能体必须在只有部分自我中心世界观的情况下进行合作。
为了在这种情况下有效地制定计划，与在单个智能体场景中学习世界动态不同，他们必须在仅对世界进行部分自我中心视觉观察的情况下，模拟以任意数量智能体的行动为条件的世界动态。
为了解决部分可观测性这一问题，他们首先训练生成模型，以便在部分自我中心观测条件下估计整体世界状态。为了能够在此世界状态下精确模拟多组行动，他们建议通过对多个智能体的自然可组合联合行动进行因式分解并组合生成视频，从而学习多智能体合作的组合世界模型。利用这种组合世界模型，结合视觉语言模型来推断其他智能体的行动，就可以使用树状搜索程序来整合这些模块，促进在线合作规划。
为了评估这一方法的有效性，他们使用 ThreeDWorld 模拟器创建了两个具有挑战性的多智能体长视野合作任务，并用 2-4 个智能体进行了实验。结果表明，他们的合成世界模型是有效的，而且该框架能使智能体在各种任务和任意数量的智能体中与不同智能体进行高效合作。
论文链接：https://arxiv.org/abs/2404.10775
GitHub 地址：https://vis-www.cs.umass.edu/combo/
7. MoA：新型“文生图”扩散模型个性化架构Snap 研究团队提出了一种新的架构——注意力混合（MoA），用于文本到图像扩散模型个性化。受大型语言模型（LLMs）中使用的专家混合机制（Mixture-of-Experts mechanism）的启发，MoA 在两个注意力路径之间分配生成工作量：个性化分支和非个性化先验分支。
MoA 的设计目的是通过将注意力层固定在先验分支中来保留原始模型的先验性，同时通过个性化分支尽量减少对生成过程的干预，该分支学会将主题嵌入到先验分支生成的布局和上下文中。一种新颖的路由机制可以管理各层像素在这些分支中的分布，从而优化个性化和通用内容创建的融合。
经过训练后，MoA 就能帮助创建高质量的个性化图像，这些图像包含多个主体，其构图和交互方式与原始模型生成的图像一样多样化。重要的是，MoA 增强了模型原有能力与新增强的个性化干预之间的区别，从而提供了一种以前无法实现的更加独立的主体—语境控制。
论文链接：https://arxiv.org/abs/2404.11565
项目地址：https://snap-research.github.io/mixture-of-attention/
8.Google DeepMind 新研究：多样本上下文学习目前，大型语言模型（LLMs）最擅长的是 “少样本上下文学习”（ICL）—— 即在推理时从上下文中提供的少数几个示例中学习，而不进行任何权重更新。新扩展的上下文窗口允许用数百或数千个案例来研究 ICL——多样本模式，从少样本到多样本，在各种生成性和判别性任务中观察到显著的性能提升。
然而，虽然多样本 ICL 前景广阔，但可能会受到人类生成示例可用数量的瓶颈制约。
为此，来自 Google DeepMind 的研究团队探索了两种新的设置：强化 ICL 和无监督 ICL。强化 ICL 使用模型生成的思维链理由来代替人类示例；无监督 ICL 则完全取消了提示中的理由，只用特定领域的问题来提示模型。他们发现，强化型和无监督型 ICL 在多轮推理，尤其是在复杂推理任务中都非常有效。
实验证明，与少样本学习不同，多样本学习在覆盖预训练偏差方面是有效的，并可以学习具有数值输入的高维函数。他们的分析还揭示了下一个 token 预测损失作为下游 ICL 性能指标的局限性。
论文链接：https://arxiv.org/abs/2404.11018
9.MIT、谷歌新研究：基于零样本跨语言对齐的奖励模型转移根据人类标注的偏好数据对语言模型（LMs）进行对齐，是获得基于 LM 的系统实用且性能良好的的关键一步。
然而，多语言人类偏好数据很难大规模获取，因此将这一框架扩展到多种语言具有挑战性。
来自麻省理工大学和谷歌的研究团队评估了一种简单的零样本跨语言对齐方法，即在一种源语言的偏好数据上训练奖励模型，然后直接应用于其他目标语言。在总结和开放式对话生成方面，他们表明这种方法在包括人工评估在内的综合评估设置下始终是成功的，在多达 > 70% 的评估实例中，跨语言对齐模型比非对齐模型更受人类青睐。
此外，不同语言奖励模型有时会比同种语言奖励模型产生更好的对齐模型。他们还确定了在没有特定语言数据甚至监督微调时的最佳实践。
论文链接：https://arxiv.org/abs/2404.12318
10.TriForce：利用分层投机解码无损加速长序列生成最近，随着大型语言模型（LLMs）被广泛应用于长内容生成，对高效长序列推理支持的需求日益增长。
然而，为避免重复计算而存储的键值（KV）缓存已成为一个关键瓶颈，其大小随序列长度呈线性增长。由于 LLM 的自动递归特性，每生成一个 token 就会加载整个 KV 缓存，从而导致计算核心利用率低而延迟高。虽然已经提出了各种 KV 缓存压缩方法来缓解这一问题，但这些方法都会导致生成质量下降。
来自卡内基梅隆大学和 Meta 的研究团队提出了一种可扩展至长序列生成的分层推测解码系统 TriForce。这种方法通过检索利用原始模型权重和动态稀疏 KV 缓存作为草稿模型，作为层次结构中的中间层，并由更小的模型进一步推测，从而减少其草稿延迟。
TriForce 不仅为 Llama2-7B-128K 带来了令人印象深刻的速度提升，在 A100 GPU 上实现了高达 2.31 倍的速度，而且还展示了在处理更长上下文时的可扩展性。在两个 RTX 4090 GPU 上的卸载设置中，TriForce 实现了 0.108s/token 的速度，仅为 A100 上自动回归基线速度的一半，而在优化的卸载系统上则达到了 7.78 倍。
此外，TriForce 在单个 RTX 4090 GPU 上的性能是 DeepSpeed-Zero-Inference 的 4.86 倍。TriForce 的鲁棒性凸显了其在各种温度下始终保持出色的性能。
论文链接：https://arxiv.org/abs/2404.11912
GitHub 地址：https://github.com/Infini-AI-Lab/TriForce
