连续控制

📄 Sub-JEPA: Subspace Gaussian Regularization for Stable End-to-End World Models #世界模型 #自监督学习 #连续控制 📝 5.0/10 | 前50% | #世界模型 | #自监督学习 | #连续控制 | arxiv 学术质量 5.0/8 | 影响力 0.8/2 | 可复现性 0.8/1 | 置信度 中 👥 作者与机构 第一作者：未提及 通讯作者：未提及 作者列表：未提及（原文摘要未列出作者信息） 💡 毒舌点评 这篇论文指出了现有JEPA方法（如LeWM）在完整潜在空间施加各向同性高斯先验可能过于严格、与低维流形假设相悖的痛点，并提出了一个在多个随机子空间施加约束的简单变体。其核心思想直观且有一定启发性，但实验验证部分仅在四个连续控制任务上进行，略显单薄，且未能充分展示该“子空间”约束相对于全局约束的理论或实践优势，更像是一个经验性的有效改进（trick），而非具有坚实理论基础的解决方案。 📌 核心摘要 要解决什么问题：联合嵌入预测架构（JEPA）在训练世界模型时，面临偏差-方差权衡问题。缺乏足够的结构约束会导致模型坍缩到平凡解（表示方差过大）。最近的LeWorldModel (LeWM) 通过施加全局各向同性高斯先验来缓解坍缩，但这种在完整嵌入空间上的强约束可能过于严格，因为它与潜在表示本就存在于高维空间中的低维流形这一事实相冲突。 方法核心是什么：本文提出Sub-JEPA，核心思想是不在完整的高维潜在空间施加全局高斯约束，而是在多个随机子空间中施加该约束。通过放松全局约束、保留反坍缩效果，在训练稳定性和表示灵活性之间寻求更好的平衡点。 与已有方法相比新在哪里：新在约束的作用域。传统方法（包括LeWM）在原始（或编码后的）完整潜在空间上施加分布约束。Sub-JEPA将约束施加在随机投影的子空间中，旨在更好地适应数据的低维流形结构。 主要实验结果如何：论文声称在四个连续控制环境中，Sub-JEPA始终以明显的优势优于LeWM。但摘要中未提供具体数值、环境名称、评估指标或与其他基线的对比数据。 实际意义是什么：为训练稳定、灵活的基于JEPA的世界模型提供了一个简单有效的改进方案，有望作为未来JEPA类世界模型研究的强基线，推动其在连续控制等任务中的应用。 主要局限性是什么：根据摘要，主要局限是实验验证范围有限（仅四个连续控制环境），可能影响结论的普适性。此外，对于为何“子空间约束”优于“全局约束”的理论解释可能不够深入。 🔗 开源详情 代码：https://github.com/intcomp/Sub-JEPA 模型权重：未提及 数据集：未提及 Demo：未提及 复现材料：未提及 论文中引用的开源项目：未提及 🏗️ 方法概述和架构 1. 整体流程概述 Sub-JEPA是一个端到端的联合嵌入预测框架，用于学习世界模型。其核心流程是：给定当前观测和历史信息，模型预测未来观测的潜在表示。训练时，为了避免模型坍缩并鼓励有意义的表示学习，它在多个随机采样的子空间中对预测的潜在表示施加高斯分布约束，而非在整个潜在空间。这是一个自监督学习框架，通过预测任务本身和子空间正则化来共同优化编码器和预测器。 2. 主要组件/模块详解 感知编码器（Perception Encoder）： 功能：将高维原始观测（如图像）映射到一个潜在表示空间。这是JEPA架构中与任务相关的可学习部分。 内部结构/实现：论文中未具体说明网络结构。假设其输出为一个向量表示 z。 输入输出：输入原始观测数据，输出潜在表示 z。 预测器（Predictor）： ...