核心发现
方法论
MultiHashFormer引入多哈希签名机制,将每个词表示为由多个独立哈希函数生成的离散哈希ID序列,避免了传统哈希模型中的多对一碰撞问题。模型由三部分组成:哈希编码器(Hash Encoder)将离散签名压缩成单一潜在向量,Transformer解码器(Transformer Decoder)处理序列,哈希解码器(Hash Decoder)则逐步重建下一词的哈希签名,最后映射回文本。该框架支持因果生成,参数规模在不同规模(100M、1B、3B)下均优于标准Transformer,且在多语言词汇扩展中表现出零参数增长的优势。
关键结果
- 在100M、1B和3B参数尺度上,MultiHashFormer在多个基准测试中持续优于标准Transformer,尤其在LAMBADA、ReCoRD等任务上表现出明显优势,例如在1B模型中,MultiHashFormer(H4B16K配置)在ReCoRD上得分达到了64.90,超越标准模型的63.78,提升了约1.12分。
- 模型在词汇扩展任务中表现出色,支持在不增加参数的情况下,将词汇从32K扩展到48K,且在多语言环境(阿拉伯语、汉语、印地语)中,扩展后仍优于传统模型,显示出其在多语种、多任务场景中的潜力。
- 通过在Card-660数据集上的语义相似性评估,MultiHashFormer在识别稀有词对方面的相关系数(Pearson和Spearman)均优于标准模型,尤其在第二倒数隐藏状态中表现更佳,证明其在捕获稀有词语语义方面具有优势。
研究意义
该研究突破了传统词嵌入矩阵线性增长的瓶颈,为大规模、多语种生成模型提供了一种参数高效的解决方案。通过多哈希签名机制,有望实现更大规模的词汇覆盖、更强的稀有词表示能力,以及在多语言环境中的无缝扩展。这对推动自然语言理解与生成的研究具有深远意义,也为工业界提供了低成本、高性能的模型部署路径,尤其在多语种、多领域应用场景中具有广泛潜力。
技术贡献
技术上,MultiHashFormer首次提出支持因果生成的多哈希签名框架,结合多哈希函数和签名压缩机制,解决了哈希碰撞导致的不可判定性问题。模型设计中引入门控组合嵌入(Gated Compositional Embedding)和级联预测器(Cascade Predictor),实现离散哈希签名的自回归生成。该框架独立于具体序列处理骨架,兼容Transformer架构,并在不同参数规模下验证其优越性,展示了在参数效率和模型泛化能力上的突破。
新颖性
本研究的创新点在于引入多哈希签名机制支持因果生成,突破了传统哈希模型仅适用于编码器架构的限制。通过多哈希ID序列,避免了多对一碰撞问题,实现了大规模词汇的参数高效扩展。同时,结合门控机制和级联预测,确保生成的哈希签名唯一性和可判定性。这在生成模型领域尚属首次,将哈希编码与自回归生成无缝结合,开启了低参数、高容量的自然语言模型新路径。
局限性
- 模型在极端稀疏或长尾词汇的表示上仍存在一定挑战,尤其在哈希签名碰撞概率增加时可能影响表示质量。
- 哈希签名的离散性可能引入噪声,影响模型在某些任务中的精确性,尤其是在需要细粒度语义区分的场景中。
- 当前模型在多语言扩展中,虽然支持无参数增长,但在极大规模词汇或多模态融合场景下的表现仍需验证。
未来方向
未来可探索多哈希签名的动态调整机制,以适应不同任务的语义需求;结合更复杂的哈希函数或学习哈希策略,提升签名唯一性和鲁棒性;扩展模型在多模态、多任务、多领域的应用能力,推动大规模多语种生成模型的实用化。同时,研究如何优化哈希签名的存储与检索效率,进一步降低模型部署成本。
AI 总览摘要
在自然语言处理领域,构建大规模、参数高效的生成模型一直是核心挑战之一。传统的词嵌入矩阵随着词汇规模线性增长,导致模型参数迅速膨胀,限制了模型的扩展性和多语种适应能力。为解决这一瓶颈,本文提出了MultiHashFormer,一种基于多哈希签名的生成框架,支持因果语言建模。该方法通过多个独立哈希函数,将每个词映射为唯一的离散哈希ID序列,避免了多对一碰撞问题。模型由三大模块组成:哈希编码器(Hash Encoder)将离散签名压缩为潜在向量,Transformer解码器(Transformer Decoder)处理序列,哈希解码器(Hash Decoder)逐步重建下一词的哈希签名,最终映射回文本。这一设计使得模型在不同参数规模(100M、1B、3B)下均优于标准Transformer,特别是在多任务、多语种环境中表现出色。
实验结果显示,MultiHashFormer在多个基准测试中持续超越传统模型。例如,在1B参数模型中,H4B16K配置在ReCoRD任务中得分达到64.90,超越标准模型的63.78,提升约1.12分。同时,该模型支持在不增加参数的情况下,将词汇从32K扩展到48K,且在阿拉伯语、汉语、印地语等多语种任务中保持优异表现,验证了其在多语种、多任务场景中的潜力。
技术创新方面,本文首次提出支持因果生成的多哈希签名机制,结合门控组合嵌入(Gated Compositional Embedding)和级联预测器(Cascade Predictor),实现离散哈希签名的自回归生成。该机制有效避免了哈希碰撞带来的不确定性,确保模型在生成过程中签名的唯一性和可判定性。模型架构的独立性和参数效率,为大规模、多语种生成模型提供了新的设计思路。
未来工作将集中在优化哈希签名的鲁棒性,提升多模态融合能力,以及在更大规模词汇和多任务环境中的应用探索。总体而言,MultiHashFormer为自然语言生成提供了一种低成本、高容量的解决方案,推动了多语种、多任务智能系统的快速发展。
深度解读
原文摘要
Language models (LMs) represent tokens using embedding matrices that scale linearly with the vocabulary size. To constrain the parameter footprint, prior work proposes hashing many tokens into a single vector within encoder-only models. While this offers parameter efficiency, many-to-one collisions prevent its use in causal LMs. In this paper, we propose MultiHashFormer, a new framework that allows hash-based autoregression. Each token is represented as a unique hash signature, a short sequence of discrete hash IDs, generated by multiple independent hash functions. A Hash Encoder compresses this signature into a single latent vector for processing by a Transformer decoder. Then, a Hash Decoder generates the hash signature of the next token, which is then mapped back to text. We evaluate our approach at the 100M, 1B and 3B parameter scales, demonstrating that MultiHashFormer consistently outperforms standard Transformer LMs across multiple benchmarks. Furthermore, we show that our model handles multilingual vocabulary expansion with a constant parameter footprint without any modifications.
参考文献 (20)
RACE: Large-scale ReAding Comprehension Dataset From Examinations
Guokun Lai, Qizhe Xie, Hanxiao Liu 等
Breaking the Softmax Bottleneck: A High-Rank RNN Language Model
Zhilin Yang, Zihang Dai, R. Salakhutdinov 等
HellaSwag: Can a Machine Really Finish Your Sentence?
Rowan Zellers, Ari Holtzman, Yonatan Bisk 等
Crowdsourcing Multiple Choice Science Questions
Johannes Welbl, Nelson F. Liu, Matt Gardner
Canine: Pre-training an Efficient Tokenization-Free Encoder for Language Representation
J. Clark, Dan Garrette, Iulia Turc 等
HashFormers: Towards Vocabulary-independent Pre-trained Transformers
Huiyin Xue, Nikolaos Aletras
Continually Adding New Languages to Multilingual Language Models
Abraham Toluwase Owodunni, Sachin Kumar
AdaptiVocab: Enhancing LLM Efficiency in Focused Domains through Lightweight Vocabulary Adaptation
Itay Nakash, Nitay Calderon, E. B. David 等
Adapt-and-Distill: Developing Small, Fast and Effective Pretrained Language Models for Domains
Yunzhi Yao, Shaohan Huang, Wenhui Wang 等
How brains make chaos in order to make sense of the world
Ronald Rosenfeld, D. Touretzky
ReCoRD: Bridging the Gap between Human and Machine Commonsense Reading Comprehension
Sheng Zhang, Xiaodong Liu, Jingjing Liu 等
From Bytes to Ideas: Language Modeling with Autoregressive U-Nets
Mathurin Videau, Badr Youbi Idrissi, A. Leite 等
Parsing Natural Scenes and Natural Language with Recursive Neural Networks
R. Socher, Cliff Chiung-Yu Lin, A. Ng 等
SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text Processing
Taku Kudo, John Richardson
A Corpus and Cloze Evaluation for Deeper Understanding of Commonsense Stories
N. Mostafazadeh, Nathanael Chambers, Xiaodong He 等
Compressing Transformer-Based Semantic Parsing Models using Compositional Code Embeddings
P. Prakash, Saurabh Kumar Shashidhar, Wenlong Zhao 等
Measuring Massive Multitask Language Understanding
Dan Hendrycks, Collin Burns, Steven Basart 等
ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
Zhenzhong Lan, Mingda Chen, Sebastian Goodman 等