Miras是什么
miras是由谷歌开发的专用于深度学习架构设计的通用框架,特别适用于序列建模任务。miras基于关联记忆和注意力偏差的概念,将transformer、现代线性rnn等模型重新定义为具有内部优化目标的关联记忆模块。miras通过四种关键选择构建模型:关联记忆架构、注意力偏差目标、保持门以及记忆学习算法。miras能够生成具有不同优势的新型序列模型,例如moneta、yaad和memora,这些模型在语言建模和常识推理等任务中表现优异,超越现有的transformer和线性rnn模型。
Miras的主要功能
-
整合现有架构:将现有的多种序列模型(如Transformer、RetNet、Mamba等)纳入一个统一的框架中。
-
优化记忆管理:通过引入注意力偏差(Attentional Bias)和保持门(Retention Gate)的概念,Miras能够更好地平衡学习新信息和保留旧信息,从而优化模型的记忆管理能力。
-
设计新型模型:支持设计具有不同注意力偏差和保持机制的全新序列模型,如Moneta、Yaad和Memora。
-
提升模型性能:在处理长序列任务时提升模型性能,同时保持快速的并行化训练能力。
Miras的技术原理
-
关联记忆:一种将输入(键,Keys)映射到输出(值,Values)的机制。在Miras中,序列模型被视为关联记忆模块,基于学习输入和输出之间的映射关系来存储和检索信息。关联记忆是Miras的核心,决定了模型如何存储和利用序列数据中的信息。
-
注意力偏差:作为关联记忆的内部优化目标,用于衡量模型如何优先关注某些事件或刺激。通过选择不同的注意力偏差目标(如ℓ2回归、ℓ1回归、Huber损失等),调整模型对数据的敏感度和鲁棒性。
-
保持门:一种正则化机制,控制模型在学习新信息时如何保留旧信息。通过引入保持正则化项(如ℓ2正则化、KL散度等)来平衡学习和保留,防止模型过度遗忘旧信息,从而在长序列任务中保持更好的性能。
-
记忆学习算法:用于优化关联记忆的目标函数。常见的算法包括梯度下降、动量梯度下降等。选择合适的优化算法可以提高模型的训练效率和收敛速度。
Miras的项目地址
Miras的应用场景
-
语言建模:适用于NLP研究人员和文本生成开发者,用于高效处理长文本并捕捉长距离依赖。
-
常识推理:适用于AI研究者和智能助手开发者,用于提升对隐含信息的理解和推理能力。
-
长文本处理:适用于文本分析工程师和信息检索专家,用于优化长文本处理效率并减少资源消耗。
-
多模态任务:适用于多模态研究者和多媒体内容分析工程师,用于融合多种模态信息并提升跨模态推理能力。
以上就是Miras— 谷歌推出的深度学习架构设计通用框架的详细内容,更多请关注php中文网其它相关文章!