用可组合专家模型实现鲁棒生成,避免调参难题。
A Theoretical Framework for Modular Learning of Robust Generative Models
- 用门控机制拼接预训练专家,实现模块化生成
- 理论证明门控函数能抵御最差数据混合,性能超越单一模型
- 适合需要灵活适配多场景的生成系统开发者
大规模生成模型训练成本高且依赖启发式数据加权。本文解决两个根本问题:能否通过组合小型领域专家模块化训练大语言模型,达到单体模型性能;能否在任意数据混合下实现鲁棒性,无需启发式调参?我们提出模块化生成建模的理论框架,通过门控机制组合预训练专家。定义归一化门控函数空间 $\ ext{\mathcal{G}}_1$,将问题建模为最小最大博弈,寻找使对最差数据混合的偏差最小的鲁棒门控。利用Kakutani不动点定理证明此类鲁棒门控的存在性,并表明模块化具有强正则化效果,泛化界与轻量门控复杂度相关。进一步证明该方法理论上优于在聚合数据上重新训练的模型,差距由Jensen-Shannon散度刻画。最后引入可扩展的随机原始-对偶算法和结构蒸馏方法,实现高效推理。在合成与真实数据集上的实验表明,该模块化架构有效缓解梯度冲突,稳健优于单体基线。
原文摘要 · Abstract (English)
Training large-scale generative models is resource-intensive and relies heavily on heuristic dataset weighting. We address two fundamental questions: Can we train Large Language Models (LLMs) modularly, combining small, domain-specific experts to match monolithic performance, and can we do so robustly for any data mixture, eliminating heuristic tuning? We present a theoretical framework for modular generative modeling where a set of pre-trained experts are combined via a gating mechanism. We define the space of normalized gating functions $\mathcal{G}_{1}$ and formulate the problem as a minimax game to find a single robust gate that minimizes divergence to the worst-case data mixture. We prove the existence of such a robust gate using Kakutani's fixed-point theorem and show that modularity acts as a strong regularizer, with generalization bounds scaling with the lightweight gate's complexity. Furthermore, we prove that this modular approach can theoretically outperform models retrained on aggregate data, with the gap characterized by the Jensen-Shannon Divergence. Finally, we introduce a scalable Stochastic Primal-Dual algorithm and a Structural Distillation method for efficient inference. Empirical results on synthetic and real-world datasets confirm that our modular architecture effectively mitigates gradient conflict and can robustly outperform monolithic baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。