arXiv:2605.19758cs.AIcs.DB2026-05被引 1

用14个可扩展合成任务快速评估模型处理序列的能力。

CogScale: Scalable Benchmark for Sequence Processing

论文配图:CogScale: Scalable Benchmark for Sequence Processing
图 1 · 摘自论文原文
  • 设计14个可调节难度的合成任务,分离测试记忆与推理能力。
  • 在1k~100k参数预算下,注意力机制和状态空间模型表现更优。
  • 适合快速验证新架构,避免盲目投入大规模训练。

维持并操作时间信息是生命体和人工智能的基本能力。尽管现代模型在自然语言处理等任务中取得显著进展,但评估新架构处理序列信息的能力仍需耗费大量计算资源且迭代缓慢。通常需扩展至超大规模数据集和模型,导致成本高昂。本文提出CogScale,一个包含14个可扩展合成任务的基准,用于在不同可调节尺度下独立评估特定认知与记忆能力。通过标准化、轻量化的框架,CogScale使研究者能在投入大规模训练前快速验证架构创新。为建立基线,我们在严格参数预算(1k、10k、100k)下评估了七种架构:GRU、LSTM、xLSTM、ESN、Mamba、Transformer Decoder 和 Transformer Encoder-Decoder,覆盖不同难度与规模。结果表明,传统RNN和ESN在小参数下擅长基础记忆,而仅注意力机制和现代状态空间模型能持续保持高性能,随推理复杂度与任务难度提升。

原文摘要 · Abstract (English)

The ability to maintain and manipulate information over time is a fundamental aspect of living beings and Artificial Intelligence. While modern models have achieved remarkable success in tasks like natural language processing, evaluating the capacity of novel architectures to process sequential information remains computationally expensive and time-consuming. Testing a new architecture often requires scaling up to massive datasets and models, leading to vast computational costs and slow iteration cycles. In this paper, we propose CogScale, a benchmark of 14 scalable synthetic tasks designed to isolate and evaluate specific cognitive and memory abilities at different parametrizable scales. By providing a standardized, lightweight framework, CogScale allows researchers to rapidly validate architectural innovations before committing to large-scale training. To establish a solid baseline, we evaluate seven distinct architectures: Gated Recurrent Unit (GRU), Long Short-Term Memory (LSTM), xLSTM, Echo State Network (ESN), Mamba, Transformer Decoder, and Transformer Encoder-Decoder. These evaluations are conducted under strict parameter budgets (1k, 10k, and 100k) and across different difficulty levels and scales. Our results show that while classical RNNs and Echo State Networks excel at basic retention within strict parameter budgets, only attention mechanisms and modern state-space models consistently maintain high performance as reasoning complexity and task difficulty scale.

序列建模基准测试状态空间轻量化评估

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。