MergePipe让大模型合并更高效,通过预算管控减少磁盘读写
MergePipe: A Budget-Aware Parameter Management System for Scalable LLM Merging
- 将模型合并视为数据管理问题,用目录抽象统一参数与执行流程
- 引入成本感知规划器,控制专家参数读取,降低总磁盘I/O达90%
- 适合需要大规模模型合并且资源受限的开发者使用
大型语言模型(LLM)合并已成为现代LLM开发的关键技术,可在不重新训练的情况下整合多个任务或领域专用的专家模型。然而,随着专家数量增加,现有实现将模型参数视为无结构文件,以无状态、一次性方式执行合并,导致过度的磁盘I/O、冗余参数扫描和可扩展性差。本文提出 extbf{MergePipe},一种面向可扩展LLM合并的参数管理系统。MergePipe是首个将LLM合并视为数据管理与执行问题的系统,引入基于目录的抽象,覆盖模型参数、合并计划和执行轨迹。其核心是一个成本感知规划器,显式建模专家参数I/O并强制执行用户指定的I/O预算,搭配流式执行引擎,在事务保证下生成合并模型。关键洞察是:基础模型读取和输出写入不可避免,但专家参数读取主导合并成本,是主要优化目标。通过在规划与执行全程实现专家访问预算感知,MergePipe缓解了原始流水线中$O(K)$的I/O增长,实现可预测的扩展行为。实验表明,MergePipe将总I/O降低达一个数量级,相比最先进方案实现高达11倍的端到端加速(最高节省90%墙钟时间)。
原文摘要 · Abstract (English)
Large language model (LLM) merging has become a key technique in modern LLM development pipelines, enabling the integration of multiple task- or domain-specific expert models without retraining. However, as the number of experts grows, existing merging implementations treat model parameters as unstructured files and execute merges in a stateless, one-shot manner, leading to excessive disk I/O, redundant parameter scans, and poor scalability. In this paper, we present \textbf{MergePipe}, a parameter management system for scalable LLM merging. MergePipe is the first system that treats LLM merging as a data management and execution problem, and introduces a catalog-driven abstraction over model parameters, merge plans, and execution lineage. At its core, MergePipe employs a cost-aware planner that explicitly models expert parameter I/O and enforces user-specified I/O budgets, followed by a streaming execution engine that materializes merged models under transactional guarantees. Our key insight is that while base model reads and output writes are unavoidable, expert parameter reads dominate merge cost and constitute the primary optimization target. By making expert access budget-aware throughout planning and execution, MergePipe mitigates the $O(K)$ I/O growth of naive pipelines and achieves predictable scaling behavior. Experiments show that MergePipe reduces total I/O by up to an order of magnitude and delivers up to $11\times$ end-to-end speedups (up to 90\% wall-time reduction) over state-of-the-art LLM merging pipelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。