Hungry Hungry Hippos: Towards Language Modeling with State Space Models

Benchmark Model Rank Results
language-modelling-on-wikitext-103Hybrid H3 (2.7B)#2Test perplexity: 10.6Number of params: 2700M
language-modelling-on-wikitext-103Hybrid H3 (1.3B)#7Test perplexity: 12.5Number of params: 1300M
language-modelling-on-wikitext-103Hybrid H3 (355M)#22Test perplexity: 16.9Number of params: 355M
language-modelling-on-wikitext-103Hybrid H3 125M#41Test perplexity: 18.5
language-modelling-on-wikitext-103Hybrid H3 (125M)#55Test perplexity: 23.7Number of params: 125M
word-sense-disambiguation-on-words-in-contextHybrid H3 125M (0-shot, logit scoring)#18Accuracy: 51.4
word-sense-disambiguation-on-words-in-contextHybrid H3 125M (0-shot, rank classification)#19Accuracy: 51.4
word-sense-disambiguation-on-words-in-contextHybrid H3 125M (3-shot, logit scoring)#26Accuracy: 49.1
question-answering-on-multircHybrid H3 355M (3-shot, logit scoring)#25EM: 59.7
question-answering-on-multircHybrid H3 355M (0-shot, logit scoring)#26EM: 59.5
question-answering-on-multircHybrid H3 125M (0-shot, logit scoring)#27EM: 51.4
question-answering-on-multircHybrid H3 125M (3-shot, logit scoring)#28EM: 48.9
language-modelling-on-the-pileHybrid H3 125M#28Test perplexity: 10.2
language-modelling-on-the-pileTransformer 125M#30Test perplexity: 10.7
question-answering-on-copaHybrid H3 2.7B (0-shot, logit scoring)#32Accuracy: 81
question-answering-on-copaHybrid H3 2.7B (3-shot, logit scoring)#38Accuracy: 77
question-answering-on-copaHybrid H3 125M (0-shot, logit scoring)#42Accuracy: 67
question-answering-on-copaHybrid H3 125M (0-shot, rank classification)#43Accuracy: 67
question-answering-on-copaH3 125M (0-shot, rank classification)#47Accuracy: 51
coreference-resolution-on-winograd-schema-challengeH3 125M (3-shot, rank classification)#40Accuracy: 63.5
coreference-resolution-on-winograd-schema-challengeH3 125M (0-shot, rank classification)#46Accuracy: 61.5
coreference-resolution-on-winograd-schema-challengeHybrid H3 125M (3-shot, logit scoring)#60Accuracy: 43.3
natural-language-inference-on-rteHybrid H3 125M (0-shot, logit scoring)#69Accuracy: 59.2%
natural-language-inference-on-rteHybrid H3 125M (3-shot, logit scoring)#72Accuracy: 58.1%
natural-language-inference-on-rteHybrid H3 125M (3-shot, rank classification)#73Accuracy: 58.1%
natural-language-inference-on-rteH3 125M (0-shot, rank classification)#81Accuracy: 53.1%
natural-language-inference-on-rteH3 125M (3-shot, rank classification)#82Accuracy: 52.3%
question-answering-on-boolqHybrid H3 1.3B (0-shot, logit scoring)#92Accuracy: 61.7
question-answering-on-boolqHybrid H3 2.7B (3-shot, logit scoring)#99Accuracy: 60.6
question-answering-on-boolqHybrid H3 125M (0-shot, logit scoring)#106Accuracy: 59.6
question-answering-on-boolqHybrid H3 125M (3-shot, logit scoring)#111Accuracy: 56.1
question-answering-on-boolqHybrid H3 125M (3-shot, rank classification)#112Accuracy: 56.1