中文

Pico:面向假设驱动的小型语言模型研究的模块化框架

计算与语言 2025-09-23 v1 人工智能

摘要

构建语言模型,尤其是中小型模型,目前仍是一门艺术而非科学。尽管大型语言模型通常通过纯粹的规模扩展来改进,但许多设计选择为何有效仍不清楚。对于小型语言模型,这种不确定性更具限制性:紧张的参数预算使每个决策都至关重要,然而研究人员仍然缺乏系统、科学的方法来测试和改进新想法。我们引入了 Pico,一个轻量级、模块化的框架,支持面向中小型语言模型开发的系统性、假设驱动研究。Pico 由两个库组成,它们共同提供了一个实用的沙盒,研究人员可以在其中对模型的架构或训练过程进行针对性修改,并直接观察其对模型行为的影响。为了支持可复现的实验,我们还发布了一套基准模型 pico-decoder,该模型在标准化条件下训练并面向社区开源。案例研究突出了 Pico 如何支持迭代的小型语言模型设计与分析。

关键词

引用

@article{arxiv.2509.16413,
  title  = {Pico: A Modular Framework for Hypothesis-Driven Small Language Model Research},
  author = {Richard Diehl Martinez and David Demitri Africa and Yuval Weiss and Suchir Salhan and Ryan Daniels and Paula Buttery},
  journal= {arXiv preprint arXiv:2509.16413},
  year   = {2025}
}