arXiv:2510.01796cs.LGcs.AI2025-10被引 2

提出倒金字塔结构MLP,提升生成性能且更高效

Rethinking the shape convention of an MLP

  • 跳接在宽层间进行,残差计算走窄瓶颈
  • 在图像生成任务中,参数越少效果越好
  • 适合追求高效率的生成模型设计者

多层感知机(MLPs)传统采用窄-宽-窄结构,跳接在输入/输出维度操作,计算在扩展的隐藏空间中进行。本文挑战这一惯例,提出宽-窄-宽(小时钟形)MLP模块:跳接在扩展维度进行,残差计算通过窄瓶颈流动。该设计利用高维空间进行增量优化,同时通过参数匹配保持计算效率。实现小时钟形MLP需初始投影将输入信号升维,我们提出该投影可固定为随机初始化,无需训练,从而实现高效训练与推理。我们在主流图像数据集上评估两种架构的生成性能,通过系统性架构搜索构建性能-参数帕累托前沿。结果表明,小时钟形架构在所有参数预算下均优于传统设计;随着参数增加,最优小时钟形配置倾向于更深网络、更宽跳接和更窄瓶颈——这一缩放规律与传统MLP明显不同。研究提示应重新思考现代架构中跳接位置,潜在适用于Transformer及其他残差网络。

原文摘要 · Abstract (English)

Multi-layer perceptrons (MLPs) conventionally follow a narrow-wide-narrow design where skip connections operate at the input/output dimensions while processing occurs in expanded hidden spaces. We challenge this convention by proposing wide-narrow-wide (Hourglass) MLP blocks where skip connections operate at expanded dimensions while residual computation flows through narrow bottlenecks. This inversion leverages higher-dimensional spaces for incremental refinement while maintaining computational efficiency through parameter-matched designs. Implementing Hourglass MLPs requires an initial projection to lift input signals to expanded dimensions. We propose that this projection can remain fixed at random initialization throughout training, enabling efficient training and inference implementations. We evaluate both architectures on generative tasks over popular image datasets, characterizing performance-parameter Pareto frontiers through systematic architectural search. Results show that Hourglass architectures consistently achieve superior Pareto frontiers compared to conventional designs. As parameter budgets increase, optimal Hourglass configurations favor deeper networks with wider skip connections and narrower bottlenecks-a scaling pattern distinct from conventional MLPs. Our findings suggest reconsidering skip connection placement in modern architectures, with potential applications extending to Transformers and other residual networks.

MLP生成模型架构设计

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