JEPA-Anything:跨领域通用预测框架,用正交分解拆解状态推演
开源框架 JEPA-Anything 试图用统一逻辑搞定不同学科的预测推演。它基于 JEPA 架构提出「正交预测分解」,将复杂状态拆解为独立要素分头推算再拼合,各领域仅需适配专属接口即可复用核心预测机制。目前仓库提供代码脚手架与合成示例,未包含预训练权重。

JEPA-Anything 是一个跨领域的通用预测框架,能帮你为生物、物理、医疗等不同领域构建统一的世界预测模型。
不管是单细胞变化、分子动力学演化,还是临床医疗轨迹分析,本质上都在做一件事:根据当前已知状态推测未来的演变。JEPA-Anything 试图用同一套学习逻辑,搞定不同学科里的预测推演。
它是怎么做到的
传统基于自监督或 JEPA 架构的模型,往往把要预测的目标当成一个不可分割的整体去硬猜。这种做法遇到复杂的多变量系统,很容易顾此失彼。
JEPA-Anything 的核心机制叫「正交预测分解」(Orthogonal Predictive Factorization,简称 OPF)。模型会先用编码器把观测数据映射到潜在世界状态,再把这个状态拆解成若干组互不干扰的预测坐标要素。
每个要素都走独立的预测通道推演变化,最后再重新拼回完整的潜在状态。拼合后的状态可以直接用于下游的属性读出、干预模拟、决策规划或多步长期推演。
各个领域的数据形态天差地别,所以每个学科依旧保留专属的数据观察接口和评估适配器,但核心的预测算子与状态接口完全共享。换句话说,不用为每个领域从头重写一套预测机制。
怎么跑起来与当前限制
这个项目目前在 GitHub 开源,环境要求 Python 3.10 或更高版本。核心预测库基于 PyTorch,而任务设计工具和示例只依赖 Python 标准库。
克隆仓库后,通过 pip 安装核心库,就能运行内置的合成线性动力学示例。工具包里还包含了设计校验工具和脚手架生成脚本,可以用 JSON 格式配置任务规范,一键生成适配器、测试和模型骨架。
必须说清楚的是,仓库目前提供的是通用的核心代码、脚手架工具以及用来跑通流程的合成数据示例。官方并没有放出在单细胞或临床数据集上训练好的大模型权重,图表中展示的生物学对应关系需要开发者自己用实验数据去训练和验证。
想要试用的话,可以访问 GitHub 上的 Gen-Verse/JEPA-Anything 仓库,按照文档在 Linux 或 macOS 终端中完成脚手架部署。
为什么重要
我们认为 JEPA-Anything 最聪明的地方,在于它没有尝试做一个包打天下的巨型端到端模型,而是把各领域的接口抽象出来,只统一底层的预测流。
对于从事 AI for Science、生物计算或复杂物理系统建模的研究者来说,它提供了一套现成的架构规范,省去了自己从零搭建潜在预测通道的重复劳动。不过在缺乏官方预训练权重的情况下,门槛仍然不低,普通开发者很难直接开箱即用,它更适合有自有领域数据并打算落地世界模型的科研团队。
来源 (1)
- github.com(在新标签页打开)原文github.com